Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

How to use generate_key_pairs method in lisa

Best Python code snippet using lisa_python

rsa.py

Source:rsa.py

...45        gcd, x, y = self.extended_euclid_gcd(A, M)46        while x < 0 :47            x += M48        return x49    def generate_key_pairs(self, bits: int = 16) :50        """51        Generate two key pairs: public key (e, n) and private key (d, n)52        """53        if bits < 4 :54            raise Exception("The number of bits must be greater than 4")55        self.p = prime.generate_prime_number(bits)56        self.q = prime.generate_prime_number(bits)57        self.n = self.p * self.q58        self.phi_n = (self.p-1) * (self.q-1)59        while not self.is_coprime(self.e, self.phi_n) :60            self.e = prime.generate_n_bits_number(bits)61        self.d = self.modulo_multiplicative_inverse(self.e, self.phi_n)62        self.public_key = (self.e, self.n)63        self.private_key = (self.d, self.n)64    def save_public_key(self, filename: str) :65        """66        Save public key as *.pub file67        """68        public_key = self.get_public_key()69        with open(os.path.join(RSA.datadir, filename, (filename + '.pub')), 'w') as f_pub :70            f_pub.write(str(public_key[0]) + '\n')71            f_pub.write(str(public_key[1]))72    def save_private_key(self, filename: str) :73        """74        Save private key as *.pri file75        """76        private_key = self.get_private_key()77        with open(os.path.join(RSA.datadir, filename, (filename + '.pri')), 'w') as f_pri :78            f_pri.write(str(private_key[0]) + '\n')79            f_pri.write(str(private_key[1]))80    def save_key_pairs(self, filename: str) :81        """82        Save both key pairs: public and private key83        """84        if not os.path.exists(os.path.join(RSA.datadir, filename)) :85            os.makedirs(os.path.join(RSA.datadir, filename))86        self.save_public_key(filename)87        self.save_private_key(filename)88    def get_public_key(self) :89        """90        Returns the current public key91        """92        return self.public_key93    def get_private_key(self) :94        """95        Returns the current private key96        """97        return self.private_key98    def encrypt(self, plaintext: bytearray) -> str:99        """100        Encrypt plaintext by public key pair using RSA algorithm101        """102        if len(plaintext) == 0 :103            raise Exception("Plaintext cannot be empty")104        if not isinstance(plaintext, (bytes, bytearray)):105            raise Exception("Plaintext must be instance of bytes or bytearray")106        e, n = self.get_public_key()107        ciphertext = ''108        for i in range(len(plaintext)) :109            c = pow(plaintext[i], e, n)110            ciphertext += r'\{}'.format(hex(c)[1:])111        return ciphertext112    def decrypt(self, ciphertext: str) :113        """114        Decrypt ciphertext by private key pair using RSA algorithm115        """116        if len(ciphertext) == 0 :117            raise Exception("Ciphertext cannot be empty")118        d, n = self.get_private_key()119        ciphertext_list = ciphertext.split("\\x")120        plaintext = []121        for i in range(len(ciphertext_list)) :122            try :123                p = pow(int(ciphertext_list[i], 16), d, n)124                plaintext.append(p)125            except ValueError :126                continue127        return plaintext128# rsa = RSA()129# # rsa.generate_key_pairs(16)130# rsa.generate_key_pairs(64)131# # rsa.generate_key_pairs(256)132# # rsa.generate_key_pairs(1024)133# plaintext = "Kriptografi dan Koding"#.encode("utf-8")134# encrypted = rsa.encrypt(plaintext)135# print(encrypted)136# decrypted = rsa.decrypt(encrypted)137# print(decrypted)...

keys_generator.py

Source:keys_generator.py

...6        self.type = type7        self.private_key = None8        self.public_key = None9        self.path = path10    def generate_key_pairs(self):11        self.private_key = rsa.generate_private_key(12            public_exponent=65537,13            key_size=4096,14            backend=default_backend()15        )16        self.public_key = self.private_key.public_key()17        if self.type == 1:18            self.save_pem_keys()19        elif self.type == 2:20            self.save_der_keys()21    def save_pem_keys(self):22        encrypted_privatekey_pem = self.private_key.private_bytes(23            encoding=serialization.Encoding.PEM,24            format=serialization.PrivateFormat.PKCS8,25            encryption_algorithm=serialization.BestAvailableEncryption(password=b'CY6740')26        )27        publickey_pem = self.public_key.public_bytes(28            encoding=serialization.Encoding.PEM,29            format=serialization.PublicFormat.SubjectPublicKeyInfo30        )31        with open(self.path + "privatekey.pem", "w") as prikey_file:32            prikey_file.write(encrypted_privatekey_pem.decode())33        with open(self.path + "publickey-pem.pub", "w") as pubkey_file:34            pubkey_file.write(publickey_pem.decode())35    def save_der_keys(self):36        encrypted_privatekey_der = self.private_key.private_bytes(37            encoding=serialization.Encoding.DER,38            format=serialization.PrivateFormat.PKCS8,39            encryption_algorithm=serialization.BestAvailableEncryption(password=b'CY6740')40        )41        publickey_der = self.public_key.public_bytes(42            encoding=serialization.Encoding.DER,43            format=serialization.PublicFormat.SubjectPublicKeyInfo44        )45        with open(self.path + "privatekey.der", "w") as prikey_file:46            prikey_file.write(encrypted_privatekey_der.decode())47        with open(self.path + "publickey-der.pub", "w") as pubkey_file:48            pubkey_file.write(publickey_der.decode())49if __name__ == "__main__":50    type_of_keys = 2  # Type 1 is for PEM and 2 for DER51    if type_of_keys == 1:52        sender_keys_gen = KeysGenerator(path="/home/abhiram/Desktop/CY6740/ProblemSet-2/target/senderkeys/", type=type_of_keys)53        sender_keys_gen.generate_key_pairs()54        receiver_keys_gen = KeysGenerator(path="/home/abhiram/Desktop/CY6740/ProblemSet-2/target/receiverkeys/", type=type_of_keys)55        receiver_keys_gen.generate_key_pairs()56        attacker_keys_gen = KeysGenerator(path="/home/abhiram/Desktop/CY6740/ProblemSet-2/target/attackerkeys/", type=type_of_keys)57        attacker_keys_gen.generate_key_pairs()58    elif type_of_keys == 2:59        sender_keys_gen = KeysGenerator(path="/home/abhiram/Desktop/CY6740/ProblemSet-2/target/senderkeys/", type=type_of_keys)60        sender_keys_gen.generate_key_pairs()61        receiver_keys_gen = KeysGenerator(path="/home/abhiram/Desktop/CY6740/ProblemSet-2/target/receiverkeys/", type=type_of_keys)62        receiver_keys_gen.generate_key_pairs()63        attacker_keys_gen = KeysGenerator(path="/home/abhiram/Desktop/CY6740/ProblemSet-2/target/attackerkeys/", type=type_of_keys)...

ecc.py

Source:ecc.py

1import sys2from Crypto.PublicKey import ECC3def generate_key_pairs(id):4    private_key = ECC.generate(curve='P-256')5    public_key = private_key.public_key()6    with open(f'{id}_key.pub', 'w') as f:7        f.write(public_key.export_key(format='PEM'))8    with open(f'{id}_key.pri', 'w') as f:9        f.write(private_key.export_key(format='PEM'))10def main():11    generate_key_pairs(sys.argv[1])12if __name__ == '__main__':...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.