Merge branch 'dev' of https://github.com/OpenWaterAnalytics/EPANET into dev-list

src/epanet.c

@@ -30,6 +30,8 @@
 #include "text.h"
 #include "enumstxt.h"
 
+#include "util/cstr_helper.h"
+
 #ifdef _WIN32
 #define snprintf _snprintf
 #endif
@@ -1731,6 +1733,9 @@ int DLLEXPORT EN_addnode(EN_Project p, char *id, int nodeType)
     if (!p->Openflag) return 102;
     if (hyd->OpenHflag || qual->OpenQflag) return 262;
 
+    // Check if id contains invalid characters
+    if (!cstr_isvalid(id)) return 252;
+
     // Check if a node with same id already exists
     if (EN_getnodeindex(p, id, &i) == 0) return 215;
 
@@ -2016,7 +2021,7 @@ int DLLEXPORT EN_setnodeid(EN_Project p, int index, char *newid)
     if (index <= 0 || index > net->Nnodes) return 203;
     n = strlen(newid);
     if (n < 1 || n > MAXID) return 209;
-    if (strcspn(newid, " ;") < n) return 209;
+    if (!cstr_isvalid(newid)) return 252;
 
     // Check if another node with same name exists
     if (hashtable_find(net->NodeHashTable, newid) > 0) return 215;
