gfwleak/zhangyang-libzt
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'py_send_binary3' of https://github.com/linsomniac/libzt into linsomniac-py_send_binary3

Browse Source
This commit is contained in:
Joseph Henry
2021-03-26 20:08:13 -07:00
parent e69956a230 5f6faaa5e8
commit 46eb8be174
1 changed files with 6 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
src/bindings/python/PythonSockets.cpp
View File

@@ -193,40 +193,16 @@ PyObject * zts_py_recv(int fd, int len, int flags)
int zts_py_send(int fd, PyObject *buf, int flags)
{
int bytes_sent = ZTS_ERR_OK;
char *bytes = NULL;
PyObject *encodedStr = NULL;
Py_buffer output;
int bytes_sent;
// Check for various encodings, or lack thereof
if (PyByteArray_Check(buf)) {
bytes = PyByteArray_AsString(buf);
if (PyObject_GetBuffer(buf, &output, PyBUF_SIMPLE) != 0) {
return 0;
}
if (PyUnicode_Check(buf)) {
encodedStr = PyUnicode_AsEncodedString(buf, "UTF-8", "strict");
if (!encodedStr) {
return ZTS_ERR_ARG;
}
bytes = PyBytes_AsString(encodedStr);
}
bytes_sent = zts_send(fd, output.buf, output.len, flags);
PyBuffer_Release(&output);
if (!bytes) {
// No encoding detected
bytes = PyBytes_AsString(buf);
}
// If we still don't have a valid pointer to a C-string, fail
if (!bytes) {
bytes_sent = ZTS_ERR_ARG;
}
else {
bytes_sent = zts_send(fd, bytes, strlen(bytes), flags);
}
if (encodedStr) {
Py_DECREF(encodedStr);
}
return bytes_sent;
}