handle memory allocation errors

chipitsine · Oct 19, 2024 · 77a621e · 77a621e
1 parent 9667dc1
commit 77a621e
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/ini/iniObjectInsert.c b/ini/iniObjectInsert.c
@@ -30,6 +30,8 @@ int iniObjectInsert( HINI hIni, char *pszObject )
 
 	/* CREATE OBJECT STRUCT */
 	hObject = malloc( sizeof(INIOBJECT) );
+        if ( !hObject )
+            return INI_ERROR;
 	hIni->hCurProperty			= NULL;
 	hObject->hFirstProperty		= NULL;
 	hObject->hLastProperty		= NULL;

diff --git a/ini/iniOpen.c b/ini/iniOpen.c
@@ -184,6 +184,8 @@ int iniOpen( HINI *hIni, char *pszFileName, char *cComment, char cLeftBracket, c
 
     /* INIT STATEMENT */
     *hIni = malloc( sizeof(INI) );
+    if ( !*hIni )
+        return INI_ERROR;
     if ( pszFileName && pszFileName != STDINFILE )
         strncpy((*hIni)->szFileName, pszFileName, ODBC_FILENAME_MAX );
     else if ( pszFileName == STDINFILE )
@@ -363,6 +365,8 @@ int iniOpen( HINI *hIni, char *pszFileName, char *cComment, char cLeftBracket, c
 
     /* INIT STATEMENT */
     *hIni = malloc( sizeof(INI) );
+    if ( !*hIni )
+        return INI_ERROR;
     if ( pszFileName && pszFileName != STDINFILE )
         strncpy((*hIni)->szFileName, pszFileName, ODBC_FILENAME_MAX );
     else if ( pszFileName == STDINFILE )

diff --git a/log/logOpen.c b/log/logOpen.c
@@ -41,6 +41,8 @@ int logOpen( HLOG *phLog, char *pszProgramName, char *pszLogFile, long nMaxMsgs
 
 	/* LOG STRUCT */
 	*phLog = malloc( sizeof(LOG) );
+        if ( !*phLog )
+            return LOG_ERROR;
     (*phLog)->nMaxMsgs			= nMaxMsgs;
     (*phLog)->hMessages			= lstOpen();
     (*phLog)->bOn				= 0;