391 lines
9.7 KiB
C
391 lines
9.7 KiB
C
#include "../../include/package.h"
|
|
#include "../../include/error.h"
|
|
#include "../../include/pool.h"
|
|
#include "../../include/util.h"
|
|
#include "../../include/ctx.h"
|
|
|
|
#include <spawn.h>
|
|
#include <wait.h>
|
|
#include <archive.h>
|
|
#include <archive_entry.h>
|
|
#include <errno.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/stat.h>
|
|
#include <unistd.h>
|
|
|
|
bool __lm_ctx_save_install(lm_ctx_t *ctx, lm_pkg_t *pkg, char *install_path){
|
|
if(NULL == ctx->data){
|
|
lm_error_set(LM_ERR_CtxDataNULL);
|
|
return false;
|
|
}
|
|
|
|
if(!exists(install_path, NULL) || is_empty(install_path))
|
|
return true;
|
|
|
|
char script_name[strlen(pkg->data.name)+15];
|
|
char script_dir[strlen(ctx->data)+15];
|
|
|
|
sprintf(script_name, "install_%s", pkg->data.name);
|
|
join(script_dir, ctx->data, "scripts");
|
|
|
|
if(!mkdir_ifnot(script_dir, 0755)){
|
|
lm_error_set(LM_ERR_InstallDirFail);
|
|
return false;
|
|
}
|
|
|
|
char dst[sizeof(script_name)+sizeof(script_dir)+10];
|
|
join(dst, script_dir, script_name);
|
|
|
|
return copy_file(dst, install_path);
|
|
}
|
|
|
|
bool __lm_ctx_run_install(char *root, char *install_path) {
|
|
if(!exists(install_path, NULL) || is_empty(install_path))
|
|
return true;
|
|
|
|
char *args[] = {"/bin/bash", install_path, NULL}, *oldpwd = NULL;
|
|
bool ret = false;
|
|
int status = 0;
|
|
pid_t pid = 0;
|
|
|
|
if((oldpwd = getcwd(NULL, 0)) == NULL){
|
|
lm_error_set(LM_ERR_InstallCwdFail);
|
|
goto end;
|
|
}
|
|
|
|
if(chdir(root) < 0){
|
|
lm_error_set(LM_ERR_InstallRootChdirFail);
|
|
goto end;
|
|
}
|
|
|
|
if (posix_spawn(&pid, "/bin/bash", NULL, NULL, args, NULL) != 0) {
|
|
lm_error_set(LM_ERR_InstallSpawnFail);
|
|
goto end;
|
|
}
|
|
|
|
waitpid(pid, &status, 0);
|
|
|
|
if(chdir(root) < 0){
|
|
lm_error_set(LM_ERR_InstallBackChdirFail);
|
|
goto end;
|
|
}
|
|
|
|
if(status != 0){
|
|
lm_error_set(LM_ERR_InstallStatusFail);
|
|
goto end;
|
|
}
|
|
|
|
ret = true;
|
|
end:
|
|
free(oldpwd);
|
|
return ret;
|
|
}
|
|
|
|
bool __lm_ctx_extract_files(lm_ctx_t *ctx, lm_pkg_t *pkg, char *files, lm_ctx_install_callback_t callback, void *data){
|
|
struct archive *reader = NULL, *writer = NULL;
|
|
struct archive_entry *entry = NULL;
|
|
int flags = 0, rc = 0;
|
|
char *oldpwd = NULL, *entry_path = NULL;
|
|
bool ret = false;
|
|
size_t total = 0, current = 0;
|
|
mode_t type = 0;
|
|
struct stat st;
|
|
|
|
if(stat(files, &st) < 0){
|
|
lm_error_set(LM_ERR_ExtractStatFail);
|
|
goto end;
|
|
}
|
|
|
|
oldpwd = getcwd(NULL, 0);
|
|
total = st.st_size;
|
|
|
|
if (NULL == oldpwd) {
|
|
lm_error_set(LM_ERR_GetCwdFail);
|
|
goto end;
|
|
}
|
|
|
|
if(chdir(ctx->root) < 0){
|
|
lm_error_set(LM_ERR_ExtractRootChdirFail);
|
|
goto end;
|
|
}
|
|
|
|
flags = ARCHIVE_EXTRACT_PERM;
|
|
flags |= ARCHIVE_EXTRACT_UNLINK;
|
|
|
|
reader = archive_read_new();
|
|
writer = archive_write_disk_new();
|
|
|
|
if (NULL == reader || NULL == writer) {
|
|
lm_error_set(LM_ERR_ArcNewFail);
|
|
goto end;
|
|
}
|
|
|
|
archive_write_disk_set_options(writer, flags);
|
|
archive_write_disk_set_standard_lookup(writer);
|
|
|
|
archive_read_support_format_tar(reader);
|
|
archive_read_support_filter_gzip(reader);
|
|
|
|
if (archive_read_open_filename(reader, files, 1024 * 10) != ARCHIVE_OK) {
|
|
lm_error_set(LM_ERR_ArcOpenFail);
|
|
goto end;
|
|
}
|
|
|
|
while ((rc = archive_read_next_header(reader, &entry)) == ARCHIVE_OK) {
|
|
entry_path = (char *)archive_entry_pathname(entry);
|
|
current += archive_entry_size(entry);
|
|
type = archive_entry_filetype(entry);
|
|
|
|
if(exists(entry_path, NULL) && lm_package_data_keep_contains(&pkg->data, entry_path)){
|
|
pdebug(__func__, "not extracting %s, file is set as KEEP", entry_path);
|
|
continue;
|
|
}
|
|
|
|
switch (type) {
|
|
case AE_IFLNK:
|
|
if(!lm_database_files_add(ctx->db, &pkg->data, "################################", entry_path)){
|
|
char *suberr = lm_strerror_dup();
|
|
pdebug(__func__, "failed to add link to the database for %s: %s", pkg->data.name, suberr);
|
|
lm_error_set(LM_ERR_PkgFilesAddFail, entry_path, suberr);
|
|
free(suberr);
|
|
goto end;
|
|
}
|
|
break;
|
|
|
|
case AE_IFREG:
|
|
if(!lm_database_files_contains(ctx->db, &pkg->data, entry_path)){
|
|
pdebug(__func__, "archive file not included in the database: %s (%s)", entry_path, pkg->data.name);
|
|
goto next;
|
|
}
|
|
break;
|
|
}
|
|
|
|
rc = archive_write_header(writer, entry);
|
|
if (rc != ARCHIVE_OK) {
|
|
lm_error_set(LM_ERR_ArcWHeaderFail);
|
|
goto end;
|
|
}
|
|
|
|
if (!copy_blocks(writer, reader))
|
|
goto end;
|
|
|
|
rc = archive_write_finish_entry(writer);
|
|
if (rc != ARCHIVE_OK) {
|
|
lm_error_set(LM_ERR_ArcWEntryFail);
|
|
goto end;
|
|
}
|
|
|
|
if(NULL != callback)
|
|
if(!callback(ctx, pkg, entry_path, current, total, data))
|
|
goto end;
|
|
|
|
next:
|
|
continue;
|
|
}
|
|
|
|
if (rc != ARCHIVE_EOF) {
|
|
lm_error_set(LM_ERR_ArcNextHeaderFail);
|
|
goto end;
|
|
}
|
|
|
|
ret = true;
|
|
|
|
end:
|
|
if (NULL != reader) {
|
|
archive_read_close(reader);
|
|
archive_read_free(reader);
|
|
}
|
|
|
|
if (NULL != writer) {
|
|
archive_write_close(writer);
|
|
archive_write_free(writer);
|
|
}
|
|
|
|
if (NULL != oldpwd) {
|
|
if(chdir(oldpwd) < 0){
|
|
lm_error_set(LM_ERR_ExtractOldChdirFail);
|
|
ret = false;
|
|
}
|
|
free(oldpwd);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
bool lm_ctx_install(lm_ctx_t *ctx, lm_pkg_t *pkg, bool run_install, lm_ctx_install_callback_t callback, void *data) {
|
|
if(NULL == ctx || NULL == pkg){
|
|
lm_error_set(LM_ERR_ArgNULL);
|
|
return false;
|
|
}
|
|
|
|
if(NULL == ctx->temp){
|
|
lm_error_set(LM_ERR_CtxTempNULL);
|
|
return false;
|
|
}
|
|
|
|
if(NULL == ctx->root){
|
|
lm_error_set(LM_ERR_CtxRootNULL);
|
|
return false;
|
|
}
|
|
|
|
if(!lm_ctx_temp_clear(ctx))
|
|
return false; // error set by function
|
|
|
|
if(!lm_ctx_database_init(ctx))
|
|
return false; // error set by function
|
|
|
|
if(lm_ctx_database_is_installed(ctx, pkg->data.name, NULL)){
|
|
lm_error_set(LM_ERR_PkgAlreadyInstalled);
|
|
return false;
|
|
}
|
|
|
|
if(lm_package_path_is_empty(pkg)){
|
|
lm_error_set(LM_ERR_PkgPathsEmpty);
|
|
return false;
|
|
}
|
|
|
|
lm_pkg_files_t *files;
|
|
char *line = NULL, *hash = NULL, *file = NULL;
|
|
ssize_t line_len = 0, file_len = 0;
|
|
FILE *hashes = NULL;
|
|
bool ret = false;
|
|
lm_pkg_t temp;
|
|
|
|
lm_ctx_temp_clear(ctx);
|
|
lm_package_init(&temp);
|
|
|
|
if((files = lm_package_extract(pkg, ctx->temp)) == NULL){
|
|
pdebug(__func__, "failed to extract %s: %s", pkg->data.name, lm_strerror());
|
|
goto end;
|
|
}
|
|
|
|
if(!lm_package_data_load(&temp.data, files->data_file))
|
|
goto end; // error set by function
|
|
|
|
if(!lm_package_is_same(&temp, pkg)){
|
|
pdebug(__func__, "DATA file does not match with stored %s data", pkg->data.name);
|
|
lm_error_set(LM_ERR_PkgDataNotMatch);
|
|
goto end;
|
|
}
|
|
|
|
pdebug(__func__, "updating changes file for %s", pkg->data.name);
|
|
|
|
if(!lm_database_changes_update(ctx->db, &pkg->data, files->changes_file)){
|
|
char *suberr = lm_strerror_dup();
|
|
pdebug(__func__, "failed to update changes file for %s: %s", pkg->data.name, suberr);
|
|
lm_error_set(LM_ERR_PkgChangesUpdateFail, suberr);
|
|
goto end;
|
|
}
|
|
|
|
if((hashes = fopen(files->hashes_file, "r")) == NULL){
|
|
pdebug(__func__, "failed to open hash file for %s", pkg->data.name);
|
|
lm_error_set(LM_ERR_PkgHashesOpenFail);
|
|
goto end;
|
|
}
|
|
|
|
pdebug(__func__, "deleting file list for %s", pkg->data.name);
|
|
|
|
if(!lm_database_files_del_all(ctx->db, &pkg->data)){
|
|
pdebug(__func__, "failed to remove file list for %s: %s", pkg->data.name, lm_error());
|
|
goto end;
|
|
}
|
|
|
|
while((line_len = getline(&line, (size_t*)&line_len, hashes)) > 0){
|
|
if(NULL == line)
|
|
goto next;
|
|
|
|
if(HASH_LEN+2 >= line_len)
|
|
goto next;
|
|
|
|
line[HASH_LEN] = 0;
|
|
hash = line;
|
|
|
|
file = line+HASH_LEN+2;
|
|
file_len = strlen(file);
|
|
file[file_len-1] = 0;
|
|
|
|
if(file[0] == '.')
|
|
file++;
|
|
|
|
if(file_len >= 2 && file[0] == '/')
|
|
file++;
|
|
|
|
pdebug(__func__, "(%lu) %s => %s", line_len, file, hash);
|
|
|
|
if(!lm_database_files_add(ctx->db, &pkg->data, hash, file)){
|
|
char *suberr = lm_strerror_dup();
|
|
pdebug(__func__, "failed to add file to the database for %s: %s", pkg->data.name, suberr);
|
|
lm_error_set(LM_ERR_PkgFilesAddFail, file, suberr);
|
|
free(suberr);
|
|
goto end;
|
|
}
|
|
|
|
next:
|
|
free(line);
|
|
line = NULL;
|
|
line_len = 0;
|
|
}
|
|
|
|
pdebug(__func__, "extracting files for %s", pkg->data.name);
|
|
|
|
if(!__lm_ctx_extract_files(ctx, pkg, files->files_archive, callback, data)){
|
|
char *suberr = lm_strerror_dup();
|
|
pdebug(__func__, "failed to extract the files archive for %s: %s", pkg->data.name, suberr);
|
|
lm_error_set(LM_ERR_PkgExtractFilesFail, suberr);
|
|
free(suberr);
|
|
goto end;
|
|
}
|
|
|
|
pdebug(__func__, "adding an entry for %s", pkg->data.name);
|
|
|
|
if(!lm_database_entry_add(ctx->db, &pkg->data)){
|
|
char *suberr = lm_strerror_dup();
|
|
pdebug(__func__, "failed to add %s to the database: %s", pkg->data.name, lm_strerror());
|
|
lm_error_set(LM_ERR_PkgDatabaseAddFail, suberr);
|
|
free(suberr);
|
|
goto end;
|
|
}
|
|
|
|
if(run_install)
|
|
pdebug(__func__, "running the install script for %s", pkg->data.name);
|
|
else
|
|
pdebug(__func__, "saving the install script for %s", pkg->data.name);
|
|
|
|
if(run_install && !__lm_ctx_run_install(ctx->root, files->install_file)){
|
|
char *suberr = lm_strerror_dup();
|
|
pdebug(__func__, "failed to run install script: %s", lm_strerror());
|
|
lm_error_set(LM_ERR_InstallRunFail, suberr);
|
|
free(suberr);
|
|
goto end;
|
|
}
|
|
|
|
if(!run_install && !__lm_ctx_save_install(ctx, pkg, files->install_file)){
|
|
char *suberr = lm_strerror_dup();
|
|
pdebug(__func__, "failed to save install script: %s", lm_strerror());
|
|
lm_error_set(LM_ERR_InstallSaveFail, suberr);
|
|
free(suberr);
|
|
goto end;
|
|
}
|
|
|
|
ret = true;
|
|
end:
|
|
free(line);
|
|
|
|
if(NULL != hashes)
|
|
fclose(hashes);
|
|
|
|
lm_package_free(&temp);
|
|
lm_package_extract_free(files);
|
|
|
|
if(!ret){
|
|
lm_database_entry_del(ctx->db, &pkg->data);
|
|
lm_database_changes_del(ctx->db, &pkg->data);
|
|
lm_database_files_del(ctx->db, &pkg->data);
|
|
}
|
|
|
|
unlink(pkg->archive);
|
|
unlink(pkg->signature);
|
|
return ret;
|
|
}
|