Fix memory leak
[platform/core/appfw/slp-pkgmgr.git] / client / src / pkgmgr_client_connection.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 #include <stdio.h>
19 #include <stdlib.h>
20 #include <sys/types.h>
21 #include <unistd.h>
22
23 #include <glib.h>
24 #include <gio/gio.h>
25
26 #include "package-manager.h"
27 #include "pkgmgr_client_debug.h"
28 #include "pkgmgr_client_internal.h"
29 #include "../../installer/pkgmgr_installer.h"
30 #include "../../installer/pkgmgr_installer_config.h"
31
32 #define CONNECTION_RETRY_MAX 5
33 #define CONNECTION_WAIT_USEC (1000000 / 2) /* 0.5 sec */
34 #define CONNECTION_TIMEOUT_MSEC 5000 /* 5 sec */
35 #define REGULAR_USER 5000
36
37 static int _is_system_user(void)
38 {
39         uid_t uid = getuid();
40
41         if (uid < REGULAR_USER)
42                 return 1;
43         else
44                 return 0;
45 }
46
47 static GBusType __get_bus_type(pkgmgr_client_type type)
48 {
49         if (type == PC_REQUEST || _is_system_user())
50                 return G_BUS_TYPE_SYSTEM;
51         else
52                 return G_BUS_TYPE_SESSION;
53 }
54
55 int pkgmgr_client_connection_connect(struct pkgmgr_client_t *pc)
56 {
57         GError *error = NULL;
58         GBusType bus_type;
59
60 #if !GLIB_CHECK_VERSION(2, 35, 0)
61         g_type_init();
62 #endif
63         bus_type = __get_bus_type(pc->pc_type);
64         pc->conn = g_bus_get_sync(bus_type, NULL, &error);
65         if (error) {
66                 ERR("gdbus connection error (%s)", error->message);
67                 g_error_free(error);
68                 return PKGMGR_R_ECOMM;
69         }
70
71         return PKGMGR_R_OK;
72 }
73
74 void pkgmgr_client_connection_disconnect(struct pkgmgr_client_t *pc)
75 {
76         /* flush remaining buffer: blocking mode */
77         g_dbus_connection_flush_sync(pc->conn, NULL, NULL);
78         g_object_unref(pc->conn);
79         pc->conn = NULL;
80 }
81
82 struct signal_map {
83         const char *signal_str;
84         int signal_type;
85 };
86
87 struct signal_map map[] = {
88         {PKGMGR_INSTALLER_INSTALL_EVENT_STR, PKGMGR_CLIENT_STATUS_INSTALL},
89         {PKGMGR_INSTALLER_UNINSTALL_EVENT_STR, PKGMGR_CLIENT_STATUS_UNINSTALL},
90         {PKGMGR_INSTALLER_UPGRADE_EVENT_STR, PKGMGR_CLIENT_STATUS_UPGRADE},
91         {PKGMGR_INSTALLER_CLEAR_EVENT_STR, PKGMGR_CLIENT_STATUS_CLEAR_DATA},
92         {PKGMGR_INSTALLER_MOVE_EVENT_STR, PKGMGR_CLIENT_STATUS_MOVE},
93         {PKGMGR_INSTALLER_INSTALL_PERCENT_KEY_STR,
94                 PKGMGR_CLIENT_STATUS_INSTALL_PROGRESS},
95         {PKGMGR_INSTALLER_GET_SIZE_KEY_STR, PKGMGR_CLIENT_STATUS_GET_SIZE},
96         {PKGMGR_INSTALLER_CLEAR_CACHE_KEY_STR,
97                 PKGMGR_CLIENT_STATUS_CLEAR_CACHE},
98         {PKGMGR_INSTALLER_APP_ENABLE_EVENT_STR,
99                 PKGMGR_CLIENT_STATUS_ENABLE_APP},
100         {PKGMGR_INSTALLER_APP_DISABLE_EVENT_STR,
101                 PKGMGR_CLIENT_STATUS_DISABLE_APP},
102         {PKGMGR_INSTALLER_APP_ENABLE_SPLASH_SCREEN_EVENT_STR,
103                 PKGMGR_CLIENT_STATUS_ENABLE_APP_SPLASH_SCREEN},
104         {PKGMGR_INSTALLER_APP_DISABLE_SPLASH_SCREEN_EVENT_STR,
105                 PKGMGR_CLIENT_STATUS_DISABLE_APP_SPLASH_SCREEN},
106         {NULL, -1}
107 };
108
109 static int __get_signal_type(const char *name)
110 {
111         int i;
112
113         if (name == NULL)
114                 return -1;
115
116         for (i = 0; map[i].signal_str != NULL; i++) {
117                 if (strcmp(map[i].signal_str, name) == 0)
118                         return map[i].signal_type;
119         }
120
121         return -1;
122 }
123
124 static void __handle_size_info_callback(struct cb_info *cb_info,
125                 const char *pkgid, const char *val)
126 {
127         pkg_size_info_t size_info;
128         char buf[BUFMAX];
129         char *saveptr;
130         char *token;
131         pkgmgr_total_pkg_size_info_receive_cb callback;
132
133         snprintf(buf, sizeof(buf), "%s", val);
134
135         DBG("%s, %s", pkgid, val);
136
137         token = strtok_r(buf, ":", &saveptr);
138         if (token == NULL) {
139                 ERR("failed to parse size info");
140                 return;
141         }
142         size_info.data_size = atoll(token);
143         token = strtok_r(NULL, ":", &saveptr);
144         if (token == NULL) {
145                 ERR("failed to parse size info");
146                 return;
147         }
148         size_info.cache_size = atoll(token);
149         token = strtok_r(NULL, ":", &saveptr);
150         if (token == NULL) {
151                 ERR("failed to parse size info");
152                 return;
153         }
154         size_info.app_size = atoll(token);
155         token = strtok_r(NULL, ":", &saveptr);
156         if (token == NULL) {
157                 ERR("failed to parse size info");
158                 return;
159         }
160         size_info.ext_data_size = atoll(token);
161         token = strtok_r(NULL, ":", &saveptr);
162         if (token == NULL) {
163                 ERR("failed to parse size info");
164                 return;
165         }
166         size_info.ext_cache_size = atoll(token);
167         token = strtok_r(NULL, ":", &saveptr);
168         if (token == NULL) {
169                 ERR("failed to parse size info");
170                 return;
171         }
172         size_info.ext_app_size = atoll(token);
173
174         DBG("data: %lld, cache: %lld, app: %lld, ext_data: %lld, "
175                         "ext_cache: %lld, ext_app: %lld",
176                         size_info.data_size, size_info.cache_size,
177                         size_info.app_size, size_info.ext_data_size,
178                         size_info.ext_cache_size, size_info.ext_app_size);
179
180         if (strcmp(pkgid, PKG_SIZE_INFO_TOTAL) == 0) {
181                 callback = (pkgmgr_total_pkg_size_info_receive_cb)
182                         cb_info->size_info_cb;
183                 callback(cb_info->client, &size_info, cb_info->data);
184         } else {
185                 cb_info->size_info_cb(cb_info->client, pkgid, &size_info,
186                                 cb_info->data);
187         }
188 }
189
190 static void __signal_handler(GDBusConnection *conn, const gchar *sender_name,
191                 const gchar *object_path, const gchar *interface_name,
192                 const gchar *signal_name, GVariant *parameters,
193                 gpointer user_data)
194 {
195         uid_t target_uid;
196         char *req_id;
197         char *pkg_type = NULL;
198         char *appid = NULL;
199         char *pkgid = NULL;
200         char *key = NULL;
201         char *val = NULL;
202         struct cb_info *cb_info = (struct cb_info *)user_data;
203         GVariantIter *iter = NULL;
204         int signal_type;
205
206         g_variant_get(parameters, "(u&sa(sss)&s&s)", &target_uid, &req_id, &iter,
207                         &key, &val);
208         while (g_variant_iter_loop(iter, "(&s&s&s)", &pkgid, &appid, &pkg_type)) {
209                 if (cb_info->req_key) {
210                         if (strcmp(cb_info->req_key, req_id) != 0)
211                                 continue;
212                 } else {
213                         signal_type = __get_signal_type(signal_name);
214                         if (signal_type < 0 || !(cb_info->status_type & signal_type)) {
215                                 g_variant_iter_free(iter);
216                                 return;
217                         }
218                 }
219
220                 /* each cb_data can only has one callback */
221                 if (cb_info->event_cb) {
222                         cb_info->event_cb(target_uid, cb_info->req_id,
223                                         pkg_type, pkgid, key, val, NULL, cb_info->data);
224                 } else if (cb_info->app_event_cb && strcmp(appid, "") != 0) {
225                         cb_info->app_event_cb(target_uid, cb_info->req_id,
226                                         pkg_type, pkgid, appid, key, val, NULL,
227                                         cb_info->data);
228                 } else if (cb_info->size_info_cb)
229                         __handle_size_info_callback(cb_info, pkgid, val);
230
231                 /* TODO: unsubscribe request callback */
232         }
233         g_variant_iter_free(iter);
234 }
235
236 static void __set_signal_list(int event, GList **signal_list)
237 {
238         int i;
239         if (event == PKGMGR_CLIENT_STATUS_ALL)
240                 return;
241
242         for (i = 0; map[i].signal_str != NULL; i++) {
243                 if (event & map[i].signal_type)
244                         *signal_list = g_list_append(*signal_list,
245                                         (char *)map[i].signal_str);
246         }
247 }
248
249 static int __subscribe_signal(struct pkgmgr_client_t *pc,
250                 struct cb_info *cb_info, const char *signal)
251 {
252         guint sid = g_dbus_connection_signal_subscribe(pc->conn, NULL,
253                         PKGMGR_INSTALLER_DBUS_INTERFACE, signal,
254                         PKGMGR_INSTALLER_DBUS_OBJECT_PATH, NULL,
255                         G_DBUS_SIGNAL_FLAGS_NONE, __signal_handler,
256                         (gpointer)cb_info, NULL);
257         if (!sid) {
258                 ERR("failed to subscribe singal");
259                 return PKGMGR_R_ERROR;
260         }
261         cb_info->sid_list = g_list_append(cb_info->sid_list,
262                         GUINT_TO_POINTER(sid));
263
264         return PKGMGR_R_OK;
265 }
266
267 int pkgmgr_client_connection_set_callback(struct pkgmgr_client_t *pc,
268                 struct cb_info *cb_info)
269 {
270         GList *signal_list = NULL;
271         GList *tmp_list = NULL;
272         int ret;
273         char *signal_type;
274         __set_signal_list(pc->status_type, &signal_list);
275
276         if (g_list_length(signal_list) == 0)
277                 return __subscribe_signal(pc, cb_info, NULL);
278
279         for (tmp_list = signal_list; tmp_list != NULL;
280                         tmp_list = g_list_next(tmp_list)) {
281                 signal_type = (char *)tmp_list->data;
282                 ret = __subscribe_signal(pc, cb_info, signal_type);
283                 if (ret != 0) {
284                         g_list_free(signal_list);
285                         return PKGMGR_R_ERROR;
286                 }
287         }
288
289         g_list_free(signal_list);
290         return PKGMGR_R_OK;
291 }
292
293 void pkgmgr_client_connection_unset_callback(struct pkgmgr_client_t *pc,
294                 struct cb_info *cb_info)
295 {
296         GList *tmp = NULL;
297         guint tmp_sid;
298         for (tmp = cb_info->sid_list; tmp != NULL; tmp = g_list_next(tmp)) {
299                 tmp_sid = GPOINTER_TO_UINT(tmp->data);
300                 g_dbus_connection_signal_unsubscribe(pc->conn, tmp_sid);
301         }
302 }
303
304 int pkgmgr_client_connection_send_request(struct pkgmgr_client_t *pc,
305                 const char *method, GVariant *params, GVariant **result)
306 {
307         GError *error = NULL;
308         GDBusProxy *proxy;
309         GVariant *r = NULL;
310         int retry_cnt = 0;
311         int ret = PKGMGR_R_ECOMM;
312
313         /* convert floating ref into normal ref */
314         g_variant_ref_sink(params);
315
316         do {
317                 proxy = g_dbus_proxy_new_sync(pc->conn,
318                                 G_DBUS_PROXY_FLAGS_DO_NOT_LOAD_PROPERTIES,
319                                 NULL, PKGMGR_DBUS_SERVICE,
320                                 PKGMGR_DBUS_OBJECT_PATH,
321                                 PKGMGR_DBUS_INTERFACE, NULL, &error);
322                 if (proxy == NULL) {
323                         ERR("failed to get proxy object, sleep and retry[%s]",
324                                         error->message);
325                         g_error_free(error);
326                         error = NULL;
327                         usleep(CONNECTION_WAIT_USEC);
328                         retry_cnt++;
329                         continue;
330                 }
331
332                 r = g_dbus_proxy_call_sync(proxy, method, params,
333                                 G_DBUS_CALL_FLAGS_NONE,
334                                 CONNECTION_TIMEOUT_MSEC, NULL, &error);
335                 g_object_unref(proxy);
336                 if (error && error->code == G_DBUS_ERROR_ACCESS_DENIED) {
337                         ERR("failed to send request, privilege denied[%s]",
338                                         error->message);
339                         g_error_free(error);
340                         ret = PKGMGR_R_EPRIV;
341                         break;
342                 }
343                 if (r) {
344                         *result = r;
345                         ret = PKGMGR_R_OK;
346                         break;
347                 }
348
349                 ERR("failed to send request, sleep and retry[%s]",
350                                 error->message);
351                 g_error_free(error);
352                 error = NULL;
353                 usleep(CONNECTION_WAIT_USEC);
354                 retry_cnt++;
355         } while (retry_cnt <= CONNECTION_RETRY_MAX);
356
357         /* decrease ref count to 0 to free resource */
358         g_variant_unref(params);
359
360         return ret;
361 }