set all of decode_context in decoder

Signed-off-by: Somtochi Onyekwere <[email protected]>

Author: somtochiama

src/messages/codec.rs

@@ -70,7 +70,7 @@ where
 {
     if let Some(msg_len) = get_length(buf, offset) {
         if msg_len > max_size {
-            return Err(PgWireError::MessageTooLarge(msg_len, max_size));
+            return Err(PgWireError::MessageTooLarge(max_size, msg_len));
         }
 
         if buf.remaining() >= msg_len + offset {

src/messages/mod.rs

@@ -103,8 +103,8 @@ pub trait Message: Sized {
         let len = self.message_length();
         if len > Self::max_message_length() {
             return Err(PgWireError::MessageTooLarge(
-                len,
                 Self::max_message_length(),
+                len,
             ));
         }
 

src/tokio/server.rs

@@ -48,14 +48,19 @@ impl<S> Decoder for PgWireMessageServerCodec<S> {
         self.decode_context.protocol_version = self.client_info.protocol_version;
 
         match self.client_info.state() {
-            PgWireConnectionState::AwaitingSslRequest => {}
+            PgWireConnectionState::AwaitingSslRequest => {
+                self.decode_context.awaiting_ssl = true;
+                self.decode_context.awaiting_startup = true;
+            }
 
             PgWireConnectionState::AwaitingStartup => {
                 self.decode_context.awaiting_ssl = false;
+                self.decode_context.awaiting_startup = true;
             }
 
             _ => {
                 self.decode_context.awaiting_startup = false;
+                self.decode_context.awaiting_ssl = false;
             }
         }