mirror of
https://github.com/joel16/VitaShell.git
synced 2024-11-23 03:39:39 +00:00
7f2d1134d9
A new menu feature "Refresh license database" is now available from the top context menu. Its action is to parse all the .rif files under ux0:licence/app and ux0:licence/addcont, and import them to the ux0:licence/license.db SQLite database (which gets created it it doesn't already exists). Used in conjunction with the new App/DLC refresh feature, this allows Vita users to restore their content straight from PKG downloads, and let VitaShell do the job of also restoring the relevant license from the DB backup. This also fixes a small typo in LANGUAGE_CONTAINER_SIZE and adds a .editorconfig, to ensure that the indentation settings of VitaShell are followed by modern code editors.
112 lines
2.8 KiB
C
112 lines
2.8 KiB
C
/*
|
|
VitaShell - RIF and license.db handling
|
|
Copyright (C) 2017 VitaSmith
|
|
|
|
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/>.
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "rif.h"
|
|
#include "sqlite3.h"
|
|
|
|
#define MAX_QUERY_LENGTH 128
|
|
|
|
int create_db(const char* db_path, const char* schema)
|
|
{
|
|
int rc;
|
|
sqlite3 *db = NULL;
|
|
|
|
rc = sqlite3_open_v2(db_path, &db, SQLITE_OPEN_CREATE | SQLITE_OPEN_READWRITE, NULL);
|
|
if (rc != SQLITE_OK)
|
|
goto out;
|
|
rc = sqlite3_exec(db, schema, NULL, NULL, NULL);
|
|
if (rc != SQLITE_OK)
|
|
goto out;
|
|
|
|
out:
|
|
sqlite3_close(db);
|
|
return rc;
|
|
}
|
|
|
|
// Insert a new RIF into the DB
|
|
int insert_rif(const char* db_path, const uint8_t* rif)
|
|
{
|
|
int rc;
|
|
sqlite3 *db = NULL;
|
|
sqlite3_stmt *stmt;
|
|
char query[MAX_QUERY_LENGTH];
|
|
|
|
rc = sqlite3_open_v2(db_path, &db, SQLITE_OPEN_READWRITE, NULL);
|
|
if (rc != SQLITE_OK)
|
|
goto out;
|
|
|
|
snprintf(query, sizeof(query), "INSERT INTO Licenses VALUES('%s', ?)", &rif[0x10]);
|
|
|
|
rc = sqlite3_prepare_v2(db, query, -1, &stmt, NULL);
|
|
if (rc != SQLITE_OK)
|
|
goto out;
|
|
rc = sqlite3_bind_blob(stmt, 1, rif, RIF_SIZE, SQLITE_STATIC);
|
|
if (rc != SQLITE_OK)
|
|
goto out;
|
|
rc = sqlite3_step(stmt);
|
|
if (rc != SQLITE_DONE)
|
|
goto out;
|
|
rc = sqlite3_finalize(stmt);
|
|
|
|
out:
|
|
sqlite3_close(db);
|
|
return rc;
|
|
}
|
|
|
|
// Query a RIF from the license database. Returned data must be freed by the caller.
|
|
uint8_t* query_rif(const char* db_path, const char* content_id)
|
|
{
|
|
int rc, rif_size = 0;
|
|
sqlite3 *db = NULL;
|
|
sqlite3_stmt *stmt;
|
|
char query[MAX_QUERY_LENGTH];
|
|
uint8_t *rif = NULL;
|
|
const uint8_t *db_rif;
|
|
|
|
rc = sqlite3_open_v2(db_path, &db, SQLITE_OPEN_READONLY, NULL);
|
|
if (rc != SQLITE_OK)
|
|
goto out;
|
|
|
|
snprintf(query, sizeof(query), "SELECT RIF FROM Licenses WHERE CONTENT_ID = '%s'", content_id);
|
|
|
|
rc = sqlite3_prepare_v2(db, query, -1, &stmt, NULL);
|
|
if (rc != SQLITE_OK)
|
|
goto out;
|
|
|
|
rc = sqlite3_step(stmt);
|
|
if (rc == SQLITE_ROW)
|
|
rif_size = sqlite3_column_bytes(stmt, 0);
|
|
if (rif_size != RIF_SIZE)
|
|
goto out;
|
|
|
|
rif = malloc(rif_size);
|
|
db_rif = sqlite3_column_blob(stmt, 0);
|
|
|
|
if ((rif != NULL) && (db_rif != NULL))
|
|
memcpy(rif, db_rif, rif_size);
|
|
rc = sqlite3_finalize(stmt);
|
|
|
|
out:
|
|
sqlite3_close(db);
|
|
return rif;
|
|
}
|