From 75ff91d0e4e2df3b0ae1b651ac726399559e92f1 Mon Sep 17 00:00:00 2001
From: neon443 <69979447+neon443@users.noreply.github.com>
Date: Tue, 1 Jul 2025 13:26:02 +0100
Subject: [PATCH] YES asdkfghjaksdhfjgkjadhfg ASDFJASODJFSDJLFASDILFAH PRIVKEYS
 CAN BE IMPORTED!!!! added extractField to help with extracting the correct
 number of bytes (using the length prefix)

---
 ShhShell/Keys/KeyManager.swift | 21 +++++++++++++--------
 1 file changed, 13 insertions(+), 8 deletions(-)

diff --git a/ShhShell/Keys/KeyManager.swift b/ShhShell/Keys/KeyManager.swift
index 2fd2dd7..de9829f 100644
--- a/ShhShell/Keys/KeyManager.swift
+++ b/ShhShell/Keys/KeyManager.swift
@@ -108,8 +108,7 @@ class KeyManager: ObservableObject {
 		
 		removeField(&dataBlob) //remove key type for pubkey str
 		
-		let pubkeyData = Data((dataBlob as NSData)[0..<Int((dataBlob as NSData)[3])])
-		removeField(&dataBlob) //remove pubkey field
+		let pubkeyData = extractField(&dataBlob) //extract pubkey field
 		
 		dataBlob.removeFirst(4) //remove lenght header for the privkey data blob
 		
@@ -120,15 +119,13 @@ class KeyManager: ObservableObject {
 		removeField(&dataBlob) // remove pubkey type header
 		removeField(&dataBlob) // remove pubkey
 		
-		let privatekeyData = Data((dataBlob as NSData)[0..<32])
-		dataBlob.removeFirst(4 + 32) //remove privkey
-		
-		guard (dataBlob as NSData).subdata(with: NSRange(0...31)) == pubkeyData else {
+		var privatekeyData = extractField(&dataBlob) //extract privkey
+		guard Data((privatekeyData as NSData)[32..<64]) == pubkeyData else {
 			fatalError("pubkeys dont match")
 		}
-		dataBlob.removeFirst(32) //remove pubkey
+		privatekeyData.removeLast(32) //remove pubkey from privkey
 		
-		let comment = String(data: Data((dataBlob as NSData)[0..<Int((dataBlob as NSData)[3])]), encoding: .utf8)!
+		let comment = String(data: extractField(&dataBlob), encoding: .utf8)!
 		
 		return Keypair(type: .ecdsa, name: comment, publicKey: pubkeyData, privateKey: privatekeyData)
 	}
@@ -213,4 +210,12 @@ class KeyManager: ObservableObject {
 		data.removeFirst(4)
 		data.removeFirst(Int(length))
 	}
+	
+	static func extractField(_ data: inout Data) -> Data {
+		let nsdata = data as NSData
+		let lenght = Int(nsdata[3])
+		let extracted = Data(nsdata[4..<(lenght+4)])
+		data.removeFirst(4 + lenght)
+		return extracted
+	}
 }