144 lines
6.8 KiB
Scala
144 lines
6.8 KiB
Scala
/*******************************************************************************/
|
|
/* */
|
|
/* 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.content.Context
|
|
import android.content.pm.ApplicationInfo
|
|
import android.database.sqlite.SQLiteDatabase
|
|
import com.github.shadowsocks.ShadowsocksApplication.app
|
|
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
|
|
|
|
import scala.collection.JavaConverters._
|
|
import scala.collection.mutable
|
|
|
|
object DBHelper {
|
|
final val PROFILE = "profile.db"
|
|
private var apps: mutable.Buffer[ApplicationInfo] = _
|
|
|
|
def isAllDigits(x: String): Boolean = !x.isEmpty && (x forall Character.isDigit)
|
|
|
|
def updateProxiedApps(context: Context, old: String): String = {
|
|
synchronized(if (apps == null) apps = context.getPackageManager.getInstalledApplications(0).asScala)
|
|
val uidSet = old.split('|').filter(isAllDigits).map(_.toInt).toSet
|
|
apps.filter(ai => uidSet.contains(ai.uid)).map(_.packageName).mkString("\n")
|
|
}
|
|
}
|
|
|
|
class DBHelper(val context: Context)
|
|
extends OrmLiteSqliteOpenHelper(context, DBHelper.PROFILE, null, 22) {
|
|
import DBHelper._
|
|
|
|
lazy val profileDao: Dao[Profile, Int] = getDao(classOf[Profile])
|
|
|
|
def onCreate(database: SQLiteDatabase, connectionSource: ConnectionSource) {
|
|
TableUtils.createTable(connectionSource, classOf[Profile])
|
|
}
|
|
|
|
def recreate(database: SQLiteDatabase, connectionSource: ConnectionSource) {
|
|
TableUtils.dropTable(connectionSource, classOf[Profile], true)
|
|
onCreate(database, connectionSource)
|
|
}
|
|
|
|
def onUpgrade(database: SQLiteDatabase, connectionSource: ConnectionSource, oldVersion: Int,
|
|
newVersion: Int) {
|
|
if (oldVersion != newVersion) {
|
|
if (oldVersion < 7) {
|
|
recreate(database, connectionSource)
|
|
return
|
|
}
|
|
|
|
try {
|
|
if (oldVersion < 8) {
|
|
profileDao.executeRawNoArgs("ALTER TABLE `profile` ADD COLUMN udpdns SMALLINT;")
|
|
}
|
|
if (oldVersion < 9) {
|
|
profileDao.executeRawNoArgs("ALTER TABLE `profile` ADD COLUMN route VARCHAR DEFAULT 'all';")
|
|
} else if (oldVersion < 19) {
|
|
profileDao.executeRawNoArgs("UPDATE `profile` SET route = 'all' WHERE route IS NULL;")
|
|
}
|
|
if (oldVersion < 11) {
|
|
profileDao.executeRawNoArgs("ALTER TABLE `profile` ADD COLUMN ipv6 SMALLINT;")
|
|
}
|
|
if (oldVersion < 12) {
|
|
profileDao.executeRawNoArgs("BEGIN TRANSACTION;")
|
|
profileDao.executeRawNoArgs("ALTER TABLE `profile` RENAME TO `tmp`;")
|
|
TableUtils.createTable(connectionSource, classOf[Profile])
|
|
profileDao.executeRawNoArgs(
|
|
"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.executeRawNoArgs("DROP TABLE `tmp`;")
|
|
profileDao.executeRawNoArgs("COMMIT;")
|
|
} else if (oldVersion < 13) {
|
|
profileDao.executeRawNoArgs("ALTER TABLE `profile` ADD COLUMN tx LONG;")
|
|
profileDao.executeRawNoArgs("ALTER TABLE `profile` ADD COLUMN rx LONG;")
|
|
profileDao.executeRawNoArgs("ALTER TABLE `profile` ADD COLUMN date VARCHAR;")
|
|
}
|
|
|
|
if (oldVersion < 15) {
|
|
if (oldVersion >= 12) profileDao.executeRawNoArgs("ALTER TABLE `profile` ADD COLUMN userOrder LONG;")
|
|
var i = 0
|
|
for (profile <- profileDao.queryForAll.asScala) {
|
|
if (oldVersion < 14) profile.individual = updateProxiedApps(context, profile.individual)
|
|
profile.userOrder = i
|
|
profileDao.update(profile)
|
|
i += 1
|
|
}
|
|
}
|
|
|
|
if (oldVersion < 16) {
|
|
profileDao.executeRawNoArgs("UPDATE `profile` SET route = 'bypass-lan-china' WHERE route = 'bypass-china'")
|
|
}
|
|
|
|
if (oldVersion < 21) {
|
|
profileDao.executeRawNoArgs("ALTER TABLE `profile` ADD COLUMN remoteDns VARCHAR DEFAULT '8.8.8.8';")
|
|
}
|
|
|
|
if (oldVersion < 17) {
|
|
profileDao.executeRawNoArgs("ALTER TABLE `profile` ADD COLUMN plugin VARCHAR;")
|
|
} else if (oldVersion < 22) {
|
|
// upgrade kcptun to SIP003 plugin
|
|
profileDao.executeRawNoArgs("BEGIN TRANSACTION;")
|
|
profileDao.executeRawNoArgs("ALTER TABLE `profile` RENAME TO `tmp`;")
|
|
TableUtils.createTable(connectionSource, classOf[Profile])
|
|
profileDao.executeRawNoArgs(
|
|
"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.executeRawNoArgs("DROP TABLE `tmp`;")
|
|
profileDao.executeRawNoArgs("COMMIT;")
|
|
}
|
|
} catch {
|
|
case ex: Exception =>
|
|
app.track(ex)
|
|
recreate(database, connectionSource)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|