From 1d61e446cbc84b41feac27d5e08e52a9c8af6644 Mon Sep 17 00:00:00 2001
From: whitequark <whitequark@whitequark.org>
Date: Sat, 8 Aug 2015 14:12:28 +0300
Subject: [PATCH] session.c: ensure session reset on out buffer overrun during
 RPC.

---
 soc/runtime/session.c | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/soc/runtime/session.c b/soc/runtime/session.c
index eea96fee5..1ad70c272 100644
--- a/soc/runtime/session.c
+++ b/soc/runtime/session.c
@@ -474,7 +474,8 @@ static int process_input(void)
 
 static int send_rpc_value(const char **tag, void *value)
 {
-    out_packet_int8(**tag);
+    if(!out_packet_int8(**tag))
+        return -1;
 
     int size = 0;
     switch(**tag) {
@@ -484,23 +485,27 @@ static int send_rpc_value(const char **tag, void *value)
 
         case 'b': // bool
             size = 1;
-            out_packet_chunk(value, size);
+            if(!out_packet_chunk(value, size))
+                return -1;
             break;
 
         case 'i': // int(width=32)
             size = 4;
-            out_packet_chunk(value, size);
+            if(!out_packet_chunk(value, size))
+                return -1;
             break;
 
         case 'I': // int(width=64)
         case 'f': // float
             size = 8;
-            out_packet_chunk(value, size);
+            if(!out_packet_chunk(value, size))
+                return -1;
             break;
 
         case 'F': // Fraction
             size = 16;
-            out_packet_chunk(value, size);
+            if(!out_packet_chunk(value, size))
+                return -1;
             break;
 
         case 'l': { // list(elt='a)
@@ -604,7 +609,7 @@ static int process_kmsg(struct msg_base *umsg)
 
             if(!send_rpc_request(msg->rpc_num, msg->args)) {
                 log("Failed to send RPC request");
-                return 0;
+                return 0; // restart session
             }
 
             user_kernel_state = USER_KERNEL_WAIT_RPC;
@@ -617,6 +622,7 @@ static int process_kmsg(struct msg_base *umsg)
                 umsg->type);
             return 0;
     }
+
     return 1;
 }