a
�tb � @ s� d dl Z d dlZd dlZd dlmZmZ d dlZd dlZd dlm Z ej
� ej
� e
�d�Ze�ee��Zed Zed Ze�e�ed��Zedks�J �dadZd d
� Zed �d d
�Zee e f Z e d �dd�Z dee e d�dd�Z ed �dd�Z dS )� N)�Optional�Tuple)�inv_mod_curve_sizezpedersen_params.json�EC_ORDER� FIELD_PRIME� �� c C sd t j�tj�| ��at jt jt jgtj_ t jt jt jt jgtj
_ t tj
_ t jt jt jt jgtj
_ d S �N)�ctypes�cdll� LoadLibrary�os�path�abspath� CPP_LIB_PATH�c_void_p�Hash�argtypes�Verify�bool�restype�Sign)Zcrypto_c_exports_path� r ��/Users/okudajunta/Library/Mobile Documents/com~apple~CloudDocs/solidity/dydx-v3-rust/./src/stark/starkex/starkex_resources/cpp_signature.py� get_cpp_lib s ���r )�returnc C s t d uS r )r r r r r �check_cpp_lib_path& s r c C s^ t �t�}t�| jdddd�|jdddd�|�dkrDt|j�d���t j
|jd d� ddd�S )N� �littleF��signedr � ) r
�create_string_buffer�OUT_BUFFER_SIZEr r �to_bytes�
ValueError�raw�rstrip�int�
from_bytes)�left�right�resr r r �cpp_hash4 s
��r- r )�seedr c C s� t �t�}t�|�}t�|jdddd�| jdddd�||�dkrPt|j �
d���t j |j dd� ddd�}t
|�}t j |j dd� ddd�|fS ) z�
Note that this uses the secrets module to generate cryptographically strong random numbers.
Note that the same seed will give a different signature compared with the sign function in
signature.py.
r r Fr r r! �@ N)r
r" r# �secrets� token_bytesr r r$ r% r& r' r( r) r )�msg_hash�priv_keyr. r, Z random_bytes�w�sr r r �cpp_sign> s
��r6 c
C s� t |�}d| kr dt k s.n J d| ��d| krFdt k sTn J d| ��d| krldt k szn J d| ��d| kr�tk s�n J d| ��t�|jddd d
�| jddd d
�|jddd d
�|jddd d
��S ) N� r zstark_key = %sz
msg_hash = %szr = %szw = %sr r Fr )r �N_ELEMENT_BITS_ECDSAr r r r$ )r2 �rr5 � stark_keyr4 r r r �
cpp_verifyP s &&&"�r; )r )!r
r0 r
�typingr r �json�math�*starkex.starkex_resources.python_signaturer r �join�dirname�__file__� PEDERSEN_HASH_POINT_FILENAME�load�open�PEDERSEN_PARAMSr r �floor�logr8 r r# r r r r( � ECSignaturer- r6 r; r r r r �<module> s, �