X-Git-Url: http://v3vee.org/palacios/gitweb/gitweb.cgi?a=blobdiff_plain;f=palacios%2Fsrc%2Fpalacios%2Fvmm_string.c;h=01c408c8c3ccbd10bd98cd81b2591e8e9ba97b86;hb=bbfa5bc2f12eab6c8dc8f2640bcfd66d18475272;hp=725fa17a40c87bc0f20e1c9520ee03285c39c792;hpb=94f67717b6461df514dc225ed84f03b44c44061b;p=palacios.git

diff --git a/palacios/src/palacios/vmm_string.c b/palacios/src/palacios/vmm_string.c
index 725fa17..01c408c 100644
--- a/palacios/src/palacios/vmm_string.c
+++ b/palacios/src/palacios/vmm_string.c
@@ -71,6 +71,11 @@ void * memcpy(void * dst, const void * src, size_t n) {
 #ifdef V3_CONFIG_BUILT_IN_MEMMOVE
 void * memmove(void * dst, const void * src, size_t n) {
     uint8_t * tmp = (uint8_t *)V3_Malloc(n);
+
+    if (!tmp) {
+	PrintError(info->vm_info, info, "Cannot allocate in built-in memmove\n");
+	return NULL;
+    }
     
     memcpy(tmp, src, n);
     memcpy(dst, tmp, n);
@@ -285,6 +290,12 @@ char * strdup(const char * s1) {
     char *ret;
 
     ret = V3_Malloc(strlen(s1) + 1);
+
+    if (!ret) {
+        PrintError(VM_NONE, VCORE_NONE, "Cannot allocate in built-in strdup\n");
+	return NULL;
+    }
+
     strcpy(ret, s1);
 
     return ret;
@@ -460,10 +471,14 @@ size_t strcspn(const char * s, const char * reject) {
 	for (i = 0; i < reject_len; i++) {
 	    if (s[cnt] == reject[i]) {
 		match = 1;
-		cnt++;
 		break;
 	    }
 	}
+
+	if (!match) {
+	    cnt++;
+	}
+
     }
 
     return cnt;