check dst bump while incrementing, remove needless bump at end (#9)

* check dst bump while incrementing, remove needless bump at end * fix
charlesnicholson · Apr 22, 2021 · 4515f44 · 4515f44
1 parent 39a130b
commit 4515f44
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/cobs.c b/cobs.c
@@ -78,8 +78,8 @@ cobs_ret_t cobs_encode(void const *dec,
   while (dec_len--) {
     cobs_byte_t const byte = *src;
     if (byte) {
-      *dst++ = byte;
-      if (dst >= dst_max) { return COBS_RET_ERR_EXHAUSTED; }
+      *dst = byte;
+      if (++dst >= dst_max) { return COBS_RET_ERR_EXHAUSTED; }
       ++code;
     }
 
@@ -89,8 +89,7 @@ cobs_ret_t cobs_encode(void const *dec,
       code = 1;
 
       if ((byte == 0) || dec_len) {
-        ++dst;
-        if (dst >= dst_max) { return COBS_RET_ERR_EXHAUSTED; }
+        if (++dst >= dst_max) { return COBS_RET_ERR_EXHAUSTED; }
       }
     }
     ++src;