mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-23 08:11:00 +00:00
42 lines
830 B
Python
42 lines
830 B
Python
from __future__ import (absolute_import, print_function, division)
|
|
from cffi import FFI
|
|
import OpenSSL
|
|
|
|
xffi = 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
|