add pkcs5_v15_password

1 files changed, 32 insertions, 0 deletions

diff --git a/ossl_cipher.c b/ossl_cipher.c
index 3d63a2a..50ae909 100644
--- a/ossl_cipher.c
+++ b/ossl_cipher.c
@@ -212,6 +212,36 @@ ossl_cipher_decrypt(int argc, VALUE *argv, VALUE self)
     return self;
 }
 
+static VALUE
+ossl_cipher_pkcs5_v15_password(int argc, VALUE *argv, VALUE self)
+{
+    EVP_CIPHER_CTX *ctx;
+    unsigned char iv[EVP_MAX_IV_LENGTH], key[EVP_MAX_KEY_LENGTH];
+    VALUE digest, rounds, pass, salt;
+    const EVP_MD *md;
+
+    GetCipher(self, ctx);
+    rb_scan_args(argc, argv, "31", &digest, &rounds, &pass, &salt);
+
+    md = EVP_get_digestbyname(STR2CSTR(digest));
+    if (md == NULL)
+        ossl_raise(eCipherError, "invalid digest");
+      
+    StringValue(pass);
+
+    if (!NIL_P(salt)) {
+        StringValue(salt);
+        if (RSTRING(salt)->len != 8)
+            ossl_raise(eCipherError, "salt must be 8 bytes");
+    }
+
+    EVP_BytesToKey(EVP_CIPHER_CTX_cipher(ctx), md, NIL_P(salt) ? NULL : RSTRING(salt)->ptr, RSTRING(pass)->ptr, RSTRING(pass)->len, NUM2INT(rounds), key, iv);
+    if (EVP_CipherInit(ctx, NULL, key, iv, -1) != 1)
+        ossl_raise(eCipherError, "");
+
+    return self;
+}
+
 static VALUE 
 ossl_cipher_update(VALUE self, VALUE data)
 {
@@ -360,6 +390,8 @@ Init_ossl_cipher(void)
 
     rb_define_method(cCipher, "name", ossl_cipher_name, 0);
 
+    rb_define_method(cCipher, "pkcs5_v15_password", ossl_cipher_pkcs5_v15_password, -1);
+
     rb_define_method(cCipher, "key=", ossl_cipher_set_key, 1);
     rb_define_method(cCipher, "key_len", ossl_cipher_key_length, 0);
 /*