Add ability to transmute girl to boy

Resolves issue #45

Author: Speykious

src/main.rs

@@ -8,6 +8,7 @@ use clap::{
 	Command,
 };
 use clap_complete::{generate, Generator, Shell};
+
 fn build_cli() -> Command {
 	Command::new(crate_name!())
 		.arg_required_else_help(true)
@@ -19,7 +20,12 @@ fn build_cli() -> Command {
 		.subcommand(
 			Command::new("bo").about("Run the buffer overflow exploit. Optionally take a shower"),
 		)
-		.subcommand(Command::new("transition").about("Safely transmute a Boy to a Girl"))
+		.subcommand(
+			Command::new("transition")
+				.about("Safely transmute A to B")
+				.subcommand(Command::new("boy-to-girl").about("Safely transmute a Boy to a Girl"))
+				.subcommand(Command::new("girl-to-boy").about("Safely transmute a Girl to a Boy")),
+		)
 		.subcommand(Command::new("segfault").about("Segfault yourself"))
 		.subcommand(
 			Command::new("completions")
@@ -34,6 +40,7 @@ fn build_cli() -> Command {
 		.styles(STYLE)
 		.help_template(TEMPLATE)
 }
+
 fn main() {
 	let mut command = build_cli();
 	let matches = build_cli().clone().get_matches();
@@ -42,7 +49,10 @@ fn main() {
 		"uaf" => cve_rs::use_after_free(),
 		"segfault" => cve_rs::segfault(),
 		"bo" => cve_rs::buffer_overflow().unwrap(),
-		"transition" => transmute_demo().unwrap(),
+		"transition" => {
+			let command = subcommand.1.subcommand().expect("Invalid transition").0;
+			transmute_demo(command).unwrap()
+		}
 		"completions" => print_completions(
 			subcommand.1.get_one::<Shell>("shell").copied().unwrap(),
 			&mut command,
@@ -96,14 +106,18 @@ struct Girl {
 	github_username: String,
 }
 
-fn transmute_demo() -> std::io::Result<()> {
+fn transmute_demo(command: &str) -> std::io::Result<()> {
 	use std::io::Write as _;
 
 	let stdin = std::io::stdin();
 	let mut stdout = std::io::stdout();
 	let mut input_buf = String::new();
 
-	stdout.write_all(b"Creating a Boy struct\n")?;
+	match command {
+		"boy-to-girl" => stdout.write_all(b"Creating a Boy struct\n")?,
+		"girl-to-boy" => stdout.write_all(b"Creating a Girl struct\n")?,
+		_ => unreachable!(),
+	}
 
 	let age = {
 		stdout.write_all(b"Enter age: ")?;
@@ -135,17 +149,33 @@ fn transmute_demo() -> std::io::Result<()> {
 		input_buf.trim().to_owned()
 	};
 
-	let boy: Boy = Boy {
-		age,
-		name,
-		github_username,
-	};
+	match command {
+		"boy-to-girl" => {
+			let boy: Boy = Boy {
+				age,
+				name,
+				github_username,
+			};
+			println!("Before transmute: {boy:?}");
+
+			let girl: Girl = cve_rs::transmute(boy);
+			println!("After transmute: {girl:?}");
+		}
 
-	println!("Before transmute: {boy:?}");
+		"girl-to-boy" => {
+			let girl: Girl = Girl {
+				age,
+				name,
+				github_username,
+			};
+			println!("Before transmute: {girl:?}");
 
-	let girl: Girl = cve_rs::transmute(boy);
+			let boy: Boy = cve_rs::transmute(girl);
+			println!("After transmute: {boy:?}");
+		}
 
-	println!("After transmute: {girl:?}");
+		_ => unreachable!(),
+	}
 
 	Ok(())
 }