diff options
Diffstat (limited to 'src/main/java/net/elytrium/limboauth')
5 files changed, 288 insertions, 38 deletions
diff --git a/src/main/java/net/elytrium/limboauth/LimboAuth.java b/src/main/java/net/elytrium/limboauth/LimboAuth.java index a42322a..46b0864 100644 --- a/src/main/java/net/elytrium/limboauth/LimboAuth.java +++ b/src/main/java/net/elytrium/limboauth/LimboAuth.java @@ -26,8 +26,8 @@ import com.j256.ormlite.dao.DaoManager; import com.j256.ormlite.dao.GenericRawResults; import com.j256.ormlite.db.DatabaseType; import com.j256.ormlite.field.FieldType; -import com.j256.ormlite.jdbc.JdbcPooledConnectionSource; import com.j256.ormlite.stmt.QueryBuilder; +import com.j256.ormlite.support.ConnectionSource; import com.j256.ormlite.table.TableInfo; import com.j256.ormlite.table.TableUtils; import com.velocitypowered.api.command.CommandManager; @@ -90,6 +90,7 @@ import net.elytrium.limboauth.command.LimboAuthCommand; import net.elytrium.limboauth.command.PremiumCommand; import net.elytrium.limboauth.command.TotpCommand; import net.elytrium.limboauth.command.UnregisterCommand; +import net.elytrium.limboauth.dependencies.DatabaseLibrary; import net.elytrium.limboauth.event.AuthPluginReloadEvent; import net.elytrium.limboauth.event.PreAuthorizationEvent; import net.elytrium.limboauth.event.PreEvent; @@ -158,7 +159,7 @@ public class LimboAuth { private ScheduledTask purgeCacheTask; private ScheduledTask purgePremiumCacheTask; - private JdbcPooledConnectionSource connectionSource; + private ConnectionSource connectionSource; private Dao<RegisteredPlayer, String> playerDao; private Pattern nicknameValidationPattern; private Limbo authServer; @@ -267,29 +268,14 @@ public class LimboAuth { this.cachedAuthChecks.clear(); Settings.DATABASE dbConfig = Settings.IMP.DATABASE; - switch (dbConfig.STORAGE_TYPE.toLowerCase(Locale.ROOT)) { - case "h2": { - this.connectionSource = new JdbcPooledConnectionSource("jdbc:h2:" + this.dataDirectoryFile.getAbsoluteFile() + "/limboauth"); - break; - } - case "mysql": { - this.connectionSource = new JdbcPooledConnectionSource( - "jdbc:mysql://" + dbConfig.HOSTNAME + "/" + dbConfig.DATABASE + dbConfig.CONNECTION_PARAMETERS, dbConfig.USER, dbConfig.PASSWORD - ); - break; - } - case "postgresql": { - this.connectionSource = new JdbcPooledConnectionSource( - "jdbc:postgresql://" + dbConfig.HOSTNAME + "/" + dbConfig.DATABASE + dbConfig.CONNECTION_PARAMETERS, dbConfig.USER, dbConfig.PASSWORD - ); - break; - } - default: { - LOGGER.error("Wrong database type."); - this.server.shutdown(); - return; - } - } + DatabaseLibrary databaseLibrary = DatabaseLibrary.valueOf(dbConfig.STORAGE_TYPE.toUpperCase(Locale.ROOT)); + this.connectionSource = databaseLibrary.connectToORM( + this.dataDirectoryFile.toPath().toAbsolutePath(), + dbConfig.HOSTNAME, + dbConfig.DATABASE + dbConfig.CONNECTION_PARAMETERS, + dbConfig.USER, + dbConfig.PASSWORD + ); this.nicknameValidationPattern = Pattern.compile(Settings.IMP.MAIN.ALLOWED_NICKNAME_REGEX); @@ -789,7 +775,7 @@ public class LimboAuth { return this.server; } - public JdbcPooledConnectionSource getConnectionSource() { + public ConnectionSource getConnectionSource() { return this.connectionSource; } @@ -797,17 +783,6 @@ public class LimboAuth { return this.playerDao; } - static { - // requireNonNull prevents the shade plugin from excluding the drivers in minimized jar. - Objects.requireNonNull(com.mysql.cj.jdbc.Driver.class); - Objects.requireNonNull(com.mysql.cj.conf.url.SingleConnectionUrl.class); - - Objects.requireNonNull(org.h2.Driver.class); - Objects.requireNonNull(org.h2.engine.Engine.class); - - Objects.requireNonNull(org.postgresql.Driver.class); - } - private static void setLogger(Logger logger) { LOGGER = logger; } diff --git a/src/main/java/net/elytrium/limboauth/Settings.java b/src/main/java/net/elytrium/limboauth/Settings.java index f8733ad..e258da7 100644 --- a/src/main/java/net/elytrium/limboauth/Settings.java +++ b/src/main/java/net/elytrium/limboauth/Settings.java @@ -375,7 +375,7 @@ public class Settings extends YamlConfig { @Comment("Database settings") public static class DATABASE { - @Comment("Database type: mysql, postgresql or h2.") + @Comment("Database type: mysql, postgresql, sqlite or h2.") public String STORAGE_TYPE = "h2"; @Comment("Settings for Network-based database (like MySQL, PostgreSQL): ") diff --git a/src/main/java/net/elytrium/limboauth/dependencies/BaseLibrary.java b/src/main/java/net/elytrium/limboauth/dependencies/BaseLibrary.java new file mode 100644 index 0000000..e8c34b5 --- /dev/null +++ b/src/main/java/net/elytrium/limboauth/dependencies/BaseLibrary.java @@ -0,0 +1,94 @@ +/* + * Copyright (C) 2021 - 2022 Elytrium + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + */ + +package net.elytrium.limboauth.dependencies; + +import edu.umd.cs.findbugs.annotations.SuppressFBWarnings; +import java.io.InputStream; +import java.net.MalformedURLException; +import java.net.URL; +import java.nio.file.Files; +import java.nio.file.Path; +import java.nio.file.StandardCopyOption; + +public enum BaseLibrary { + H2_V1( + "com.h2database", + "h2", + "1.4.200" + ), + H2_V2( + "com.h2database", + "h2", + "2.1.214" + ), + MYSQL( + "com.mysql", + "mysql-connector-j", + "8.0.31" + ), + POSTGRESQL( + "org.postgresql", + "postgresql", + "42.5.1" + ), + SQLITE( + "org.xerial", + "sqlite-jdbc", + "3.40.0.0" + ); + + private final Path filenamePath; + private final URL mavenRepoURL; + + BaseLibrary(String groupId, String artifactId, String version) { + String mavenPath = String.format("%s/%s/%s/%s-%s.jar", + groupId.replace(".", "/"), + artifactId, + version, + artifactId, + version + ); + + this.filenamePath = Path.of("libraries/" + mavenPath); + + URL mavenRepoURL = null; + try { + mavenRepoURL = new URL("https://repo1.maven.org/maven2/" + mavenPath); + } catch (Exception e) { + e.printStackTrace(); + } + + this.mavenRepoURL = mavenRepoURL; + } + + @SuppressFBWarnings("NP_NULL_ON_SOME_PATH_FROM_RETURN_VALUE") + public URL getClassLoaderURL() throws MalformedURLException { + if (!Files.exists(this.filenamePath)) { + try { + try (InputStream in = this.mavenRepoURL.openStream()) { + Files.createDirectories(this.filenamePath.getParent()); + Files.copy(in, Files.createFile(this.filenamePath), StandardCopyOption.REPLACE_EXISTING); + } + } catch (Exception e) { + e.printStackTrace(); + } + } + + return this.filenamePath.toUri().toURL(); + } +} diff --git a/src/main/java/net/elytrium/limboauth/dependencies/DatabaseLibrary.java b/src/main/java/net/elytrium/limboauth/dependencies/DatabaseLibrary.java new file mode 100644 index 0000000..3a8a070 --- /dev/null +++ b/src/main/java/net/elytrium/limboauth/dependencies/DatabaseLibrary.java @@ -0,0 +1,149 @@ +/* + * Copyright (C) 2021 - 2022 Elytrium + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + */ + +package net.elytrium.limboauth.dependencies; + +import com.j256.ormlite.jdbc.JdbcSingleConnectionSource; +import com.j256.ormlite.support.ConnectionSource; +import java.lang.reflect.Constructor; +import java.lang.reflect.Method; +import java.net.URL; +import java.nio.file.Files; +import java.nio.file.Path; +import java.sql.Connection; +import java.sql.Driver; +import java.sql.DriverManager; +import java.sql.Statement; +import java.util.Properties; + +public enum DatabaseLibrary { + H2_LEGACY_V1( + BaseLibrary.H2_V1, + (classLoader, dir, jdbc, user, password) -> fromDriver(classLoader.loadClass("org.h2.Driver"), jdbc, null, null, false), + (dir, hostname, database) -> "jdbc:h2:" + dir + "/limboauth" + ), + H2( + BaseLibrary.H2_V2, + (classLoader, dir, jdbc, user, password) -> { + Connection modernConnection = fromDriver(classLoader.loadClass("org.h2.Driver"), jdbc, null, null, true); + + Path legacyDatabase = dir.resolve("limboauth.mv.db"); + if (Files.exists(legacyDatabase)) { + Path dumpFile = dir.resolve("limboauth.dump.sql"); + try (Connection legacyConnection = H2_LEGACY_V1.connect(dir, null, null, user, password)) { + try (Statement migrateStatement = legacyConnection.createStatement()) { + migrateStatement.execute("SCRIPT TO '" + dumpFile + "'"); + } + } + + try (Statement migrateStatement = modernConnection.createStatement()) { + migrateStatement.execute("RUNSCRIPT FROM '" + dumpFile + "'"); + } + + Files.delete(dumpFile); + Files.move(legacyDatabase, dir.resolve("limboauth-v1-backup.mv.db")); + } + + return modernConnection; + }, + (dir, hostname, database) -> "jdbc:h2:" + dir + "/limboauth-v2" + ), + MYSQL( + BaseLibrary.MYSQL, + (classLoader, dir, jdbc, user, password) -> fromDriver(classLoader.loadClass("com.mysql.cj.jdbc.Driver"), jdbc, user, password, true), + (dir, hostname, database) -> + "jdbc:mysql://" + hostname + "/" + database + ), + POSTGRESQL( + BaseLibrary.POSTGRESQL, + (classLoader, dir, jdbc, user, password) -> fromDriver(classLoader.loadClass("org.postgresql.Driver"), jdbc, user, password, true), + (dir, hostname, database) -> "jdbc:postgresql://" + hostname + "/" + database + ), + SQLITE( + BaseLibrary.SQLITE, + (classLoader, dir, jdbc, user, password) -> fromDriver(classLoader.loadClass("org.sqlite.JDBC"), jdbc, user, password, true), + (dir, hostname, database) -> "jdbc:sqlite:" + dir + "/limboauth.db" + ); + + private final BaseLibrary baseLibrary; + private final DatabaseConnector connector; + private final DatabaseStringGetter stringGetter; + + DatabaseLibrary(BaseLibrary baseLibrary, DatabaseConnector connector, DatabaseStringGetter stringGetter) { + this.baseLibrary = baseLibrary; + this.connector = connector; + this.stringGetter = stringGetter; + } + + public Connection connect(ClassLoader classLoader, Path dir, String hostname, String database, String user, String password) throws Exception { + return this.connect(classLoader, dir, this.stringGetter.getJdbcString(dir, hostname, database), user, password); + } + + public Connection connect(Path dir, String hostname, String database, String user, String password) throws Exception { + return this.connect(dir, this.stringGetter.getJdbcString(dir, hostname, database), user, password); + } + + public Connection connect(ClassLoader classLoader, Path dir, String jdbc, String user, String password) throws Exception { + return this.connector.connect(classLoader, dir, jdbc, user, password); + } + + public Connection connect(Path dir, String jdbc, String user, String password) throws Exception { + return this.connector.connect(new IsolatedClassLoader(new URL[]{this.baseLibrary.getClassLoaderURL()}), dir, jdbc, user, password); + } + + public ConnectionSource connectToORM(Path dir, String hostname, String database, String user, String password) throws Exception { + String jdbc = this.stringGetter.getJdbcString(dir, hostname, database); + URL baseLibraryURL = this.baseLibrary.getClassLoaderURL(); + ClassLoader currentClassLoader = DatabaseLibrary.class.getClassLoader(); + Method addPath = currentClassLoader.getClass().getDeclaredMethod("addPath", Path.class); + addPath.setAccessible(true); + addPath.invoke(currentClassLoader, Path.of(baseLibraryURL.toURI())); + + return new JdbcSingleConnectionSource(jdbc, this.connect(currentClassLoader, dir, jdbc, hostname, user, password)); + } + + private static Connection fromDriver(Class<?> connectionClass, String jdbc, String user, String password, boolean register) throws Exception { + Constructor<?> legacyConstructor = connectionClass.getConstructor(); + + Properties info = new Properties(); + if (user != null) { + info.put("user", user); + } + + if (password != null) { + info.put("password", password); + } + + Object driver = legacyConstructor.newInstance(); + + if (!register) { + DriverManager.deregisterDriver((Driver) driver); + } + + Method connect = connectionClass.getDeclaredMethod("connect", String.class, Properties.class); + connect.setAccessible(true); + return (Connection) connect.invoke(driver, jdbc, info); + } + + public interface DatabaseConnector { + Connection connect(ClassLoader classLoader, Path dir, String jdbc, String user, String password) throws Exception; + } + + public interface DatabaseStringGetter { + String getJdbcString(Path dir, String hostname, String database); + } +} diff --git a/src/main/java/net/elytrium/limboauth/dependencies/IsolatedClassLoader.java b/src/main/java/net/elytrium/limboauth/dependencies/IsolatedClassLoader.java new file mode 100644 index 0000000..ee53ddd --- /dev/null +++ b/src/main/java/net/elytrium/limboauth/dependencies/IsolatedClassLoader.java @@ -0,0 +1,32 @@ +/* + * Copyright (C) 2021 - 2022 Elytrium + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + */ + +package net.elytrium.limboauth.dependencies; + +import java.net.URL; +import java.net.URLClassLoader; + +public class IsolatedClassLoader extends URLClassLoader { + + public IsolatedClassLoader(URL[] urls) { + super(urls, ClassLoader.getSystemClassLoader().getParent()); + } + + static { + ClassLoader.registerAsParallelCapable(); + } +} |