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
  38 #ifdef _E
  39 #undef _E
  40 #endif
  41 #define _E(fmt, arg...) fprintf(stderr, "[AIL_INITDB][E][%s,%d] "fmt"\n", __FUNCTION__, __LINE__, ##arg);
  42
  43 #ifdef _D
  44 #undef _D
  45 #endif
  46 #define _D(fmt, arg...) fprintf(stderr, "[AIL_INITDB][D][%s,%d] "fmt"\n", __FUNCTION__, __LINE__, ##arg);
  47
  48 #define SET_DEFAULT_LABEL(x) \
  49         if(smack_setlabel((x), "*", SMACK_LABEL_ACCESS)) _E("failed chsmack -a \"*\" %s", x) \
  50         else _D("chsmack -a \"*\" %s", x)
  51
  52 static int initdb_user_count_app(void)
  53 {
  54         ail_filter_h filter;
  55         ail_error_e ret;
  56         int total = 0;
  57
  58         ret = ail_filter_new(&filter);
  59         if (ret != AIL_ERROR_OK) {
  60                 return -1;
  61         }
  62
  63         ret = ail_filter_add_bool(filter, AIL_PROP_NODISPLAY_BOOL, false);
  64         if (ret != AIL_ERROR_OK) {
  65                 ail_filter_destroy(filter);
  66                 return -1;
  67         }
  68         ret = ail_filter_count_usr_appinfo(filter, &total, getuid());
  69         if (ret != AIL_ERROR_OK) {
  70                 ail_filter_destroy(filter);
  71                 return -1;
  72         }
  73
  74         ail_filter_destroy(filter);
  75
  76         return total;
  77 }
  78
  79
  80
  81 char* _desktop_to_package(const char* desktop)
  82 {
  83         char *package, *tmp;
  84
  85         retv_if(!desktop, NULL);
  86
  87         package = strdup(desktop);
  88         retv_if(!package, NULL);
  89
  90         tmp = strrchr(package, '.');
  91         if(tmp == NULL) {
  92                 _E("[%s] is not a desktop file", package);
  93                 free(package);
  94                 return NULL;
  95         }
  96
  97         if (strcmp(tmp, ".desktop")) {
  98                 _E("%s is not a desktop file", desktop);
  99                 free(package);
 100                 return NULL;
 101         }
 102
 103         *tmp = '\0';
 104
 105         return package;
 106 }
 107
 108
 109
 110 int initdb_user_load_directory(const char *directory)
 111 {
 112         DIR *dir;
 113         struct dirent entry, *result;
 114         int len, ret;
 115         char buf[BUFSZE];
 116         int total_cnt = 0;
 117         int ok_cnt = 0;
 118
 119         // desktop file
 120         dir = opendir(directory);
 121         if (!dir) {
 122                 if (strerror_r(errno, buf, sizeof(buf)) == 0)
 123                         _E("Failed to access the [%s] because %s\n", directory, buf);
 124                 return AIL_ERROR_FAIL;
 125         }
 126
 127         len = strlen(directory) + 1;
 128         _D("Loading desktop files from %s", directory);
 129
 130         for (ret = readdir_r(dir, &entry, &result);
 131                         ret == 0 && result != NULL;
 132                         ret = readdir_r(dir, &entry, &result)) {
 133                 char *package;
 134
 135                 if (entry.d_name[0] == '.') continue;
 136                 total_cnt++;
 137                 package = _desktop_to_package(entry.d_name);
 138                 if (!package) {
 139                         _E("Failed to convert file to package[%s]", entry.d_name);
 140                         continue;
 141                 }
 142
 143                 if (ail_usr_desktop_add(package, getuid()) != AIL_ERROR_OK) {
 144                         _E("Failed to add a package[%s]", package);
 145                 } else {
 146                         ok_cnt++;
 147                 }
 148                 free(package);
 149         }
 150
 151         _D("Application-Desktop process : Success [%d], fail[%d], total[%d] \n", ok_cnt, total_cnt-ok_cnt, total_cnt);
 152         closedir(dir);
 153
 154         return AIL_ERROR_OK;
 155 }
 156
 157 static int __is_authorized()
 158 {
 159         /* ail_init db should be called by an user. */
 160
 161         uid_t uid = getuid();
 162         if ((uid_t) OWNER_ROOT != uid)
 163                 return 1;
 164         else
 165                 return 0;
 166 }
 167
 168
 169 int main(int argc, char *argv[])
 170 {
 171         int ret;
 172
 173         if (!__is_authorized()) {
 174                 fprintf(stderr, "You are not an authorized user!\n");
 175                 _D("You are root user! Please switch to a regular user\n");
 176         }
 177         else {
 178                 if(remove(ail_get_app_DB(getuid())))
 179                         _E(" %s is not removed", ail_get_app_DB(getuid()));
 180                 if(remove(ail_get_app_DB_journal(getuid())))
 181                         _E(" %s is not removed", ail_get_app_DB_journal(getuid()));
 182         }
 183         ret = setenv("AIL_INITDB", "1", 1);
 184         _D("AIL_INITDB : %d", ret);
 185         ret = initdb_user_count_app();
 186         if (ret > 0) {
 187                 _D("Some Apps in the App Info DB.");
 188         }
 189
 190         ret = initdb_user_load_directory(ail_get_desktop_path(getuid()));
 191         if (ret == AIL_ERROR_FAIL) {
 192                 _E("cannot load usr desktop directory.");
 193         }
 194
 195         return AIL_ERROR_OK;
 196 }
 197
 198
 199
 200 // END