version 3.0

ssh/ssh_filter.h

@@ -0,0 +1,55 @@
+////////////////////////////////////////////////////////////////////////////////
+//                                                                            //
+// File:           ssh_filter.h                                               //
+// Description:    SSH encryption/decryption inside a data filter pair        //
+// Project:        WANProxy XTech                                             //
+// Author:         Andreu Vidal Bramfeld-Software                             //
+// Last modified:  2015-04-01                                                 //
+//                                                                            //
+////////////////////////////////////////////////////////////////////////////////
+
+#ifndef	SSH_ENCRYPT_FILTER_H
+#define	SSH_ENCRYPT_FILTER_H
+
+#include <common/filter.h>
+#include <ssh/ssh_session.h>
+
+namespace SSH 
+{
+	const int SOURCE_ENCODED = 0x01;
+	const int ALGORITHM_NEGOTIATED = 0x2C;
+	
+	class EncryptFilter : public BufferedFilter
+	{
+	private:
+		Session session_;
+		bool encoded_, negotiated_;
+		
+	public:
+		EncryptFilter (Role role, int flg = 0);
+
+		virtual bool consume (Buffer& buf);
+		virtual bool produce (Buffer& buf);
+		virtual void flush (int flg);
+		
+		Session* current_session ()   { return &session_; }
+	};
+
+	class DecryptFilter : public LogisticFilter
+	{
+	private:
+		Buffer first_block_;
+		Session* session_;
+		bool encoded_, identified_;
+		
+	public:
+		DecryptFilter (int flg = 0);
+
+		virtual bool consume (Buffer& buf);
+		virtual void flush (int flg);
+		
+		void set_encrypter (EncryptFilter* f)   { session_ = (f ? f->current_session () : 0); set_upstream (f); }
+	};
+}
+
+#endif /* !SSH_ENCRYPT_FILTER_H */