Fix a bug in protocol

Correctly handle the impossibility to add the
tailing record separator.

Bug-AGL: SPEC-3168

Change-Id: Id611eda4c6517819c1209baeb58c2d8d8ff57183
Signed-off-by: José Bollo <jose.bollo@iot.bzh>

diff --git a/src/prot.c b/src/prot.c
index 7edcc50..2714e2f 100644 (file)
--- a/src/prot.c
+++ b/src/prot.c
@@ -387,7 +387,8 @@ prot_put_end(
                rc = 0;
        else {
                rc = buf_put_car(&prot->outbuf, RECORD_SEPARATOR);
-               prot->outfields = 0;
+               if (rc == 0)
+                       prot->outfields = 0;
        }
        return rc;
 }
@@ -440,10 +441,10 @@ prot_put(
        int rc;
 
        rc = prot_put_fields(prot, count, fields);
+       if (!rc)
+               rc = prot_put_end(prot);
        if (rc)
                prot_put_cancel(prot);
-       else
-               rc = prot_put_end(prot);
        return rc;
 }