On-the-fly certificate forgery

src/main.rs

@@ -106,7 +106,14 @@ struct OptPrint {
 #[cfg(feature = "record")]
 #[derive(FromArgs)]
 #[argp(subcommand, name = "record")]
-struct OptRecord {}
+struct OptRecord {
+	/// Path to the CAs directory
+	#[argp(positional)]
+	ca_dir: String,
+	/// Path to the directory to store certificates
+	#[argp(positional)]
+	certs_dir: String,
+}
 
 /// Record traffic
 #[derive(FromArgs)]
@@ -179,7 +186,13 @@ async fn main() {
 			record::print_records(&records, subopt.number);
 		}
 		#[cfg(feature = "record")]
-		Subcommand::Record(_subopt) => {
+		Subcommand::Record(subopt) => {
+			record::CERTIFICATES
+				.set(record::LyingCertificateRegistry::new(
+					&subopt.ca_dir,
+					&subopt.certs_dir,
+				))
+				.unwrap();
 			record::make_record(&opt.record_file);
 		}
 		Subcommand::Remove(subopt) => {