157 lines
8.2 KiB
Kotlin
157 lines
8.2 KiB
Kotlin
/*******************************************************************************
|
|
* *
|
|
* Copyright (C) 2017 by Max Lv <max.c.lv@gmail.com> *
|
|
* Copyright (C) 2017 by Mygod Studio <contact-shadowsocks-android@mygod.be> *
|
|
* *
|
|
* This program is free software: you can redistribute it and/or modify *
|
|
* it under the terms of the GNU 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 General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU General Public License *
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>. *
|
|
* *
|
|
*******************************************************************************/
|
|
|
|
package com.github.shadowsocks.database
|
|
|
|
import android.database.sqlite.SQLiteDatabase
|
|
import android.support.v7.preference.PreferenceManager
|
|
import android.text.TextUtils
|
|
import com.github.shadowsocks.App.Companion.app
|
|
import com.github.shadowsocks.utils.Key
|
|
import com.j256.ormlite.android.AndroidDatabaseConnection
|
|
import com.j256.ormlite.android.apptools.OrmLiteSqliteOpenHelper
|
|
import com.j256.ormlite.dao.Dao
|
|
import com.j256.ormlite.support.ConnectionSource
|
|
import com.j256.ormlite.table.TableUtils
|
|
|
|
object PrivateDatabase : OrmLiteSqliteOpenHelper(app, Key.DB_PROFILE, null, 25) {
|
|
@Suppress("UNCHECKED_CAST")
|
|
val profileDao: Dao<Profile, Int> by lazy { getDao(Profile::class.java) as Dao<Profile, Int> }
|
|
@Suppress("UNCHECKED_CAST")
|
|
val kvPairDao: Dao<KeyValuePair, String?> by lazy { getDao(KeyValuePair::class.java) as Dao<KeyValuePair, String?> }
|
|
|
|
override fun onCreate(database: SQLiteDatabase?, connectionSource: ConnectionSource) {
|
|
connectionSource.createTableSafe<Profile>()
|
|
connectionSource.createTableSafe<KeyValuePair>()
|
|
}
|
|
|
|
private fun recreate(database: SQLiteDatabase?, connectionSource: ConnectionSource) {
|
|
TableUtils.dropTable<Profile, Int>(connectionSource, Profile::class.java, true)
|
|
TableUtils.dropTable<KeyValuePair, String?>(connectionSource, KeyValuePair::class.java, true)
|
|
onCreate(database, connectionSource)
|
|
}
|
|
|
|
override fun onUpgrade(database: SQLiteDatabase?, connectionSource: ConnectionSource,
|
|
oldVersion: Int, newVersion: Int) {
|
|
if (oldVersion < 7) {
|
|
recreate(database, connectionSource)
|
|
return
|
|
}
|
|
|
|
try {
|
|
if (oldVersion < 8) {
|
|
profileDao.executeSafe("ALTER TABLE `profile` ADD COLUMN udpdns SMALLINT;")
|
|
}
|
|
if (oldVersion < 9) {
|
|
profileDao.executeSafe("ALTER TABLE `profile` ADD COLUMN route VARCHAR DEFAULT 'all';")
|
|
} else if (oldVersion < 19) {
|
|
profileDao.executeSafe("UPDATE `profile` SET route = 'all' WHERE route IS NULL;")
|
|
}
|
|
if (oldVersion < 11) {
|
|
profileDao.executeSafe("ALTER TABLE `profile` ADD COLUMN ipv6 SMALLINT;")
|
|
}
|
|
if (oldVersion < 12) {
|
|
profileDao.executeSafe("BEGIN TRANSACTION;")
|
|
profileDao.executeSafe("ALTER TABLE `profile` RENAME TO `tmp`;")
|
|
connectionSource.createTableSafe<Profile>()
|
|
profileDao.executeSafe(
|
|
"INSERT INTO `profile`(id, name, host, localPort, remotePort, password, method, route," +
|
|
" proxyApps, bypass, udpdns, ipv6, individual) " +
|
|
"SELECT id, name, host, localPort, remotePort, password, method, route, 1 - global," +
|
|
" bypass, udpdns, ipv6, individual FROM `tmp`;")
|
|
profileDao.executeSafe("DROP TABLE `tmp`;")
|
|
profileDao.executeSafe("COMMIT;")
|
|
} else if (oldVersion < 13) {
|
|
profileDao.executeSafe("ALTER TABLE `profile` ADD COLUMN tx LONG;")
|
|
profileDao.executeSafe("ALTER TABLE `profile` ADD COLUMN rx LONG;")
|
|
profileDao.executeSafe("ALTER TABLE `profile` ADD COLUMN date VARCHAR;")
|
|
}
|
|
|
|
if (oldVersion < 15) {
|
|
if (oldVersion >= 12) profileDao.executeSafe("ALTER TABLE `profile` ADD COLUMN userOrder LONG;")
|
|
var i = 0L
|
|
val apps by lazy { app.packageManager.getInstalledApplications(0) }
|
|
for (profile in profileDao.queryAllSafe()) {
|
|
if (oldVersion < 14) {
|
|
val uidSet = profile.individual.split('|').filter(TextUtils::isDigitsOnly)
|
|
.map(String::toInt).toSet()
|
|
profile.individual = apps.filter { uidSet.contains(it.uid) }
|
|
.joinToString("\n") { it.packageName }
|
|
}
|
|
profile.userOrder = i
|
|
profileDao.updateSafe(profile)
|
|
i += 1
|
|
}
|
|
}
|
|
|
|
if (oldVersion < 16) {
|
|
profileDao.executeSafe(
|
|
"UPDATE `profile` SET route = 'bypass-lan-china' WHERE route = 'bypass-china'")
|
|
}
|
|
|
|
if (oldVersion < 21) {
|
|
profileDao.executeSafe("ALTER TABLE `profile` ADD COLUMN remoteDns VARCHAR DEFAULT '8.8.8.8';")
|
|
}
|
|
|
|
if (oldVersion < 17) {
|
|
profileDao.executeSafe("ALTER TABLE `profile` ADD COLUMN plugin VARCHAR;")
|
|
} else if (oldVersion < 22) {
|
|
// upgrade kcptun to SIP003 plugin
|
|
profileDao.executeSafe("BEGIN TRANSACTION;")
|
|
profileDao.executeSafe("ALTER TABLE `profile` RENAME TO `tmp`;")
|
|
connectionSource.createTableSafe<Profile>()
|
|
profileDao.executeSafe(
|
|
"INSERT INTO `profile`(id, name, host, localPort, remotePort, password, method, route, " +
|
|
"remoteDns, proxyApps, bypass, udpdns, ipv6, individual, tx, rx, date, userOrder, " +
|
|
"plugin) " +
|
|
"SELECT id, name, host, localPort, " +
|
|
"CASE WHEN kcp = 1 THEN kcpPort ELSE remotePort END, password, method, route, " +
|
|
"remoteDns, proxyApps, bypass, udpdns, ipv6, individual, tx, rx, date, userOrder, " +
|
|
"CASE WHEN kcp = 1 THEN 'kcptun ' || kcpcli ELSE NULL END FROM `tmp`;")
|
|
profileDao.executeSafe("DROP TABLE `tmp`;")
|
|
profileDao.executeSafe("COMMIT;")
|
|
}
|
|
|
|
if (oldVersion < 23) {
|
|
profileDao.executeSafe("BEGIN TRANSACTION;")
|
|
connectionSource.createTableSafe<KeyValuePair>()
|
|
profileDao.executeSafe("COMMIT;")
|
|
val old = PreferenceManager.getDefaultSharedPreferences(app)
|
|
PublicDatabase.kvPairDao.replaceSafe(KeyValuePair(Key.id).put(old.getInt(Key.id, 0)))
|
|
PublicDatabase.kvPairDao.replaceSafe(KeyValuePair(Key.tfo).put(old.getBoolean(Key.tfo, false)))
|
|
}
|
|
|
|
if (oldVersion < 25) {
|
|
PublicDatabase.onUpgrade(database, 0, -1)
|
|
}
|
|
} catch (ex: Exception) {
|
|
app.track(ex)
|
|
recreate(database, connectionSource)
|
|
}
|
|
}
|
|
|
|
override fun onDowngrade(db: SQLiteDatabase?, oldVersion: Int, newVersion: Int) {
|
|
val connection = AndroidDatabaseConnection(db, true)
|
|
connectionSource.saveSpecialConnection(connection)
|
|
recreate(db, connectionSource)
|
|
connectionSource.clearSpecialConnection(connection)
|
|
}
|
|
}
|