mysql.go 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764
  1. //go:build !nomysql
  2. // +build !nomysql
  3. package dataprovider
  4. import (
  5. "context"
  6. "crypto/tls"
  7. "crypto/x509"
  8. "database/sql"
  9. "errors"
  10. "fmt"
  11. "os"
  12. "strings"
  13. "time"
  14. "github.com/go-sql-driver/mysql"
  15. "github.com/drakkan/sftpgo/v2/logger"
  16. "github.com/drakkan/sftpgo/v2/version"
  17. "github.com/drakkan/sftpgo/v2/vfs"
  18. )
  19. const (
  20. mysqlResetSQL = "DROP TABLE IF EXISTS `{{api_keys}}` CASCADE;" +
  21. "DROP TABLE IF EXISTS `{{folders_mapping}}` CASCADE;" +
  22. "DROP TABLE IF EXISTS `{{users_folders_mapping}}` CASCADE;" +
  23. "DROP TABLE IF EXISTS `{{users_groups_mapping}}` CASCADE;" +
  24. "DROP TABLE IF EXISTS `{{groups_folders_mapping}}` CASCADE;" +
  25. "DROP TABLE IF EXISTS `{{admins}}` CASCADE;" +
  26. "DROP TABLE IF EXISTS `{{folders}}` CASCADE;" +
  27. "DROP TABLE IF EXISTS `{{shares}}` CASCADE;" +
  28. "DROP TABLE IF EXISTS `{{users}}` CASCADE;" +
  29. "DROP TABLE IF EXISTS `{{groups}}` CASCADE;" +
  30. "DROP TABLE IF EXISTS `{{defender_events}}` CASCADE;" +
  31. "DROP TABLE IF EXISTS `{{defender_hosts}}` CASCADE;" +
  32. "DROP TABLE IF EXISTS `{{active_transfers}}` CASCADE;" +
  33. "DROP TABLE IF EXISTS `{{shared_sessions}}` CASCADE;" +
  34. "DROP TABLE IF EXISTS `{{schema_version}}` CASCADE;"
  35. mysqlInitialSQL = "CREATE TABLE `{{schema_version}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, `version` integer NOT NULL);" +
  36. "CREATE TABLE `{{admins}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, `username` varchar(255) NOT NULL UNIQUE, " +
  37. "`description` varchar(512) NULL, `password` varchar(255) NOT NULL, `email` varchar(255) NULL, `status` integer NOT NULL, " +
  38. "`permissions` longtext NOT NULL, `filters` longtext NULL, `additional_info` longtext NULL, `last_login` bigint NOT NULL, " +
  39. "`created_at` bigint NOT NULL, `updated_at` bigint NOT NULL);" +
  40. "CREATE TABLE `{{defender_hosts}}` (`id` bigint AUTO_INCREMENT NOT NULL PRIMARY KEY, " +
  41. "`ip` varchar(50) NOT NULL UNIQUE, `ban_time` bigint NOT NULL, `updated_at` bigint NOT NULL);" +
  42. "CREATE TABLE `{{defender_events}}` (`id` bigint AUTO_INCREMENT NOT NULL PRIMARY KEY, " +
  43. "`date_time` bigint NOT NULL, `score` integer NOT NULL, `host_id` bigint NOT NULL);" +
  44. "ALTER TABLE `{{defender_events}}` ADD CONSTRAINT `{{prefix}}defender_events_host_id_fk_defender_hosts_id` " +
  45. "FOREIGN KEY (`host_id`) REFERENCES `{{defender_hosts}}` (`id`) ON DELETE CASCADE;" +
  46. "CREATE TABLE `{{folders}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, `name` varchar(255) NOT NULL UNIQUE, " +
  47. "`description` varchar(512) NULL, `path` longtext NULL, `used_quota_size` bigint NOT NULL, " +
  48. "`used_quota_files` integer NOT NULL, `last_quota_update` bigint NOT NULL, `filesystem` longtext NULL);" +
  49. "CREATE TABLE `{{users}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, `username` varchar(255) NOT NULL UNIQUE, " +
  50. "`status` integer NOT NULL, `expiration_date` bigint NOT NULL, `description` varchar(512) NULL, `password` longtext NULL, " +
  51. "`public_keys` longtext NULL, `home_dir` longtext NOT NULL, `uid` bigint NOT NULL, `gid` bigint NOT NULL, " +
  52. "`max_sessions` integer NOT NULL, `quota_size` bigint NOT NULL, `quota_files` integer NOT NULL, " +
  53. "`permissions` longtext NOT NULL, `used_quota_size` bigint NOT NULL, `used_quota_files` integer NOT NULL, " +
  54. "`last_quota_update` bigint NOT NULL, `upload_bandwidth` integer NOT NULL, `download_bandwidth` integer NOT NULL, " +
  55. "`last_login` bigint NOT NULL, `filters` longtext NULL, `filesystem` longtext NULL, `additional_info` longtext NULL, " +
  56. "`created_at` bigint NOT NULL, `updated_at` bigint NOT NULL, `email` varchar(255) NULL);" +
  57. "CREATE TABLE `{{folders_mapping}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, `virtual_path` longtext NOT NULL, " +
  58. "`quota_size` bigint NOT NULL, `quota_files` integer NOT NULL, `folder_id` integer NOT NULL, `user_id` integer NOT NULL);" +
  59. "ALTER TABLE `{{folders_mapping}}` ADD CONSTRAINT `{{prefix}}unique_mapping` UNIQUE (`user_id`, `folder_id`);" +
  60. "ALTER TABLE `{{folders_mapping}}` ADD CONSTRAINT `{{prefix}}folders_mapping_folder_id_fk_folders_id` FOREIGN KEY (`folder_id`) REFERENCES `{{folders}}` (`id`) ON DELETE CASCADE;" +
  61. "ALTER TABLE `{{folders_mapping}}` ADD CONSTRAINT `{{prefix}}folders_mapping_user_id_fk_users_id` FOREIGN KEY (`user_id`) REFERENCES `{{users}}` (`id`) ON DELETE CASCADE;" +
  62. "CREATE TABLE `{{shares}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, " +
  63. "`share_id` varchar(60) NOT NULL UNIQUE, `name` varchar(255) NOT NULL, `description` varchar(512) NULL, " +
  64. "`scope` integer NOT NULL, `paths` longtext NOT NULL, `created_at` bigint NOT NULL, " +
  65. "`updated_at` bigint NOT NULL, `last_use_at` bigint NOT NULL, `expires_at` bigint NOT NULL, " +
  66. "`password` longtext NULL, `max_tokens` integer NOT NULL, `used_tokens` integer NOT NULL, " +
  67. "`allow_from` longtext NULL, `user_id` integer NOT NULL);" +
  68. "ALTER TABLE `{{shares}}` ADD CONSTRAINT `{{prefix}}shares_user_id_fk_users_id` " +
  69. "FOREIGN KEY (`user_id`) REFERENCES `{{users}}` (`id`) ON DELETE CASCADE;" +
  70. "CREATE TABLE `{{api_keys}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, `name` varchar(255) NOT NULL, `key_id` varchar(50) NOT NULL UNIQUE," +
  71. "`api_key` varchar(255) NOT NULL UNIQUE, `scope` integer NOT NULL, `created_at` bigint NOT NULL, `updated_at` bigint NOT NULL, `last_use_at` bigint NOT NULL, " +
  72. "`expires_at` bigint NOT NULL, `description` longtext NULL, `admin_id` integer NULL, `user_id` integer NULL);" +
  73. "ALTER TABLE `{{api_keys}}` ADD CONSTRAINT `{{prefix}}api_keys_admin_id_fk_admins_id` FOREIGN KEY (`admin_id`) REFERENCES `{{admins}}` (`id`) ON DELETE CASCADE;" +
  74. "ALTER TABLE `{{api_keys}}` ADD CONSTRAINT `{{prefix}}api_keys_user_id_fk_users_id` FOREIGN KEY (`user_id`) REFERENCES `{{users}}` (`id`) ON DELETE CASCADE;" +
  75. "CREATE INDEX `{{prefix}}users_updated_at_idx` ON `{{users}}` (`updated_at`);" +
  76. "CREATE INDEX `{{prefix}}defender_hosts_updated_at_idx` ON `{{defender_hosts}}` (`updated_at`);" +
  77. "CREATE INDEX `{{prefix}}defender_hosts_ban_time_idx` ON `{{defender_hosts}}` (`ban_time`);" +
  78. "CREATE INDEX `{{prefix}}defender_events_date_time_idx` ON `{{defender_events}}` (`date_time`);" +
  79. "INSERT INTO {{schema_version}} (version) VALUES (15);"
  80. mysqlV16SQL = "ALTER TABLE `{{users}}` ADD COLUMN `download_data_transfer` integer DEFAULT 0 NOT NULL;" +
  81. "ALTER TABLE `{{users}}` ALTER COLUMN `download_data_transfer` DROP DEFAULT;" +
  82. "ALTER TABLE `{{users}}` ADD COLUMN `total_data_transfer` integer DEFAULT 0 NOT NULL;" +
  83. "ALTER TABLE `{{users}}` ALTER COLUMN `total_data_transfer` DROP DEFAULT;" +
  84. "ALTER TABLE `{{users}}` ADD COLUMN `upload_data_transfer` integer DEFAULT 0 NOT NULL;" +
  85. "ALTER TABLE `{{users}}` ALTER COLUMN `upload_data_transfer` DROP DEFAULT;" +
  86. "ALTER TABLE `{{users}}` ADD COLUMN `used_download_data_transfer` integer DEFAULT 0 NOT NULL;" +
  87. "ALTER TABLE `{{users}}` ALTER COLUMN `used_download_data_transfer` DROP DEFAULT;" +
  88. "ALTER TABLE `{{users}}` ADD COLUMN `used_upload_data_transfer` integer DEFAULT 0 NOT NULL;" +
  89. "ALTER TABLE `{{users}}` ALTER COLUMN `used_upload_data_transfer` DROP DEFAULT;" +
  90. "CREATE TABLE `{{active_transfers}}` (`id` bigint AUTO_INCREMENT NOT NULL PRIMARY KEY, " +
  91. "`connection_id` varchar(100) NOT NULL, `transfer_id` bigint NOT NULL, `transfer_type` integer NOT NULL, " +
  92. "`username` varchar(255) NOT NULL, `folder_name` varchar(255) NULL, `ip` varchar(50) NOT NULL, " +
  93. "`truncated_size` bigint NOT NULL, `current_ul_size` bigint NOT NULL, `current_dl_size` bigint NOT NULL, " +
  94. "`created_at` bigint NOT NULL, `updated_at` bigint NOT NULL);" +
  95. "CREATE INDEX `{{prefix}}active_transfers_connection_id_idx` ON `{{active_transfers}}` (`connection_id`);" +
  96. "CREATE INDEX `{{prefix}}active_transfers_transfer_id_idx` ON `{{active_transfers}}` (`transfer_id`);" +
  97. "CREATE INDEX `{{prefix}}active_transfers_updated_at_idx` ON `{{active_transfers}}` (`updated_at`);"
  98. mysqlV16DownSQL = "ALTER TABLE `{{users}}` DROP COLUMN `used_upload_data_transfer`;" +
  99. "ALTER TABLE `{{users}}` DROP COLUMN `used_download_data_transfer`;" +
  100. "ALTER TABLE `{{users}}` DROP COLUMN `upload_data_transfer`;" +
  101. "ALTER TABLE `{{users}}` DROP COLUMN `total_data_transfer`;" +
  102. "ALTER TABLE `{{users}}` DROP COLUMN `download_data_transfer`;" +
  103. "DROP TABLE `{{active_transfers}}` CASCADE;"
  104. mysqlV17SQL = "CREATE TABLE `{{groups}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, " +
  105. "`name` varchar(255) NOT NULL UNIQUE, `description` varchar(512) NULL, `created_at` bigint NOT NULL, " +
  106. "`updated_at` bigint NOT NULL, `user_settings` longtext NULL);" +
  107. "CREATE TABLE `{{groups_folders_mapping}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, " +
  108. "`group_id` integer NOT NULL, `folder_id` integer NOT NULL, " +
  109. "`virtual_path` longtext NOT NULL, `quota_size` bigint NOT NULL, `quota_files` integer NOT NULL);" +
  110. "CREATE TABLE `{{users_groups_mapping}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, " +
  111. "`user_id` integer NOT NULL, `group_id` integer NOT NULL, `group_type` integer NOT NULL);" +
  112. "ALTER TABLE `{{folders_mapping}}` DROP FOREIGN KEY `{{prefix}}folders_mapping_folder_id_fk_folders_id`;" +
  113. "ALTER TABLE `{{folders_mapping}}` DROP FOREIGN KEY `{{prefix}}folders_mapping_user_id_fk_users_id`;" +
  114. "ALTER TABLE `{{folders_mapping}}` DROP INDEX `{{prefix}}unique_mapping`;" +
  115. "RENAME TABLE `{{folders_mapping}}` TO `{{users_folders_mapping}}`;" +
  116. "ALTER TABLE `{{users_folders_mapping}}` ADD CONSTRAINT `{{prefix}}unique_user_folder_mapping` " +
  117. "UNIQUE (`user_id`, `folder_id`);" +
  118. "ALTER TABLE `{{users_folders_mapping}}` ADD CONSTRAINT `{{prefix}}users_folders_mapping_user_id_fk_users_id` " +
  119. "FOREIGN KEY (`user_id`) REFERENCES `{{users}}` (`id`) ON DELETE CASCADE;" +
  120. "ALTER TABLE `{{users_folders_mapping}}` ADD CONSTRAINT `{{prefix}}users_folders_mapping_folder_id_fk_folders_id` " +
  121. "FOREIGN KEY (`folder_id`) REFERENCES `{{folders}}` (`id`) ON DELETE CASCADE;" +
  122. "ALTER TABLE `{{users_groups_mapping}}` ADD CONSTRAINT `{{prefix}}unique_user_group_mapping` UNIQUE (`user_id`, `group_id`);" +
  123. "ALTER TABLE `{{groups_folders_mapping}}` ADD CONSTRAINT `{{prefix}}unique_group_folder_mapping` UNIQUE (`group_id`, `folder_id`);" +
  124. "ALTER TABLE `{{users_groups_mapping}}` ADD CONSTRAINT `{{prefix}}users_groups_mapping_group_id_fk_groups_id` " +
  125. "FOREIGN KEY (`group_id`) REFERENCES `{{groups}}` (`id`) ON DELETE NO ACTION;" +
  126. "ALTER TABLE `{{users_groups_mapping}}` ADD CONSTRAINT `{{prefix}}users_groups_mapping_user_id_fk_users_id` " +
  127. "FOREIGN KEY (`user_id`) REFERENCES `{{users}}` (`id`) ON DELETE CASCADE;" +
  128. "ALTER TABLE `{{groups_folders_mapping}}` ADD CONSTRAINT `{{prefix}}groups_folders_mapping_folder_id_fk_folders_id` " +
  129. "FOREIGN KEY (`folder_id`) REFERENCES `{{folders}}` (`id`) ON DELETE CASCADE;" +
  130. "ALTER TABLE `{{groups_folders_mapping}}` ADD CONSTRAINT `{{prefix}}groups_folders_mapping_group_id_fk_groups_id` " +
  131. "FOREIGN KEY (`group_id`) REFERENCES `{{groups}}` (`id`) ON DELETE CASCADE;" +
  132. "CREATE INDEX `{{prefix}}groups_updated_at_idx` ON `{{groups}}` (`updated_at`);"
  133. mysqlV17DownSQL = "ALTER TABLE `{{groups_folders_mapping}}` DROP FOREIGN KEY `{{prefix}}groups_folders_mapping_group_id_fk_groups_id`;" +
  134. "ALTER TABLE `{{groups_folders_mapping}}` DROP FOREIGN KEY `{{prefix}}groups_folders_mapping_folder_id_fk_folders_id`;" +
  135. "ALTER TABLE `{{users_groups_mapping}}` DROP FOREIGN KEY `{{prefix}}users_groups_mapping_user_id_fk_users_id`;" +
  136. "ALTER TABLE `{{users_groups_mapping}}` DROP FOREIGN KEY `{{prefix}}users_groups_mapping_group_id_fk_groups_id`;" +
  137. "ALTER TABLE `{{groups_folders_mapping}}` DROP INDEX `{{prefix}}unique_group_folder_mapping`;" +
  138. "ALTER TABLE `{{users_groups_mapping}}` DROP INDEX `{{prefix}}unique_user_group_mapping`;" +
  139. "DROP TABLE `{{users_groups_mapping}}` CASCADE;" +
  140. "DROP TABLE `{{groups_folders_mapping}}` CASCADE;" +
  141. "DROP TABLE `{{groups}}` CASCADE;" +
  142. "ALTER TABLE `{{users_folders_mapping}}` DROP FOREIGN KEY `{{prefix}}users_folders_mapping_folder_id_fk_folders_id`;" +
  143. "ALTER TABLE `{{users_folders_mapping}}` DROP FOREIGN KEY `{{prefix}}users_folders_mapping_user_id_fk_users_id`;" +
  144. "ALTER TABLE `{{users_folders_mapping}}` DROP INDEX `{{prefix}}unique_user_folder_mapping`;" +
  145. "RENAME TABLE `{{users_folders_mapping}}` TO `{{folders_mapping}}`;" +
  146. "ALTER TABLE `{{folders_mapping}}` ADD CONSTRAINT `{{prefix}}unique_mapping` UNIQUE (`user_id`, `folder_id`);" +
  147. "ALTER TABLE `{{folders_mapping}}` ADD CONSTRAINT `{{prefix}}folders_mapping_user_id_fk_users_id` " +
  148. "FOREIGN KEY (`user_id`) REFERENCES `{{users}}` (`id`) ON DELETE CASCADE;" +
  149. "ALTER TABLE `{{folders_mapping}}` ADD CONSTRAINT `{{prefix}}folders_mapping_folder_id_fk_folders_id` " +
  150. "FOREIGN KEY (`folder_id`) REFERENCES `{{folders}}` (`id`) ON DELETE CASCADE;"
  151. mysqlV19SQL = "CREATE TABLE `{{shared_sessions}}` (`key` varchar(128) NOT NULL PRIMARY KEY, " +
  152. "`data` longtext NOT NULL, `type` integer NOT NULL, `timestamp` bigint NOT NULL);" +
  153. "CREATE INDEX `{{prefix}}shared_sessions_type_idx` ON `{{shared_sessions}}` (`type`);" +
  154. "CREATE INDEX `{{prefix}}shared_sessions_timestamp_idx` ON `{{shared_sessions}}` (`timestamp`);"
  155. mysqlV19DownSQL = "DROP TABLE `{{shared_sessions}}` CASCADE;"
  156. )
  157. // MySQLProvider defines the auth provider for MySQL/MariaDB database
  158. type MySQLProvider struct {
  159. dbHandle *sql.DB
  160. }
  161. func init() {
  162. version.AddFeature("+mysql")
  163. }
  164. func initializeMySQLProvider() error {
  165. var err error
  166. connString, err := getMySQLConnectionString(false)
  167. if err != nil {
  168. return err
  169. }
  170. redactedConnString, err := getMySQLConnectionString(true)
  171. if err != nil {
  172. return err
  173. }
  174. dbHandle, err := sql.Open("mysql", connString)
  175. if err == nil {
  176. providerLog(logger.LevelDebug, "mysql database handle created, connection string: %#v, pool size: %v",
  177. redactedConnString, config.PoolSize)
  178. dbHandle.SetMaxOpenConns(config.PoolSize)
  179. if config.PoolSize > 0 {
  180. dbHandle.SetMaxIdleConns(config.PoolSize)
  181. } else {
  182. dbHandle.SetMaxIdleConns(2)
  183. }
  184. dbHandle.SetConnMaxLifetime(240 * time.Second)
  185. provider = &MySQLProvider{dbHandle: dbHandle}
  186. } else {
  187. providerLog(logger.LevelError, "error creating mysql database handler, connection string: %#v, error: %v",
  188. redactedConnString, err)
  189. }
  190. return err
  191. }
  192. func getMySQLConnectionString(redactedPwd bool) (string, error) {
  193. var connectionString string
  194. if config.ConnectionString == "" {
  195. password := config.Password
  196. if redactedPwd && password != "" {
  197. password = "[redacted]"
  198. }
  199. sslMode := getSSLMode()
  200. if sslMode == "custom" && !redactedPwd {
  201. tlsConfig := &tls.Config{}
  202. if config.RootCert != "" {
  203. rootCAs, err := x509.SystemCertPool()
  204. if err != nil {
  205. rootCAs = x509.NewCertPool()
  206. }
  207. rootCrt, err := os.ReadFile(config.RootCert)
  208. if err != nil {
  209. return "", fmt.Errorf("unable to load root certificate %#v: %v", config.RootCert, err)
  210. }
  211. if !rootCAs.AppendCertsFromPEM(rootCrt) {
  212. return "", fmt.Errorf("unable to parse root certificate %#v", config.RootCert)
  213. }
  214. tlsConfig.RootCAs = rootCAs
  215. }
  216. if config.ClientCert != "" && config.ClientKey != "" {
  217. clientCert := make([]tls.Certificate, 0, 1)
  218. tlsCert, err := tls.LoadX509KeyPair(config.ClientCert, config.ClientKey)
  219. if err != nil {
  220. return "", fmt.Errorf("unable to load key pair %#v, %#v: %v", config.ClientCert, config.ClientKey, err)
  221. }
  222. clientCert = append(clientCert, tlsCert)
  223. tlsConfig.Certificates = clientCert
  224. }
  225. if config.SSLMode == 2 {
  226. tlsConfig.InsecureSkipVerify = true
  227. }
  228. providerLog(logger.LevelInfo, "registering custom TLS config, root cert %#v, client cert %#v, client key %#v",
  229. config.RootCert, config.ClientCert, config.ClientKey)
  230. if err := mysql.RegisterTLSConfig("custom", tlsConfig); err != nil {
  231. return "", fmt.Errorf("unable to register tls config: %v", err)
  232. }
  233. }
  234. connectionString = fmt.Sprintf("%v:%v@tcp([%v]:%v)/%v?charset=utf8mb4&interpolateParams=true&timeout=10s&parseTime=true&tls=%v&writeTimeout=60s&readTimeout=60s",
  235. config.Username, password, config.Host, config.Port, config.Name, sslMode)
  236. } else {
  237. connectionString = config.ConnectionString
  238. }
  239. return connectionString, nil
  240. }
  241. func (p *MySQLProvider) checkAvailability() error {
  242. return sqlCommonCheckAvailability(p.dbHandle)
  243. }
  244. func (p *MySQLProvider) validateUserAndPass(username, password, ip, protocol string) (User, error) {
  245. return sqlCommonValidateUserAndPass(username, password, ip, protocol, p.dbHandle)
  246. }
  247. func (p *MySQLProvider) validateUserAndTLSCert(username, protocol string, tlsCert *x509.Certificate) (User, error) {
  248. return sqlCommonValidateUserAndTLSCertificate(username, protocol, tlsCert, p.dbHandle)
  249. }
  250. func (p *MySQLProvider) validateUserAndPubKey(username string, publicKey []byte, isSSHCert bool) (User, string, error) {
  251. return sqlCommonValidateUserAndPubKey(username, publicKey, isSSHCert, p.dbHandle)
  252. }
  253. func (p *MySQLProvider) updateTransferQuota(username string, uploadSize, downloadSize int64, reset bool) error {
  254. return sqlCommonUpdateTransferQuota(username, uploadSize, downloadSize, reset, p.dbHandle)
  255. }
  256. func (p *MySQLProvider) updateQuota(username string, filesAdd int, sizeAdd int64, reset bool) error {
  257. return sqlCommonUpdateQuota(username, filesAdd, sizeAdd, reset, p.dbHandle)
  258. }
  259. func (p *MySQLProvider) getUsedQuota(username string) (int, int64, int64, int64, error) {
  260. return sqlCommonGetUsedQuota(username, p.dbHandle)
  261. }
  262. func (p *MySQLProvider) setUpdatedAt(username string) {
  263. sqlCommonSetUpdatedAt(username, p.dbHandle)
  264. }
  265. func (p *MySQLProvider) updateLastLogin(username string) error {
  266. return sqlCommonUpdateLastLogin(username, p.dbHandle)
  267. }
  268. func (p *MySQLProvider) updateAdminLastLogin(username string) error {
  269. return sqlCommonUpdateAdminLastLogin(username, p.dbHandle)
  270. }
  271. func (p *MySQLProvider) userExists(username string) (User, error) {
  272. return sqlCommonGetUserByUsername(username, p.dbHandle)
  273. }
  274. func (p *MySQLProvider) addUser(user *User) error {
  275. return sqlCommonAddUser(user, p.dbHandle)
  276. }
  277. func (p *MySQLProvider) updateUser(user *User) error {
  278. return sqlCommonUpdateUser(user, p.dbHandle)
  279. }
  280. func (p *MySQLProvider) deleteUser(user User) error {
  281. return sqlCommonDeleteUser(user, p.dbHandle)
  282. }
  283. func (p *MySQLProvider) updateUserPassword(username, password string) error {
  284. return sqlCommonUpdateUserPassword(username, password, p.dbHandle)
  285. }
  286. func (p *MySQLProvider) dumpUsers() ([]User, error) {
  287. return sqlCommonDumpUsers(p.dbHandle)
  288. }
  289. func (p *MySQLProvider) getRecentlyUpdatedUsers(after int64) ([]User, error) {
  290. return sqlCommonGetRecentlyUpdatedUsers(after, p.dbHandle)
  291. }
  292. func (p *MySQLProvider) getUsers(limit int, offset int, order string) ([]User, error) {
  293. return sqlCommonGetUsers(limit, offset, order, p.dbHandle)
  294. }
  295. func (p *MySQLProvider) getUsersForQuotaCheck(toFetch map[string]bool) ([]User, error) {
  296. return sqlCommonGetUsersForQuotaCheck(toFetch, p.dbHandle)
  297. }
  298. func (p *MySQLProvider) dumpFolders() ([]vfs.BaseVirtualFolder, error) {
  299. return sqlCommonDumpFolders(p.dbHandle)
  300. }
  301. func (p *MySQLProvider) getFolders(limit, offset int, order string, minimal bool) ([]vfs.BaseVirtualFolder, error) {
  302. return sqlCommonGetFolders(limit, offset, order, minimal, p.dbHandle)
  303. }
  304. func (p *MySQLProvider) getFolderByName(name string) (vfs.BaseVirtualFolder, error) {
  305. ctx, cancel := context.WithTimeout(context.Background(), defaultSQLQueryTimeout)
  306. defer cancel()
  307. return sqlCommonGetFolderByName(ctx, name, p.dbHandle)
  308. }
  309. func (p *MySQLProvider) addFolder(folder *vfs.BaseVirtualFolder) error {
  310. return sqlCommonAddFolder(folder, p.dbHandle)
  311. }
  312. func (p *MySQLProvider) updateFolder(folder *vfs.BaseVirtualFolder) error {
  313. return sqlCommonUpdateFolder(folder, p.dbHandle)
  314. }
  315. func (p *MySQLProvider) deleteFolder(folder vfs.BaseVirtualFolder) error {
  316. return sqlCommonDeleteFolder(folder, p.dbHandle)
  317. }
  318. func (p *MySQLProvider) updateFolderQuota(name string, filesAdd int, sizeAdd int64, reset bool) error {
  319. return sqlCommonUpdateFolderQuota(name, filesAdd, sizeAdd, reset, p.dbHandle)
  320. }
  321. func (p *MySQLProvider) getUsedFolderQuota(name string) (int, int64, error) {
  322. return sqlCommonGetFolderUsedQuota(name, p.dbHandle)
  323. }
  324. func (p *MySQLProvider) getGroups(limit, offset int, order string, minimal bool) ([]Group, error) {
  325. return sqlCommonGetGroups(limit, offset, order, minimal, p.dbHandle)
  326. }
  327. func (p *MySQLProvider) getGroupsWithNames(names []string) ([]Group, error) {
  328. return sqlCommonGetGroupsWithNames(names, p.dbHandle)
  329. }
  330. func (p *MySQLProvider) getUsersInGroups(names []string) ([]string, error) {
  331. return sqlCommonGetUsersInGroups(names, p.dbHandle)
  332. }
  333. func (p *MySQLProvider) groupExists(name string) (Group, error) {
  334. return sqlCommonGetGroupByName(name, p.dbHandle)
  335. }
  336. func (p *MySQLProvider) addGroup(group *Group) error {
  337. return sqlCommonAddGroup(group, p.dbHandle)
  338. }
  339. func (p *MySQLProvider) updateGroup(group *Group) error {
  340. return sqlCommonUpdateGroup(group, p.dbHandle)
  341. }
  342. func (p *MySQLProvider) deleteGroup(group Group) error {
  343. return sqlCommonDeleteGroup(group, p.dbHandle)
  344. }
  345. func (p *MySQLProvider) dumpGroups() ([]Group, error) {
  346. return sqlCommonDumpGroups(p.dbHandle)
  347. }
  348. func (p *MySQLProvider) adminExists(username string) (Admin, error) {
  349. return sqlCommonGetAdminByUsername(username, p.dbHandle)
  350. }
  351. func (p *MySQLProvider) addAdmin(admin *Admin) error {
  352. return sqlCommonAddAdmin(admin, p.dbHandle)
  353. }
  354. func (p *MySQLProvider) updateAdmin(admin *Admin) error {
  355. return sqlCommonUpdateAdmin(admin, p.dbHandle)
  356. }
  357. func (p *MySQLProvider) deleteAdmin(admin Admin) error {
  358. return sqlCommonDeleteAdmin(admin, p.dbHandle)
  359. }
  360. func (p *MySQLProvider) getAdmins(limit int, offset int, order string) ([]Admin, error) {
  361. return sqlCommonGetAdmins(limit, offset, order, p.dbHandle)
  362. }
  363. func (p *MySQLProvider) dumpAdmins() ([]Admin, error) {
  364. return sqlCommonDumpAdmins(p.dbHandle)
  365. }
  366. func (p *MySQLProvider) validateAdminAndPass(username, password, ip string) (Admin, error) {
  367. return sqlCommonValidateAdminAndPass(username, password, ip, p.dbHandle)
  368. }
  369. func (p *MySQLProvider) apiKeyExists(keyID string) (APIKey, error) {
  370. return sqlCommonGetAPIKeyByID(keyID, p.dbHandle)
  371. }
  372. func (p *MySQLProvider) addAPIKey(apiKey *APIKey) error {
  373. return sqlCommonAddAPIKey(apiKey, p.dbHandle)
  374. }
  375. func (p *MySQLProvider) updateAPIKey(apiKey *APIKey) error {
  376. return sqlCommonUpdateAPIKey(apiKey, p.dbHandle)
  377. }
  378. func (p *MySQLProvider) deleteAPIKey(apiKey APIKey) error {
  379. return sqlCommonDeleteAPIKey(apiKey, p.dbHandle)
  380. }
  381. func (p *MySQLProvider) getAPIKeys(limit int, offset int, order string) ([]APIKey, error) {
  382. return sqlCommonGetAPIKeys(limit, offset, order, p.dbHandle)
  383. }
  384. func (p *MySQLProvider) dumpAPIKeys() ([]APIKey, error) {
  385. return sqlCommonDumpAPIKeys(p.dbHandle)
  386. }
  387. func (p *MySQLProvider) updateAPIKeyLastUse(keyID string) error {
  388. return sqlCommonUpdateAPIKeyLastUse(keyID, p.dbHandle)
  389. }
  390. func (p *MySQLProvider) shareExists(shareID, username string) (Share, error) {
  391. return sqlCommonGetShareByID(shareID, username, p.dbHandle)
  392. }
  393. func (p *MySQLProvider) addShare(share *Share) error {
  394. return sqlCommonAddShare(share, p.dbHandle)
  395. }
  396. func (p *MySQLProvider) updateShare(share *Share) error {
  397. return sqlCommonUpdateShare(share, p.dbHandle)
  398. }
  399. func (p *MySQLProvider) deleteShare(share Share) error {
  400. return sqlCommonDeleteShare(share, p.dbHandle)
  401. }
  402. func (p *MySQLProvider) getShares(limit int, offset int, order, username string) ([]Share, error) {
  403. return sqlCommonGetShares(limit, offset, order, username, p.dbHandle)
  404. }
  405. func (p *MySQLProvider) dumpShares() ([]Share, error) {
  406. return sqlCommonDumpShares(p.dbHandle)
  407. }
  408. func (p *MySQLProvider) updateShareLastUse(shareID string, numTokens int) error {
  409. return sqlCommonUpdateShareLastUse(shareID, numTokens, p.dbHandle)
  410. }
  411. func (p *MySQLProvider) getDefenderHosts(from int64, limit int) ([]DefenderEntry, error) {
  412. return sqlCommonGetDefenderHosts(from, limit, p.dbHandle)
  413. }
  414. func (p *MySQLProvider) getDefenderHostByIP(ip string, from int64) (DefenderEntry, error) {
  415. return sqlCommonGetDefenderHostByIP(ip, from, p.dbHandle)
  416. }
  417. func (p *MySQLProvider) isDefenderHostBanned(ip string) (DefenderEntry, error) {
  418. return sqlCommonIsDefenderHostBanned(ip, p.dbHandle)
  419. }
  420. func (p *MySQLProvider) updateDefenderBanTime(ip string, minutes int) error {
  421. return sqlCommonDefenderIncrementBanTime(ip, minutes, p.dbHandle)
  422. }
  423. func (p *MySQLProvider) deleteDefenderHost(ip string) error {
  424. return sqlCommonDeleteDefenderHost(ip, p.dbHandle)
  425. }
  426. func (p *MySQLProvider) addDefenderEvent(ip string, score int) error {
  427. return sqlCommonAddDefenderHostAndEvent(ip, score, p.dbHandle)
  428. }
  429. func (p *MySQLProvider) setDefenderBanTime(ip string, banTime int64) error {
  430. return sqlCommonSetDefenderBanTime(ip, banTime, p.dbHandle)
  431. }
  432. func (p *MySQLProvider) cleanupDefender(from int64) error {
  433. return sqlCommonDefenderCleanup(from, p.dbHandle)
  434. }
  435. func (p *MySQLProvider) addActiveTransfer(transfer ActiveTransfer) error {
  436. return sqlCommonAddActiveTransfer(transfer, p.dbHandle)
  437. }
  438. func (p *MySQLProvider) updateActiveTransferSizes(ulSize, dlSize, transferID int64, connectionID string) error {
  439. return sqlCommonUpdateActiveTransferSizes(ulSize, dlSize, transferID, connectionID, p.dbHandle)
  440. }
  441. func (p *MySQLProvider) removeActiveTransfer(transferID int64, connectionID string) error {
  442. return sqlCommonRemoveActiveTransfer(transferID, connectionID, p.dbHandle)
  443. }
  444. func (p *MySQLProvider) cleanupActiveTransfers(before time.Time) error {
  445. return sqlCommonCleanupActiveTransfers(before, p.dbHandle)
  446. }
  447. func (p *MySQLProvider) getActiveTransfers(from time.Time) ([]ActiveTransfer, error) {
  448. return sqlCommonGetActiveTransfers(from, p.dbHandle)
  449. }
  450. func (p *MySQLProvider) addSharedSession(session Session) error {
  451. return sqlCommonAddSession(session, p.dbHandle)
  452. }
  453. func (p *MySQLProvider) deleteSharedSession(key string) error {
  454. return sqlCommonDeleteSession(key, p.dbHandle)
  455. }
  456. func (p *MySQLProvider) getSharedSession(key string) (Session, error) {
  457. return sqlCommonGetSession(key, p.dbHandle)
  458. }
  459. func (p *MySQLProvider) cleanupSharedSessions(sessionType SessionType, before int64) error {
  460. return sqlCommonCleanupSessions(sessionType, before, p.dbHandle)
  461. }
  462. func (p *MySQLProvider) close() error {
  463. return p.dbHandle.Close()
  464. }
  465. func (p *MySQLProvider) reloadConfig() error {
  466. return nil
  467. }
  468. // initializeDatabase creates the initial database structure
  469. func (p *MySQLProvider) initializeDatabase() error {
  470. dbVersion, err := sqlCommonGetDatabaseVersion(p.dbHandle, false)
  471. if err == nil && dbVersion.Version > 0 {
  472. return ErrNoInitRequired
  473. }
  474. if errors.Is(err, sql.ErrNoRows) {
  475. return errSchemaVersionEmpty
  476. }
  477. logger.InfoToConsole("creating initial database schema, version 15")
  478. providerLog(logger.LevelInfo, "creating initial database schema, version 15")
  479. initialSQL := strings.ReplaceAll(mysqlInitialSQL, "{{schema_version}}", sqlTableSchemaVersion)
  480. initialSQL = strings.ReplaceAll(initialSQL, "{{admins}}", sqlTableAdmins)
  481. initialSQL = strings.ReplaceAll(initialSQL, "{{folders}}", sqlTableFolders)
  482. initialSQL = strings.ReplaceAll(initialSQL, "{{users}}", sqlTableUsers)
  483. initialSQL = strings.ReplaceAll(initialSQL, "{{folders_mapping}}", sqlTableFoldersMapping)
  484. initialSQL = strings.ReplaceAll(initialSQL, "{{api_keys}}", sqlTableAPIKeys)
  485. initialSQL = strings.ReplaceAll(initialSQL, "{{shares}}", sqlTableShares)
  486. initialSQL = strings.ReplaceAll(initialSQL, "{{defender_events}}", sqlTableDefenderEvents)
  487. initialSQL = strings.ReplaceAll(initialSQL, "{{defender_hosts}}", sqlTableDefenderHosts)
  488. initialSQL = strings.ReplaceAll(initialSQL, "{{prefix}}", config.SQLTablesPrefix)
  489. return sqlCommonExecSQLAndUpdateDBVersion(p.dbHandle, strings.Split(initialSQL, ";"), 15, true)
  490. }
  491. func (p *MySQLProvider) migrateDatabase() error { //nolint:dupl
  492. dbVersion, err := sqlCommonGetDatabaseVersion(p.dbHandle, true)
  493. if err != nil {
  494. return err
  495. }
  496. switch version := dbVersion.Version; {
  497. case version == sqlDatabaseVersion:
  498. providerLog(logger.LevelDebug, "sql database is up to date, current version: %v", version)
  499. return ErrNoInitRequired
  500. case version < 15:
  501. err = fmt.Errorf("database version %v is too old, please see the upgrading docs", version)
  502. providerLog(logger.LevelError, "%v", err)
  503. logger.ErrorToConsole("%v", err)
  504. return err
  505. case version == 15:
  506. return updateMySQLDatabaseFromV15(p.dbHandle)
  507. case version == 16:
  508. return updateMySQLDatabaseFromV16(p.dbHandle)
  509. case version == 17:
  510. return updateMySQLDatabaseFromV17(p.dbHandle)
  511. case version == 18:
  512. return updateMySQLDatabaseFromV18(p.dbHandle)
  513. default:
  514. if version > sqlDatabaseVersion {
  515. providerLog(logger.LevelError, "database version %v is newer than the supported one: %v", version,
  516. sqlDatabaseVersion)
  517. logger.WarnToConsole("database version %v is newer than the supported one: %v", version,
  518. sqlDatabaseVersion)
  519. return nil
  520. }
  521. return fmt.Errorf("database version not handled: %v", version)
  522. }
  523. }
  524. func (p *MySQLProvider) revertDatabase(targetVersion int) error {
  525. dbVersion, err := sqlCommonGetDatabaseVersion(p.dbHandle, true)
  526. if err != nil {
  527. return err
  528. }
  529. if dbVersion.Version == targetVersion {
  530. return errors.New("current version match target version, nothing to do")
  531. }
  532. switch dbVersion.Version {
  533. case 16:
  534. return downgradeMySQLDatabaseFromV16(p.dbHandle)
  535. case 17:
  536. return downgradeMySQLDatabaseFromV17(p.dbHandle)
  537. case 18:
  538. return downgradeMySQLDatabaseFromV18(p.dbHandle)
  539. case 19:
  540. return downgradeMySQLDatabaseFromV19(p.dbHandle)
  541. default:
  542. return fmt.Errorf("database version not handled: %v", dbVersion.Version)
  543. }
  544. }
  545. func (p *MySQLProvider) resetDatabase() error {
  546. sql := sqlReplaceAll(mysqlResetSQL)
  547. return sqlCommonExecSQLAndUpdateDBVersion(p.dbHandle, strings.Split(sql, ";"), 0, false)
  548. }
  549. func updateMySQLDatabaseFromV15(dbHandle *sql.DB) error {
  550. if err := updateMySQLDatabaseFrom15To16(dbHandle); err != nil {
  551. return err
  552. }
  553. return updateMySQLDatabaseFromV16(dbHandle)
  554. }
  555. func updateMySQLDatabaseFromV16(dbHandle *sql.DB) error {
  556. if err := updateMySQLDatabaseFrom16To17(dbHandle); err != nil {
  557. return err
  558. }
  559. return updateMySQLDatabaseFromV17(dbHandle)
  560. }
  561. func updateMySQLDatabaseFromV17(dbHandle *sql.DB) error {
  562. if err := updateMySQLDatabaseFrom17To18(dbHandle); err != nil {
  563. return err
  564. }
  565. return updateMySQLDatabaseFromV18(dbHandle)
  566. }
  567. func updateMySQLDatabaseFromV18(dbHandle *sql.DB) error {
  568. return updateMySQLDatabaseFrom18To19(dbHandle)
  569. }
  570. func downgradeMySQLDatabaseFromV16(dbHandle *sql.DB) error {
  571. return downgradeMySQLDatabaseFrom16To15(dbHandle)
  572. }
  573. func downgradeMySQLDatabaseFromV17(dbHandle *sql.DB) error {
  574. if err := downgradeMySQLDatabaseFrom17To16(dbHandle); err != nil {
  575. return err
  576. }
  577. return downgradeMySQLDatabaseFromV16(dbHandle)
  578. }
  579. func downgradeMySQLDatabaseFromV18(dbHandle *sql.DB) error {
  580. if err := downgradeMySQLDatabaseFrom18To17(dbHandle); err != nil {
  581. return err
  582. }
  583. return downgradeMySQLDatabaseFromV17(dbHandle)
  584. }
  585. func downgradeMySQLDatabaseFromV19(dbHandle *sql.DB) error {
  586. if err := downgradeMySQLDatabaseFrom19To18(dbHandle); err != nil {
  587. return err
  588. }
  589. return downgradeMySQLDatabaseFromV18(dbHandle)
  590. }
  591. func updateMySQLDatabaseFrom15To16(dbHandle *sql.DB) error {
  592. logger.InfoToConsole("updating database version: 15 -> 16")
  593. providerLog(logger.LevelInfo, "updating database version: 15 -> 16")
  594. sql := strings.ReplaceAll(mysqlV16SQL, "{{users}}", sqlTableUsers)
  595. sql = strings.ReplaceAll(sql, "{{active_transfers}}", sqlTableActiveTransfers)
  596. sql = strings.ReplaceAll(sql, "{{prefix}}", config.SQLTablesPrefix)
  597. return sqlCommonExecSQLAndUpdateDBVersion(dbHandle, strings.Split(sql, ";"), 16, true)
  598. }
  599. func updateMySQLDatabaseFrom16To17(dbHandle *sql.DB) error {
  600. logger.InfoToConsole("updating database version: 16 -> 17")
  601. providerLog(logger.LevelInfo, "updating database version: 16 -> 17")
  602. sql := strings.ReplaceAll(mysqlV17SQL, "{{users}}", sqlTableUsers)
  603. sql = strings.ReplaceAll(sql, "{{groups}}", sqlTableGroups)
  604. sql = strings.ReplaceAll(sql, "{{folders}}", sqlTableFolders)
  605. sql = strings.ReplaceAll(sql, "{{folders_mapping}}", sqlTableFoldersMapping)
  606. sql = strings.ReplaceAll(sql, "{{users_folders_mapping}}", sqlTableUsersFoldersMapping)
  607. sql = strings.ReplaceAll(sql, "{{users_groups_mapping}}", sqlTableUsersGroupsMapping)
  608. sql = strings.ReplaceAll(sql, "{{groups_folders_mapping}}", sqlTableGroupsFoldersMapping)
  609. sql = strings.ReplaceAll(sql, "{{prefix}}", config.SQLTablesPrefix)
  610. return sqlCommonExecSQLAndUpdateDBVersion(dbHandle, strings.Split(sql, ";"), 17, true)
  611. }
  612. func updateMySQLDatabaseFrom17To18(dbHandle *sql.DB) error {
  613. logger.InfoToConsole("updating database version: 17 -> 18")
  614. providerLog(logger.LevelInfo, "updating database version: 17 -> 18")
  615. if err := importGCSCredentials(); err != nil {
  616. return err
  617. }
  618. return sqlCommonExecSQLAndUpdateDBVersion(dbHandle, nil, 18, true)
  619. }
  620. func updateMySQLDatabaseFrom18To19(dbHandle *sql.DB) error {
  621. logger.InfoToConsole("updating database version: 18 -> 19")
  622. providerLog(logger.LevelInfo, "updating database version: 18 -> 19")
  623. sql := strings.ReplaceAll(mysqlV19SQL, "{{shared_sessions}}", sqlTableSharedSessions)
  624. sql = strings.ReplaceAll(sql, "{{prefix}}", config.SQLTablesPrefix)
  625. return sqlCommonExecSQLAndUpdateDBVersion(dbHandle, strings.Split(sql, ";"), 19, true)
  626. }
  627. func downgradeMySQLDatabaseFrom16To15(dbHandle *sql.DB) error {
  628. logger.InfoToConsole("downgrading database version: 16 -> 15")
  629. providerLog(logger.LevelInfo, "downgrading database version: 16 -> 15")
  630. sql := strings.ReplaceAll(mysqlV16DownSQL, "{{users}}", sqlTableUsers)
  631. sql = strings.ReplaceAll(sql, "{{active_transfers}}", sqlTableActiveTransfers)
  632. return sqlCommonExecSQLAndUpdateDBVersion(dbHandle, strings.Split(sql, ";"), 15, false)
  633. }
  634. func downgradeMySQLDatabaseFrom17To16(dbHandle *sql.DB) error {
  635. logger.InfoToConsole("downgrading database version: 17 -> 16")
  636. providerLog(logger.LevelInfo, "downgrading database version: 17 -> 16")
  637. sql := strings.ReplaceAll(mysqlV17DownSQL, "{{users}}", sqlTableUsers)
  638. sql = strings.ReplaceAll(sql, "{{groups}}", sqlTableGroups)
  639. sql = strings.ReplaceAll(sql, "{{folders}}", sqlTableFolders)
  640. sql = strings.ReplaceAll(sql, "{{folders_mapping}}", sqlTableFoldersMapping)
  641. sql = strings.ReplaceAll(sql, "{{users_folders_mapping}}", sqlTableUsersFoldersMapping)
  642. sql = strings.ReplaceAll(sql, "{{users_groups_mapping}}", sqlTableUsersGroupsMapping)
  643. sql = strings.ReplaceAll(sql, "{{groups_folders_mapping}}", sqlTableGroupsFoldersMapping)
  644. sql = strings.ReplaceAll(sql, "{{prefix}}", config.SQLTablesPrefix)
  645. return sqlCommonExecSQLAndUpdateDBVersion(dbHandle, strings.Split(sql, ";"), 16, false)
  646. }
  647. func downgradeMySQLDatabaseFrom18To17(dbHandle *sql.DB) error {
  648. logger.InfoToConsole("downgrading database version: 18 -> 17")
  649. providerLog(logger.LevelInfo, "downgrading database version: 18 -> 17")
  650. return sqlCommonExecSQLAndUpdateDBVersion(dbHandle, nil, 17, false)
  651. }
  652. func downgradeMySQLDatabaseFrom19To18(dbHandle *sql.DB) error {
  653. logger.InfoToConsole("downgrading database version: 19 -> 18")
  654. providerLog(logger.LevelInfo, "downgrading database version: 19 -> 18")
  655. sql := strings.ReplaceAll(mysqlV19DownSQL, "{{shared_sessions}}", sqlTableSharedSessions)
  656. return sqlCommonExecSQLAndUpdateDBVersion(dbHandle, []string{sql}, 18, false)
  657. }