dh: move state changes inside DH code

Signed-off-by: Aris Adamantiadis <aris@0xbadc0de.be>
Reviewed-by: Jakub Jelen <jjelen@redhat.com>
Reviewed-by: Andreas Schneider <asn@cryptomilk.org>

src/client.c

@@ -274,7 +274,6 @@ static int dh_handshake(ssh_session session) {
           return SSH_ERROR;
       }
 
-      session->dh_handshake_state = DH_STATE_INIT_SENT;
     case DH_STATE_INIT_SENT:
     	/* wait until ssh_packet_dh_reply is called */
     	break;

src/curve25519.c

@@ -79,6 +79,7 @@ int ssh_client_curve25519_init(ssh_session session){
   }
   /* register the packet callbacks */
   ssh_packet_set_callbacks(session, &ssh_curve25519_client_callbacks);
+  session->dh_handshake_state = DH_STATE_INIT_SENT;
   rc = ssh_packet_send(session);
 
   return rc;

src/dh.c

@@ -516,6 +516,7 @@ int ssh_client_dh_init(ssh_session session){
 
   /* register the packet callbacks */
   ssh_packet_set_callbacks(session, &ssh_dh_client_callbacks);
+  session->dh_handshake_state = DH_STATE_INIT_SENT;
 
   rc = ssh_packet_send(session);
   return rc;