1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/* Copyright (c) 1999 Ng Pheng Siong. All rights reserved. */
/* $Id: _rc4.i 522 2007-05-08 22:21:51Z heikki $ */
%{
#include <openssl/rc4.h>
%}
%apply Pointer NONNULL { RC4_KEY * };
%inline %{
RC4_KEY *rc4_new(void) {
RC4_KEY *key;
if (!(key = (RC4_KEY *)PyMem_Malloc(sizeof(RC4_KEY))))
PyErr_SetString(PyExc_MemoryError, "rc4_new");
return key;
}
void rc4_free(RC4_KEY *key) {
PyMem_Free((void *)key);
}
PyObject *rc4_set_key(RC4_KEY *key, PyObject *value) {
const void *vbuf;
int vlen;
if (m2_PyObject_AsReadBufferInt(value, &vbuf, &vlen) == -1)
return NULL;
RC4_set_key(key, vlen, vbuf);
Py_INCREF(Py_None);
return Py_None;
}
PyObject *rc4_update(RC4_KEY *key, PyObject *in) {
PyObject *ret;
const void *buf;
Py_ssize_t len;
void *out;
if (PyObject_AsReadBuffer(in, &buf, &len) == -1)
return NULL;
if (!(out = PyMem_Malloc(len))) {
PyErr_SetString(PyExc_MemoryError, "expected a string object");
return NULL;
}
RC4(key, len, buf, out);
ret = PyString_FromStringAndSize(out, len);
PyMem_Free(out);
return ret;
}
int rc4_type_check(RC4_KEY *key) {
return 1;
}
%}
|
