X-Git-Url: https://code.wpia.club/?a=blobdiff_plain;f=src%2Fio%2Frecord.h;h=5d6e659b7439689157c704a303eefbf7cfcc0d64;hb=5163a9658ba7ef4a875ee1103e82c0be81698689;hp=88576651ea574c25c2c7f2ebf83ad44e8ec6d5f1;hpb=f69f31caeda734d6d9c8ab00e693671ac7512bea;p=cassiopeia.git

diff --git a/src/io/record.h b/src/io/record.h
index 8857665..5d6e659 100644
--- a/src/io/record.h
+++ b/src/io/record.h
@@ -4,8 +4,9 @@
 
 #include <memory>
 #include <string>
-
+#include <exception>
 #include "bios.h"
+#include "io/opensslBIO.h"
 
 #define RECORD_HEADER_SIZE 17
 
@@ -59,10 +60,10 @@ public:
     template <class T>
     static void read( std::string::const_iterator& it, T& val ) {
         union typeConversion {
-            char buf[sizeof(T)];
+            char buf[sizeof( T )];
             T value;
 
-            typeConversion(const T& v) : value(v) {}
+            typeConversion( const T& v ) : value( v ) {}
         };
 
         typeConversion data( 0 );
@@ -89,7 +90,7 @@ public:
 
     void unpackFromString( const std::string& str ) {
         if( str.size() != RECORD_HEADER_SIZE ) {
-            throw "Invalid string length";
+            throw std::runtime_error( "Invalid string length" );
         }
 
         auto it =  str.cbegin();
@@ -101,9 +102,12 @@ public:
         read( it, offset );
         read( it, payloadLength );
     }
-
+    bool isFollowupOf( const RecordHeader& head ) {
+        return head.command == command && head.flags == flags && head.sessid == sessid && head.command_count == command_count && head.totalLength == totalLength && head.offset + 1 == offset;
+    }
 };
 
-std::string parseCommand( RecordHeader& head, const std::string& input, std::shared_ptr<std::ostream> log );
+std::string parseCommand( RecordHeader& head, const std::string& input );
+std::string parseCommandChunked( RecordHeader& head, std::shared_ptr<OpensslBIOWrapper> conn );
 
-void sendCommand( RecordHeader& head, const std::string& data, std::shared_ptr<OpensslBIO> bio, std::shared_ptr<std::ostream> log );
+void sendCommand( RecordHeader& head, const std::string& data, std::shared_ptr<OpensslBIO> bio );