psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2025-02-12 07:10:33 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix off-by-one error in PyMem_Malloc() call in

Browse Source 
psyco_conn_set_client_encoding().  Fixes #211
This commit is contained in:
James Henstridge 2008-02-13 16:01:10 +00:00
parent 6c2e3ab49b
commit b5f4a5f6c2
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
psycopg/connection_type.c
View File

@ -195,12 +195,12 @@ psyco_conn_set_client_encoding(connectionObject *self, PyObject *args)
if (!PyArg_ParseTuple(args, "s", &enc)) return NULL; if (!PyArg_ParseTuple(args, "s", &enc)) return NULL;
/* convert to upper case and remove '-' and '_' from string */ /* convert to upper case and remove '-' and '_' from string */
buffer = PyMem_Malloc(strlen(enc)); buffer = PyMem_Malloc(strlen(enc)+1);
for (i=j=0 ; i < strlen(enc) ; i++) { for (i=j=0 ; i < strlen(enc) ; i++) {
if (enc[i] == '_' || enc[i] == '-') if (enc[i] == '_' || enc[i] == '-')
continue; continue;
else else
buffer[j++] = toupper(enc[i]); buffer[j++] = toupper(enc[i]);
} }
buffer[j] = '\0'; buffer[j] = '\0';