mitmproxy/netlib/certffi.py

37 lines
757 B
Python

import cffi
import OpenSSL
xffi = cffi.FFI()
xffi.cdef ("""
struct rsa_meth_st {
int flags;
...;
};
struct rsa_st {
int pad;
long version;
struct rsa_meth_st *meth;
...;
};
""")
xffi.verify(
"""#include <openssl/rsa.h>""",
extra_compile_args=['-w']
)
def handle(privkey):
new = xffi.new("struct rsa_st*")
newbuf = xffi.buffer(new)
rsa = OpenSSL.SSL._lib.EVP_PKEY_get1_RSA(privkey._pkey)
oldbuf = OpenSSL.SSL._ffi.buffer(rsa)
newbuf[:] = oldbuf[:]
return new
def set_flags(privkey, val):
hdl = handle(privkey)
hdl.meth.flags = val
return privkey
def get_flags(privkey):
hdl = handle(privkey)
return hdl.meth.flags