diff options
| author | Serhiy Storchaka <storchaka@gmail.com> | 2015-02-16 11:33:32 (GMT) |
|---|---|---|
| committer | Serhiy Storchaka <storchaka@gmail.com> | 2015-02-16 11:33:32 (GMT) |
| commit | 4d0d9829851915e97ae392dd803976be6c95c8d1 (patch) | |
| tree | e93666c54592b95dbca422ec66d0896f827957b3 /Modules/socketmodule.c | |
| parent | 53fa8b2a4bbb589d3d761284c70f93e0f852df23 (diff) | |
| parent | 1a1ff29659f068659dea07f1bd67b8fd4331071c (diff) | |
| download | cpython-4d0d9829851915e97ae392dd803976be6c95c8d1.zip cpython-4d0d9829851915e97ae392dd803976be6c95c8d1.tar.gz cpython-4d0d9829851915e97ae392dd803976be6c95c8d1.tar.bz2 | |
Issue #23446: Use PyMem_New instead of PyMem_Malloc to avoid possible integer
overflows. Added few missed PyErr_NoMemory().
Diffstat (limited to 'Modules/socketmodule.c')
| -rw-r--r-- | Modules/socketmodule.c | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/Modules/socketmodule.c b/Modules/socketmodule.c index 5913e65..7c7cb7d 100644 --- a/Modules/socketmodule.c +++ b/Modules/socketmodule.c @@ -4213,9 +4213,11 @@ socket_gethostname(PyObject *self, PyObject *unused) /* MSDN says ERROR_MORE_DATA may occur because DNS allows longer names */ - name = PyMem_Malloc(size * sizeof(wchar_t)); - if (!name) + name = PyMem_New(wchar_t, size); + if (!name) { + PyErr_NoMemory(); return NULL; + } if (!GetComputerNameExW(ComputerNamePhysicalDnsHostname, name, &size)) |