Add end2end test

.cargo/config.toml

@@ -0,0 +1,5 @@
+[target.aarch64-apple-darwin]
+rustflags = [
+  "-C", "link-arg=-undefined",
+  "-C", "link-arg=dynamic_lookup",
+]

.gitignore

@@ -1 +1,2 @@
 /target
+.venv

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "end2end/PPPwn"]
+	path = end2end/PPPwn
+	url = https://github.com/fedebuonco/PPPwn.git

Cargo.toml

@@ -1,9 +1,21 @@
 [package]
-name = "yappwn"
+name = "yapppwn"
 version = "0.1.0"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
 pcap = "2.0.0"
+pyo3 = { version = "0.21.2", features = ["extension-module"] }
+
+
+[lib]
+name = "py_yapppwn"
+# "cdylib" is necessary to produce a shared library for Python to import from.
+crate-type = ["cdylib"]
+path = "src/py_yapppwn.rs"
+
+[[bin]]
+name = "yapppwn"
+path = "src/main.rs"

src/exploit.rs

@@ -102,7 +102,7 @@ fn extract_ps4_source_mac(packet: &Packet) -> Result<MacAddress64, Box<dyn std::
 
     Ok(MacAddress64(mac_address))
 }
-fn build_fake_ifnet(pppoe_softc: u64) -> Vec<u8> {
+pub fn build_fake_ifnet(pppoe_softc: u64) -> Vec<u8> {
     // Fake ifnet
     let mut fake_ifnet = vec![0x41; 0x48]; // Fill with 'A' initially
     fake_ifnet.extend(&constants::ZERO.to_le_bytes()); // if_addrhead

src/py_yapppwn.rs

@@ -0,0 +1,20 @@
+mod constants;
+mod exploit;
+mod macaddress;
+use crate::exploit::build_fake_ifnet;
+use macaddress::MacAddress64;
+use pcap::Device;
+
+use pyo3::prelude::*;
+use pyo3::wrap_pyfunction;
+
+#[pyfunction]
+fn py_build_fake_ifnet(pppoe_softc: u64) -> PyResult<Vec<u8>> {
+    Ok(build_fake_ifnet(pppoe_softc))
+}
+
+#[pymodule]
+fn py_yapppwn(_py: Python, m: &PyModule) -> PyResult<()> {
+    m.add_function(wrap_pyfunction!(py_build_fake_ifnet, m)?)?;
+    Ok(())
+}

src/test.py

@@ -0,0 +1,4 @@
+import py_yapppwn
+
+result = py_yapppwn.py_build_fake_ifnet(1)
+print(result)