@@ -2939,6 +2944,9 @@ int DLLEXPORT EN_addlink(EN_Project p, char *id, int linkType,
     if (!p->Openflag) return 102;
     if (p->hydraul.OpenHflag || p->quality.OpenQflag) return 262;
 
+    // Check if id contains invalid characters
+    if (!cstr_isvalid(id)) return 252;
+
     // Check if a link with same id already exists
     if (EN_getlinkindex(p, id, &i) == 0) return 215;
 
@@ -3200,7 +3208,7 @@ int DLLEXPORT EN_setlinkid(EN_Project p, int index, char *newid)
     if (index <= 0 || index > net->Nlinks) return 204;
     n = strlen(newid);
     if (n < 1 || n > MAXID) return 211;
-    if (strcspn(newid, " ;") < n) return 211;
+    if (!cstr_isvalid(newid)) return 252;
 
     // Check if another link with same name exists
     if (hashtable_find(net->LinkHashTable, newid) > 0) return 215;
@@ -3951,6 +3959,9 @@ int DLLEXPORT EN_addpattern(EN_Project p, char *id)
     if (!p->Openflag) return 102;
     if (EN_getpatternindex(p, id, &i) == 0) return 215;
 
+    // Check is id name contains invalid characters
+    if (!cstr_isvalid(id)) return 252;
+
     // Check that id name is not too long
     if (strlen(id) > MAXID) return 250;
 
@@ -4074,7 +4085,12 @@ int DLLEXPORT EN_setpatternid(EN_Project p, int index, char *id)
 
     if (!p->Openflag) return 102;
     if (index < 1 || index > p->network.Npats) return 205;
+
+    // Check is id name contains invalid characters
+    if (!cstr_isvalid(id)) return 252;
+
     if (strlen(id) > MAXID) return 250;
+
     for (i = 1; i <= p->network.Npats; i++)
     {
         if (i != index && strcmp(id, p->network.Pattern[i].ID) == 0) return 215;
@@ -4219,6 +4235,9 @@ int DLLEXPORT EN_addcurve(EN_Project p, char *id)
     if (!p->Openflag) return 102;
     if (EN_getcurveindex(p, id, &i) == 0) return 215;
 
+    // Check is id name contains invalid characters
+    if (!cstr_isvalid(id)) return 252;
+
     // Check that id name is not too long
     if (strlen(id) > MAXID) return 250;
 
@@ -4338,6 +4357,10 @@ int DLLEXPORT EN_setcurveid(EN_Project p, int index, char *id)
 
     if (!p->Openflag) return 102;
     if (index < 1 || index > p->network.Ncurves) return 205;
+
+    // Check is id name contains invalid characters
+    if (!cstr_isvalid(id)) return 252;
+
     if (strlen(id) > MAXID) return 250;
     for (i = 1; i <= p->network.Ncurves; i++)
     {

src/errors.dat

@@ -49,6 +49,7 @@ DAT(240,"nonexistent source")
 DAT(241,"nonexistent control")
 DAT(250,"invalid format")
 DAT(251,"invalid parameter code")
+DAT(252,"invalid ID name")
 DAT(253,"nonexistent demand category")
 DAT(254,"node with no coordinates")
 DAT(257,"nonexistent rule")

src/util/cstr_helper.c

@@ -17,32 +17,43 @@
 #include "cstr_helper.h"
 
 
-int copy_cstr(const char *source, char **dest)
-// Determines length, allocates memory, and returns a null terminated copy
-// Be Aware: caller is responsible for freeing memory
+int cstr_duplicate(char **dest, const char *source)
+// Duplicates source string
 {
-    size_t size;
-
-    size = 1 + strlen(source);
+    size_t size = 1 + strlen(source);
     *dest = (char *) calloc(size, sizeof(char));
 
     if (*dest == NULL)
         return -1;
     else {
 #ifdef _MSC_VER
-		strncpy_s(*dest, size, source, size);
+        strncpy_s(*dest, size, source, size);
 #else
-		strncpy(*dest, source, size);
+        strncpy(*dest, source, size);
 #endif
     }
     return 0;
 }
 
 
-bool isnullterm_cstr(const char *source)
+bool cstr_isvalid(const char *element_id)
+// Determines if invalid characters are present in an element id string
 {
-	if (strchr(source, '\0'))
-		return true;
-	else
-		return false;
+    const char *invalid_chars = " \";";
+
+    // if invalid char is present a pointer to it is returned else NULL
+    if (strpbrk(element_id, invalid_chars))
+        return false;
+    else
+        return true;
+}
+
+
+bool cstr_isnullterm(const char *source)
+// Determines if the string passed is null terminated or not
+{
+    if (strchr(source, '\0'))
+        return true;
+    else
+        return false;
 }

src/util/cstr_helper.h

@@ -18,8 +18,21 @@
 #include <stdbool.h>
 
 
-int copy_cstr(const char *source, char **destination);
-bool isnullterm_cstr(const char *source);
+#if defined(__cplusplus)
+extern "C" {
+#endif
+
+
+int cstr_duplicate(char **dest, const char *source);
+
+bool cstr_isvalid(const char *element_id);
+
+bool cstr_isnullterm(const char *source);
+
+
+#if defined(__cplusplus)
+}
+#endif
 
 
 #endif /* CSTR_HELPER_H_ */

src/util/filemanager.c

@@ -63,7 +63,7 @@ int get_filename(file_handle_t *file_handle, char **filename)
 // BE AWARE: The memory allocated here must be freed by the caller
 //
 {
-    return copy_cstr(file_handle->filename, filename);
+    return cstr_duplicate(filename, file_handle->filename);
 }
 
 
@@ -73,7 +73,7 @@ int open_file(file_handle_t *file_handle, const char *filename, const char *file
     if (filename == NULL)
         _get_temp_filename(&(file_handle->filename));
     else
-        copy_cstr(filename, &(file_handle->filename));
+        cstr_duplicate(&(file_handle->filename), filename);
 
     if (file_mode == NULL)
         error = -1;
@@ -156,7 +156,7 @@ int remove_file(file_handle_t *file_handle) {
 bool is_valid(file_handle_t *file_handle)
 {
 	if ((file_handle->filename == NULL && file_handle->file == NULL) ||
-		(isnullterm_cstr(file_handle->filename) && file_handle != NULL))
+		(cstr_isnullterm(file_handle->filename) && file_handle != NULL))
 		return true;
 	else
 		return false;
@@ -195,7 +195,7 @@ int _get_temp_filename(char **tempname)
         return error;
     }
     else
-        copy_cstr(name, tempname);
+        cstr_duplicate(tempname, name);
 
     // --- free the pointer returned by _tempnam
     if (name)
@@ -204,7 +204,7 @@ int _get_temp_filename(char **tempname)
     // --- for non-Windows systems:
 #else
     // --- use system function mkstemp() to create a temporary file name
-    copy_cstr("enXXXXXX", tempname);
+    cstr_duplicate(tempname, "enXXXXXX");
     error = mkstemp(*tempname);
 #endif
     return error;