mirrors/kitty
1
0
Fork 0
mirror of https://github.com/kovidgoyal/kitty synced 2026-07-23 16:58:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

Implement signature generation

Browse Source
This commit is contained in:
Kovid Goyal
2021-09-17 17:42:25 +05:30
parent fbc8a1cdcb
commit a941b1af4e
3 changed files with 109 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

74
kittens/transfer/rsync.c
View File

@@ -8,12 +8,84 @@
#include "data-types.h"
#include <librsync.h>
#define JOB_CAPSULE "rs_job_t"
#define IO_BUFFER_SIZE (64u * 1024u)
static void
free_job_capsule(PyObject *capsule) {
rs_job_t *job = PyCapsule_GetPointer(capsule, JOB_CAPSULE);
if (job) rs_job_free(job);
}
static PyObject*
begin_signature(PyObject *self UNUSED, PyObject *args) {
long long file_size = -1;
long sl = 0;
if (!PyArg_ParseTuple(args, "|Ll", &file_size, &sl)) return NULL;
rs_magic_number magic_number = 0;
size_t block_len = 0, strong_len = sl;
rs_result res = rs_sig_args(file_size, &magic_number, &block_len, &strong_len);
if (res != RS_DONE) {
PyErr_SetString(PyExc_ValueError, rs_strerror(res));
return NULL;
}
rs_job_t *job = rs_sig_begin(block_len, strong_len, magic_number);
if (!job) return PyErr_NoMemory();
PyObject *ans = PyCapsule_New(job, JOB_CAPSULE, free_job_capsule);
if (!ans) rs_job_free(job);
return ans;
}
#define GET_JOB_FROM_CAPSULE \
rs_job_t *job = PyCapsule_GetPointer(job_capsule, JOB_CAPSULE); \
if (!job) { PyErr_SetString(PyExc_TypeError, "Not a job capsule"); return NULL; }
static PyObject*
iter_job(PyObject *self UNUSED, PyObject *args) {
PyObject *job_capsule;
Py_ssize_t input_data_size;
char *input_data;
int eof = -1;
if (!PyArg_ParseTuple(args, "O!y#|p", &PyCapsule_Type, &job_capsule, &input_data, &input_data_size, &eof)) return NULL;
GET_JOB_FROM_CAPSULE;
if (eof == -1) eof = input_data_size > 0 ? 0 : 1;
rs_buffers_t buffer = {.avail_in=input_data_size, .next_in = input_data, .eof_in=eof, .avail_out=MAX(IO_BUFFER_SIZE, 2 * input_data_size)};
PyObject *ans = PyBytes_FromStringAndSize(NULL, buffer.avail_out);
if (!ans) return NULL;
buffer.next_out = PyBytes_AS_STRING(ans);
size_t output_size = 0;
rs_result result = RS_DONE;
while (true) {
size_t before = buffer.avail_out;
result = rs_job_iter(job, &buffer);
output_size += before - buffer.avail_out;
if (result == RS_DONE) break;
if (buffer.avail_in) {
if (_PyBytes_Resize(&ans, PyBytes_GET_SIZE(ans) * 2) != 0) return NULL;
buffer.avail_out = PyBytes_GET_SIZE(ans) - output_size;
buffer.next_out = PyBytes_AS_STRING(ans) + output_size;
continue;
}
if (result == RS_BLOCKED) break;
Py_DECREF(ans);
PyErr_SetString(PyExc_RuntimeError, rs_strerror(result));
return NULL;
}
if ((ssize_t)output_size != PyBytes_GET_SIZE(ans)) {
if (_PyBytes_Resize(&ans, output_size) != 0) return NULL;
}
return Py_BuildValue("NO", ans, result == RS_DONE ? Py_True : Py_False);
}
static PyMethodDef module_methods[] = {
{"begin_signature", (PyCFunction)begin_signature, METH_VARARGS, ""},
{"iter_job", (PyCFunction)iter_job, METH_VARARGS, ""},
{NULL, NULL, 0, NULL} /* Sentinel */
};
static int
exec_module(PyObject *m UNUSED) {
exec_module(PyObject *m) {
PyModule_AddIntMacro(m, IO_BUFFER_SIZE);
return 0;
}