src/queue.c

   1 /*
   2  * Copyright (c) 2016 Samsung Electronics Co., Ltd. All rights reserved.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  *
  16  */
  17
  18 #define _XOPEN_SOURCE 500
  19 #include <ftw.h>
  20 #include <stdio.h>
  21 #include <stdlib.h>
  22 #include <string.h>
  23 #include <strings.h>
  24 #include <unistd.h>
  25 #include <sys/stat.h>
  26 #include <sys/types.h>
  27
  28 #include <glib.h>
  29
  30 #include "pkgmgr-server.h"
  31 #include "queue.h"
  32
  33 #define BACKEND_DIR "/etc/package-manager/backend"
  34 #define NOPENFD 20
  35
  36 struct backend_queue {
  37         char *path;
  38         char *type;
  39         int slot;
  40         GList *job_list;
  41 };
  42
  43 static GHashTable *queue_type_table;
  44 static GHashTable *queue_slot_table;
  45
  46 int num_of_backends;
  47
  48 int _is_queue_empty(int pos)
  49 {
  50         struct backend_queue *queue;
  51
  52         queue = (struct backend_queue *)g_hash_table_lookup(queue_slot_table,
  53                         (gconstpointer)pos);
  54         if (queue == NULL) {
  55                 ERR("cannot find queue of slot %d", pos);
  56                 return -1;
  57         }
  58
  59         if (queue->job_list == NULL)
  60                 return 1;
  61
  62         return 0;
  63 }
  64
  65 struct backend_job *_pop_queue(int pos)
  66 {
  67         struct backend_queue *queue;
  68         struct backend_job *job;
  69         GList *tmp;
  70
  71         queue = (struct backend_queue *)g_hash_table_lookup(queue_slot_table,
  72                         (gconstpointer)pos);
  73         if (queue == NULL) {
  74                 ERR("cannot find queue of slot %d", pos);
  75                 return NULL;
  76         }
  77
  78         tmp = g_list_first(queue->job_list);
  79         if (tmp == NULL)
  80                 return NULL;
  81         job = (struct backend_job *)tmp->data;
  82         queue->job_list = g_list_delete_link(queue->job_list, tmp);
  83
  84         return job;
  85 }
  86
  87 int _push_queue(uid_t target_uid, uid_t caller_uid, const char *req_id,
  88                 int req_type, const char *queue_type, const char *pkgid,
  89                 const char *args)
  90 {
  91         struct backend_queue *queue;
  92         struct backend_job *job;
  93
  94         queue = (struct backend_queue *)g_hash_table_lookup(queue_type_table,
  95                         (gconstpointer)queue_type);
  96         if (queue == NULL) {
  97                 ERR("no such queue: %s", queue_type);
  98                 return -1;
  99         }
 100
 101         job = calloc(1, sizeof(struct backend_job));
 102         job->target_uid = target_uid;
 103         job->caller_uid = caller_uid;
 104         if (req_id)
 105                 job->req_id = strdup(req_id);
 106         job->req_type = req_type;
 107         if (pkgid)
 108                 job->pkgid = strdup(pkgid);
 109         if (args)
 110                 job->args = strdup(args);
 111         /* just assign pointer value */
 112         job->backend_slot = queue->slot;
 113         job->backend_type = queue->type;
 114         job->backend_path = queue->path;
 115
 116         queue->job_list = g_list_append(queue->job_list, (gpointer)job);
 117
 118         return 0;
 119 }
 120
 121 static int __init_backends(const char *fpath, const struct stat *sb,
 122                 int typeflag, struct FTW *ftwbuf)
 123 {
 124         int r;
 125         struct backend_queue *queue;
 126
 127         if (typeflag != FTW_F && typeflag != FTW_SL)
 128                 return 0;
 129
 130         queue = calloc(1, sizeof(struct backend_queue));
 131         if (typeflag == FTW_F) {
 132                 queue->path = strdup(fpath);
 133         } else if (typeflag == FTW_SL) {
 134                 queue->path = malloc(sb->st_size + 1);
 135                 r = readlink(fpath, queue->path, sb->st_size + 1);
 136                 if (r < 0 || r > sb->st_size) {
 137                         ERR("failed to readlink for %s", fpath);
 138                         free(queue->path);
 139                         free(queue);
 140                         return -1;
 141                 }
 142                 queue->path[r] = '\0';
 143         }
 144         queue->type = strdup(fpath + ftwbuf->base);
 145         queue->slot = num_of_backends++;
 146         g_hash_table_insert(queue_type_table, (gpointer)queue->type,
 147                         (gpointer)queue);
 148         g_hash_table_insert(queue_slot_table, (gpointer)queue->slot,
 149                         (gpointer)queue);
 150
 151         DBG("backend detected: %s(%s)", queue->type, queue->path);
 152
 153         return 0;
 154 }
 155
 156 static gboolean __str_equal(gconstpointer v1, gconstpointer v2)
 157 {
 158         const char *str1 = (const char *)v1;
 159         const char *str2 = (const char *)v2;
 160
 161         /* ignore case */
 162         return strcasecmp(str1, str2) == 0;
 163 }
 164
 165 void __free_extra_info(struct backend_job *job)
 166 {
 167         if (job->extra) {
 168                 if (job->extra->getsize_io)
 169                         g_io_channel_unref(job->extra->getsize_io);
 170                 if (job->extra->getsize_fd)
 171                         close(job->extra->getsize_fd);
 172                 if (job->extra->getsize_fifo) {
 173                         unlink(job->extra->getsize_fifo);
 174                         free(job->extra->getsize_fifo);
 175                 }
 176                 free(job->extra);
 177                 job->extra = NULL;
 178         }
 179 }
 180
 181 void _free_backend_job(struct backend_job *job)
 182 {
 183         __free_extra_info(job);
 184         free(job->req_id);
 185         free(job->pkgid);
 186         free(job->appid);
 187         free(job->args);
 188         free(job);
 189 }
 190
 191 static void __free_backend_queue(gpointer data)
 192 {
 193         struct backend_queue *queue = (struct backend_queue *)data;
 194
 195         g_list_free_full(queue->job_list, (GDestroyNotify)_free_backend_job);
 196         free(queue->path);
 197         free(queue->type);
 198         free(queue);
 199 }
 200
 201 void _fini_backend_queue(void)
 202 {
 203         g_hash_table_destroy(queue_slot_table);
 204         g_hash_table_destroy(queue_type_table);
 205 }
 206
 207 int _init_backend_queue(void)
 208 {
 209         queue_type_table = g_hash_table_new_full(g_str_hash, __str_equal,
 210                         NULL, __free_backend_queue);
 211         queue_slot_table = g_hash_table_new(g_direct_hash, g_direct_equal);
 212
 213         if (nftw(BACKEND_DIR, __init_backends, NOPENFD, FTW_PHYS))
 214                 return -1;
 215
 216         DBG("number of backends: %d", num_of_backends);
 217
 218         return 0;
 219 }