add sanity checks

wolfSSL · Oct 10, 2024 · 8a1b62f · 8a1b62f
1 parent 52e4e32
commit 8a1b62f
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/internal.c b/src/internal.c
@@ -15720,16 +15720,22 @@ int wolfSSH_GetPath(const char* defaultPath, byte* in, word32 inSz,
     }
 
     if (out != NULL) {
-        WMEMCPY(out + curSz, in, inSz);
+        if (curSz + inSz < *outSz) {
+            WMEMCPY(out + curSz, in, inSz);
+        }
     }
     curSz += inSz;
     if (out != NULL) {
         if (!WOLFSSH_SFTP_IS_DELIM(out[0])) {
-            WMEMMOVE(out+1, out, curSz);
-            out[0] = '/';
+            if (curSz + 1 < *outSz) {
+                WMEMMOVE(out+1, out, curSz);
+                out[0] = '/';
+            }
             curSz++;
         }
-        out[curSz] = 0;
+        if (curSz < *outSz) {
+            out[curSz] = 0;
+        }
     }
     *outSz = curSz;