Send keep-alive request to remote after successful upload

jclehner · Feb 17, 2016 · bc0901e · bc0901e
1 parent b95bfc9
commit bc0901e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nmrp.c b/nmrp.c
@@ -531,6 +531,7 @@ int nmrp_do(struct nmrpd_args *args)
 						printf("Ignoring extra upload request.\n");
 					}
 					ethsock_set_timeout(sock, args->ul_timeout);
+					tx.msg.code = NMRP_C_KEEP_ALIVE_REQ;
 					break;
 				}
 
@@ -589,6 +590,7 @@ int nmrp_do(struct nmrpd_args *args)
 					printf("OK\nWaiting for remote to respond.\n");
 					upload_ok = 1;
 					ethsock_set_timeout(sock, args->ul_timeout);
+					tx.msg.code = NMRP_C_KEEP_ALIVE_REQ;
 					expect = NMRP_C_NONE;
 				} else if (status == -2) {
 					expect = NMRP_C_TFTP_UL_REQ;