pkey_ctx: dh params generation test

Author: huwcbjones

openssl/src/pkey_ctx.rs

@@ -994,6 +994,28 @@ mod test {
         ctx.keygen().unwrap();
     }
 
+    #[test]
+    #[cfg(not(boringssl))]
+    fn dh_paramgen() {
+        let mut ctx = PkeyCtx::new_id(Id::DH).unwrap();
+        ctx.paramgen_init().unwrap();
+        ctx.set_dh_paramgen_prime_len(512).unwrap();
+        ctx.set_dh_paramgen_generator(2).unwrap();
+        let params = ctx.paramgen().unwrap();
+
+        let size = {
+            cfg_if! {
+                if #[cfg(any(libressl, all(ossl101, not(ossl102))))] {
+                    // LibreSSL and OpenSSL 1.0.1 and earlier
+                    48
+                } else {
+                    64
+                }
+            }
+        };
+        assert_eq!(params.size(), size);
+    }
+
     #[test]
     #[cfg(not(boringssl))]
     fn dsa_paramgen() {