Merge pull request #65 from luc1ph3r/Fix_skew_write_to_buffer

Guarantee a C-string termination
jiangwen365 · Jun 17, 2018 · 5859511 · 5859511
2 parents e204b88 + 677aadb
commit 5859511
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pypyodbc.py b/pypyodbc.py
@@ -1898,6 +1898,8 @@ def fetchone(self):
                                 if target_type == SQL_C_BINARY:
                                     value_list.append(buf_cvt_func(alloc_buffer.raw[:used_buf_len.value]))
                                 elif target_type == SQL_C_WCHAR:
+                                    if used_buf_len.value < total_buf_len:
+                                        ctypes.memset(ctypes.addressof(alloc_buffer) + used_buf_len.value, 0, 1)
                                     value_list.append(buf_cvt_func(from_buffer_u(alloc_buffer)))
                                 elif alloc_buffer.value == '':
                                     value_list.append(None)