tool/src/initdb_user.c

   1 /*
   2  * ail
   3  *
   4  * Copyright (c) 2000 - 2011 Samsung Electronics Co., Ltd. All rights reserved.
   5  *
   6  * Contact: Jayoun Lee <airjany@samsung.com>, Sewook Park <sewook7.park@samsung.com>, Jaeho Lee <jaeho81.lee@samsung.com>
   7  *
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  * http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  *
  20  */
  21
  22
  23
  24 #include <string.h>
  25 #include <stdlib.h>
  26 #include <stdio.h>
  27 #include <dirent.h>
  28 #include <unistd.h>
  29 #include <sys/stat.h>
  30 #include <errno.h>
  31 #include <sys/smack.h>
  32
  33 #include "ail.h"
  34 #include "ail_db.h"
  35 #include "ail_private.h"
  36
  37 #ifdef _E
  38 #undef _E
  39 #endif
  40 #define _E(fmt, arg...) fprintf(stderr, "[AIL_INITDB][E][%s,%d] "fmt"\n", __FUNCTION__, __LINE__, ##arg);
  41
  42 #ifdef _D
  43 #undef _D
  44 #endif
  45 #define _D(fmt, arg...) fprintf(stderr, "[AIL_INITDB][D][%s,%d] "fmt"\n", __FUNCTION__, __LINE__, ##arg);
  46
  47 #define SET_DEFAULT_LABEL(x) \
  48         if(smack_setlabel((x), "*", SMACK_LABEL_ACCESS)) _E("failed chsmack -a \"*\" %s", x) \
  49         else _D("chsmack -a \"*\" %s", x)
  50
  51 static int initdb_user_count_app(void)
  52 {
  53         ail_filter_h filter;
  54         ail_error_e ret;
  55         int total = 0;
  56
  57         ret = ail_filter_new(&filter);
  58         if (ret != AIL_ERROR_OK) {
  59                 return -1;
  60         }
  61
  62         ret = ail_filter_add_bool(filter, AIL_PROP_NODISPLAY_BOOL, false);
  63         if (ret != AIL_ERROR_OK) {
  64                 ail_filter_destroy(filter);
  65                 return -1;
  66         }
  67         ret = ail_filter_count_usr_appinfo(filter, &total, getuid());
  68         if (ret != AIL_ERROR_OK) {
  69                 ail_filter_destroy(filter);
  70                 return -1;
  71         }
  72
  73         ail_filter_destroy(filter);
  74
  75         return total;
  76 }
  77
  78
  79
  80 char* _desktop_to_package(const char* desktop)
  81 {
  82         char *package, *tmp;
  83
  84         retv_if(!desktop, NULL);
  85
  86         package = strdup(desktop);
  87         retv_if(!package, NULL);
  88
  89         tmp = strrchr(package, '.');
  90         if(tmp == NULL) {
  91                 _E("[%s] is not a desktop file", package);
  92                 free(package);
  93                 return NULL;
  94         }
  95
  96         if (strcmp(tmp, ".desktop")) {
  97                 _E("%s is not a desktop file", desktop);
  98                 free(package);
  99                 return NULL;
 100         }
 101
 102         *tmp = '\0';
 103
 104         return package;
 105 }
 106
 107
 108
 109 int initdb_user_load_directory(const char *directory)
 110 {
 111         DIR *dir;
 112         struct dirent entry, *result;
 113         int len, ret;
 114         char buf[BUFSZE];
 115         int total_cnt = 0;
 116         int ok_cnt = 0;
 117
 118         // desktop file
 119         dir = opendir(directory);
 120         if (!dir) {
 121                 if (strerror_r(errno, buf, sizeof(buf)) == 0)
 122                         _E("Failed to access the [%s] because %s\n", directory, buf);
 123                 return AIL_ERROR_FAIL;
 124         }
 125
 126         len = strlen(directory) + 1;
 127         _D("Loading desktop files from %s", directory);
 128
 129         for (ret = readdir_r(dir, &entry, &result);
 130                         ret == 0 && result != NULL;
 131                         ret = readdir_r(dir, &entry, &result)) {
 132                 char *package;
 133
 134                 if (entry.d_name[0] == '.') continue;
 135                 total_cnt++;
 136                 package = _desktop_to_package(entry.d_name);
 137                 if (!package) {
 138                         _E("Failed to convert file to package[%s]", entry.d_name);
 139                         continue;
 140                 }
 141
 142                 if (ail_usr_desktop_add(package, getuid()) != AIL_ERROR_OK) {
 143                         _E("Failed to add a package[%s]", package);
 144                 } else {
 145                         ok_cnt++;
 146                 }
 147                 free(package);
 148         }
 149
 150         _D("Application-Desktop process : Success [%d], fail[%d], total[%d] \n", ok_cnt, total_cnt-ok_cnt, total_cnt);
 151         closedir(dir);
 152
 153         return AIL_ERROR_OK;
 154 }
 155
 156 static int __is_authorized()
 157 {
 158         /* ail_init db should be called by an user. */
 159         uid_t uid = getuid();
 160         if ((uid_t)OWNER_ROOT != uid)
 161                 return 1;
 162         else
 163                 return 0;
 164 }
 165
 166 int main(int argc, char *argv[])
 167 {
 168         int ret;
 169         char *db;
 170         char *path;
 171
 172         if (!__is_authorized()) {
 173                 fprintf(stderr, "You are not an authorized user!\n");
 174                 _D("You are root user! Please switch to a regular user\n");
 175                 return -1;
 176         }
 177
 178         db = ail_get_app_DB(getuid());
 179         if (db) {
 180                 if (remove(db))
 181                         _E(" %s is not removed", db);
 182
 183                 free(db);
 184         }
 185
 186         db = ail_get_app_DB_journal(getuid());
 187         if (db) {
 188                 if (remove(db))
 189                         _E(" %s is not removed", db);
 190
 191                 free(db);
 192         }
 193
 194         ret = setenv("AIL_INITDB", "1", 1);
 195         _D("AIL_INITDB : %d", ret);
 196
 197         ret = initdb_user_count_app();
 198         if (ret > 0)
 199                 _D("Some Apps in the App Info DB.");
 200
 201         path = ail_get_desktop_path(getuid());
 202         if (path == NULL) {
 203                 _E("Failed to get desktop path");
 204                 return -1;
 205         }
 206
 207         ret = initdb_user_load_directory(path);
 208         if (ret == AIL_ERROR_FAIL)
 209                 _E("cannot load usr desktop directory.");
 210
 211         free(path);
 212
 213         return AIL_ERROR_OK;
 214 }