Add PyBytes_FromStringAndSize to allow strings with NUL characters to be processed

bytes.go

@@ -35,6 +35,14 @@ func PyBytes_FromString(str string) *PyObject {
 	return togo(C.PyBytes_FromString(cstr))
 }
 
+//PyBytes_FromStringAndSize : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_FromStringAndSize
+func PyBytes_FromStringAndSize(str string) *PyObject {
+	cstr := C.CString(str)
+	defer C.free(unsafe.Pointer(cstr))
+
+	return togo(C.PyBytes_FromStringAndSize(cstr, C.Py_ssize_t(len(str))))
+}
+
 //PyBytes_FromObject : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_FromObject
 func PyBytes_FromObject(o *PyObject) *PyObject {
 	return togo(C.PyBytes_FromObject(toc(o)))

bytes_test.go

@@ -35,6 +35,17 @@ func TestBytesFromAsString(t *testing.T) {
 	assert.Equal(t, s1, PyBytes_AsString(bytes1))
 }
 
+func TestBytesFromStringAndSize(t *testing.T) {
+	Py_Initialize()
+
+	s1 := "aaaaaaaa\x00bbbb"
+
+	bytes1 := PyBytes_FromStringAndSize(s1)
+	defer bytes1.DecRef()
+
+	assert.Equal(t, s1, PyBytes_AsString(bytes1))
+}
+
 func TestBytesSize(t *testing.T) {
 	Py_Initialize()