AppleTLS: Implement AppleTLS and Apple Message Digest

1 files changed, 15 insertions, 18 deletions

diff --git a/src/LibgnutlsTLSSession.h b/src/LibgnutlsTLSSession.h
index 7118ab21..48b7f121 100644
--- a/src/LibgnutlsTLSSession.h
+++ b/src/LibgnutlsTLSSession.h
@@ -39,31 +39,28 @@
 
 #include <gnutls/gnutls.h>
 
-#include <string>
-
-#include "TLSSessionConst.h"
+#include "LibgnutlsTLSContext.h"
+#include "TLSSession.h"
 #include "a2netcompat.h"
 
 namespace aria2 {
 
-class TLSContext;
-
-class TLSSession {
+class GnuTLSSession : public TLSSession {
 public:
-  TLSSession(TLSContext* tlsContext);
-  ~TLSSession();
-  int init(sock_t sockfd);
-  int setSNIHostname(const std::string& hostname);
-  int closeConnection();
-  int checkDirection();
-  ssize_t writeData(const void* data, size_t len);
-  ssize_t readData(void* data, size_t len);
-  int tlsConnect(const std::string& hostname, std::string& handshakeErr);
-  int tlsAccept();
-  std::string getLastErrorString();
+  GnuTLSSession(GnuTLSContext* tlsContext);
+  ~GnuTLSSession();
+  virtual int init(sock_t sockfd);
+  virtual int setSNIHostname(const std::string& hostname);
+  virtual int closeConnection();
+  virtual int checkDirection();
+  virtual ssize_t writeData(const void* data, size_t len);
+  virtual ssize_t readData(void* data, size_t len);
+  virtual int tlsConnect(const std::string& hostname, std::string& handshakeErr);
+  virtual int tlsAccept();
+  virtual std::string getLastErrorString();
 private:
   gnutls_session_t sslSession_;
-  TLSContext* tlsContext_;
+  GnuTLSContext* tlsContext_;
   // Last error code from gnutls library functions
   int rv_;
 };