Free before return
[platform/core/connectivity/stc-manager.git] / src / stc-restriction.c
1 /*
2  * Copyright (c) 2016 Samsung Electronics Co., Ltd.
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 #include "stc-db.h"
18 #include "table-restrictions.h"
19 #include "stc-restriction.h"
20 #include "stc-manager-gdbus.h"
21 #include "stc-manager-plugin-monitor.h"
22
23 #define RESTRICTION_DBUS_ERROR_NAME "net.stc.restriction.Error.Failed"
24
25 #define STC_RESTRICTION_DBUS_REPLY_ERROR(invocation, err_num) \
26         g_dbus_method_invocation_return_dbus_error((invocation), \
27                                                    RESTRICTION_DBUS_ERROR_NAME, \
28                                                    stc_err_strs[-(err_num)])
29
30 static const gchar *stc_err_strs[] = {
31         "ERROR_NONE",
32         "FAIL",
33         "DB_FAILED",
34         "OUT_OF_MEMORY",
35         "INVALID_PARAMETER",
36         "NO_DATA",
37         "ALREADY_DATA",
38         "UNINITIALIZED",
39         "PERMISSION_DENIED",
40         "NOTIMPL"
41 };
42
43 void __initialize_rstn_rule(table_restrictions_info *rule)
44 {
45         rule->app_id = NULL;
46
47         rule->iftype = STC_IFACE_ALL;
48         rule->ifname = NULL;
49
50         rule->subscriber_id = NULL;
51         rule->roaming = STC_ROAMING_DISABLE;
52
53         rule->rstn_type = STC_RSTN_TYPE_UNKNOWN;
54         rule->data_limit = -1;
55         rule->data_warn_limit = -1;
56
57         rule->month_start_date = 1;
58         rule->monthly_limit = -1;
59         rule->weekly_limit = -1;
60         rule->daily_limit = -1;
61
62         rule->mac = NULL;
63 }
64
65 gboolean __validate_rstn_rule(table_restrictions_info *rule)
66 {
67         __STC_LOG_FUNC_ENTER__;
68
69         if (rule == NULL) {
70                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
71                 return FALSE; //LCOV_EXCL_LINE
72         }
73
74         if (rule->iftype <= STC_IFACE_UNKNOWN ||
75             rule->iftype >= STC_IFACE_LAST_ELEM) {
76                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
77                 return FALSE; //LCOV_EXCL_LINE
78         }
79
80         if (rule->roaming > STC_ROAMING_DISABLE) {
81                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
82                 return FALSE; //LCOV_EXCL_LINE
83         }
84
85         if (rule->subscriber_id == NULL) {
86                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
87                 return FALSE; //LCOV_EXCL_LINE
88         }
89
90         if (rule->app_id == NULL) {
91                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
92                 return FALSE; //LCOV_EXCL_LINE
93         }
94
95         __STC_LOG_FUNC_EXIT__;
96         return TRUE;
97 }
98
99 void __stc_restriction_app_info_builder_add(GVariantBuilder *builder,
100                                             const table_restrictions_info *info)
101 {
102         __STC_LOG_FUNC_ENTER__;
103
104         if (!builder || !info) {
105                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
106                 return; //LCOV_EXCL_LINE
107         }
108
109         g_variant_builder_add(builder, "{sv}", "app_id",
110                               g_variant_new_string(info->app_id));
111
112         g_variant_builder_add(builder, "{sv}", "ifname",
113                               g_variant_new_string(info->ifname));
114
115         g_variant_builder_add(builder, "{sv}", "iftype",
116                               g_variant_new_uint16(info->iftype));
117
118         g_variant_builder_add(builder, "{sv}", "rstn_type",
119                               g_variant_new_uint16(info->rstn_type));
120
121         g_variant_builder_add(builder, "{sv}", "data_limit",
122                               g_variant_new_int64(info->data_limit));
123
124         g_variant_builder_add(builder, "{sv}", "data_warn_limit",
125                               g_variant_new_int64(info->data_warn_limit));
126
127         g_variant_builder_add(builder, "{sv}", "month_start_date",
128                               g_variant_new_int32(info->month_start_date));
129
130         g_variant_builder_add(builder, "{sv}", "monthly_limit",
131                               g_variant_new_int64(info->monthly_limit));
132
133         g_variant_builder_add(builder, "{sv}", "weekly_limit",
134                               g_variant_new_int64(info->weekly_limit));
135
136         g_variant_builder_add(builder, "{sv}", "daily_limit",
137                               g_variant_new_int64(info->daily_limit));
138
139         g_variant_builder_add(builder, "{sv}", "roaming",
140                               g_variant_new_uint16(info->roaming));
141
142         g_variant_builder_add(builder, "{sv}", "subscriber_id",
143                               g_variant_new_string(info->subscriber_id));
144
145         __STC_LOG_FUNC_EXIT__;
146 }
147
148 stc_cb_ret_e __table_restrictions_foreach_app_cb(const table_restrictions_info *info,
149                                                  void *user_data)
150 {
151         __STC_LOG_FUNC_ENTER__;
152         GVariantBuilder *builder = (GVariantBuilder *)user_data;
153         GVariantBuilder sub_builder;
154
155         if (!info || !builder) {
156                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
157                 return STC_CANCEL; //LCOV_EXCL_LINE
158         }
159
160         g_variant_builder_init(&sub_builder, G_VARIANT_TYPE("a{sv}"));
161         __stc_restriction_app_info_builder_add(&sub_builder, info);
162
163         g_variant_builder_add_value(builder,
164                                     g_variant_builder_end(&sub_builder));
165
166         __STC_LOG_FUNC_EXIT__;
167         return STC_CONTINUE;
168 }
169
170 stc_cb_ret_e __table_restrictions_per_app_cb(const table_restrictions_info *info,
171                                              void *user_data)
172 {
173         __STC_LOG_FUNC_ENTER__;
174         GVariantBuilder *builder = (GVariantBuilder *)user_data;
175
176         if (!info || !builder) {
177                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
178                 return STC_CANCEL; //LCOV_EXCL_LINE
179         }
180
181         __stc_restriction_app_info_builder_add(builder, info);
182
183         __STC_LOG_FUNC_EXIT__;
184         return STC_CONTINUE;
185 }
186
187 static void __stc_extract_restriction_rule(const char *key, GVariant *value,
188                                            void *user_data)
189 {
190         table_restrictions_info *rule =
191                 (table_restrictions_info *) user_data;
192         if (rule == NULL) {
193                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
194                 return; //LCOV_EXCL_LINE
195         }
196
197         if (!g_strcmp0(key, "app_id")) {
198                 gsize str_length;
199                 const gchar *str = g_variant_get_string(value, &str_length);
200                 rule->app_id = g_strdup(str);
201                 STC_LOGD("app_id: [%s]", rule->app_id);
202
203         } else if (!g_strcmp0(key, "ifname")) {
204                 gsize str_length;
205                 const gchar *str = g_variant_get_string(value, &str_length);
206                 rule->ifname = g_strdup(str);
207                 STC_LOGD("ifname: [%s]", rule->ifname);
208
209         } else if (!g_strcmp0(key, "iftype")) {
210                 rule->iftype = g_variant_get_uint16(value);
211                 STC_LOGD("iftype: [%u]", (unsigned int) rule->iftype);
212
213         } else if (!g_strcmp0(key, "data_limit")) {
214                 rule->data_limit = g_variant_get_int64(value);
215                 STC_LOGD("data_limit: [%lld]", rule->data_limit);
216
217         } else if (!g_strcmp0(key, "data_warn_limit")) {
218                 rule->data_warn_limit = g_variant_get_int64(value);
219                 STC_LOGD("data_warn_limit: [%lld]", rule->data_warn_limit);
220
221         } else if (!g_strcmp0(key, "month_start_date")) {
222                 rule->month_start_date = g_variant_get_int32(value);
223                 STC_LOGD("month_start_date: [%d]", rule->month_start_date);
224
225         } else if (!g_strcmp0(key, "monthly_limit")) {
226                 rule->monthly_limit = g_variant_get_int64(value);
227                 STC_LOGD("monthly_limit: [%lld]", rule->monthly_limit);
228
229         } else if (!g_strcmp0(key, "weekly_limit")) {
230                 rule->weekly_limit = g_variant_get_int64(value);
231                 STC_LOGD("weekly_limit: [%lld]", rule->weekly_limit);
232
233         } else if (!g_strcmp0(key, "daily_limit")) {
234                 rule->daily_limit = g_variant_get_int64(value);
235                 STC_LOGD("daily_limit: [%lld]", rule->daily_limit);
236
237         } else if (!g_strcmp0(key, "roaming")) {
238                 rule->roaming = g_variant_get_uint16(value);
239                 STC_LOGD("roaming: [%u]", rule->roaming);
240
241         } else if (!g_strcmp0(key, "subscriber_id")) {
242                 gsize str_length;
243                 const gchar *str = g_variant_get_string(value, &str_length);
244                 rule->subscriber_id = g_strdup(str);
245                 STC_LOGD("subscriber_id: [%s]", rule->subscriber_id);
246
247         } else if (!g_strcmp0(key, "rstn_type")) {
248                 rule->rstn_type = g_variant_get_uint16(value);
249                 STC_LOGD("type: [%u]", (unsigned int) rule->rstn_type);
250
251         } else if (!g_strcmp0(key, "mac")) {
252                 gsize str_length;
253                 const gchar *str = g_variant_get_string(value, &str_length);
254                 rule->mac = g_strdup(str);
255                 STC_LOGD("mac: [%s]", rule->mac);
256
257         } else {
258                 STC_LOGD("Unknown select rule"); //LCOV_EXCL_LINE
259         }
260 }
261
262 static void __stc_free_restriction_rule_members(table_restrictions_info *rule)
263 {
264         FREE(rule->app_id);
265         FREE(rule->ifname);
266         FREE(rule->subscriber_id);
267         FREE(rule->mac);
268 }
269
270 gboolean handle_restriction_set(StcRestriction *object,
271                                 GDBusMethodInvocation *invocation,
272                                 GVariant *parameters,
273                                 void *user_data)
274 {
275         __STC_LOG_FUNC_ENTER__;
276         GVariantIter *iter = NULL;
277         table_restrictions_info rule;
278         stc_error_e ret = STC_ERROR_NONE;
279
280         stc_set_keep_alive(TRUE);
281
282         memset(&rule, 0, sizeof(table_restrictions_info));
283         __initialize_rstn_rule(&rule);
284
285         g_variant_get(parameters, "a{sv}", &iter);
286         if (iter != NULL) {
287                 stc_manager_gdbus_dict_foreach(iter,
288                                                __stc_extract_restriction_rule,
289                                                &rule);
290                 g_variant_iter_free(iter);
291         }
292
293         if (__validate_rstn_rule(&rule) == FALSE) {
294                 STC_RESTRICTION_DBUS_REPLY_ERROR(invocation, //LCOV_EXCL_LINE
295                                                  STC_ERROR_INVALID_PARAMETER);
296                 __stc_free_restriction_rule_members(&rule);
297                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
298                 return TRUE;
299         }
300
301         ret = table_restrictions_update(&rule);
302         if (ret != STC_ERROR_NONE) {
303                 STC_RESTRICTION_DBUS_REPLY_ERROR(invocation, //LCOV_EXCL_LINE
304                                                  STC_ERROR_DB_FAILED);
305                 __stc_free_restriction_rule_members(&rule);
306                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
307                 return TRUE;
308         }
309
310         stc_plugin_monitor_add_rstn(&rule);
311         STC_DBUS_REPLY_ERROR_NONE(invocation);
312         __stc_free_restriction_rule_members(&rule);
313         __STC_LOG_FUNC_EXIT__;
314         return TRUE;
315 }
316
317 gboolean handle_restriction_unset(StcRestriction *object,
318                                    GDBusMethodInvocation *invocation,
319                                    GVariant *parameters,
320                                    void *user_data)
321 {
322         __STC_LOG_FUNC_ENTER__;
323         GVariantIter *iter = NULL;
324         table_restrictions_info rule;
325         stc_error_e ret = STC_ERROR_NONE;
326
327         stc_set_keep_alive(TRUE);
328
329         memset(&rule, 0, sizeof(table_restrictions_info));
330         __initialize_rstn_rule(&rule);
331
332         g_variant_get(parameters, "a{sv}", &iter);
333         if (iter != NULL) {
334                 stc_manager_gdbus_dict_foreach(iter,
335                                                __stc_extract_restriction_rule,
336                                                &rule);
337                 g_variant_iter_free(iter);
338         }
339
340         if (__validate_rstn_rule(&rule) == FALSE) {
341                 STC_RESTRICTION_DBUS_REPLY_ERROR(invocation, //LCOV_EXCL_LINE
342                                                  STC_ERROR_INVALID_PARAMETER);
343                 __stc_free_restriction_rule_members(&rule);
344                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
345                 return TRUE;
346         }
347
348         ret = table_restrictions_delete(rule.app_id, rule.iftype, rule.ifname,
349                                         rule.subscriber_id, rule.roaming);
350         if (ret != STC_ERROR_NONE) {
351                 STC_RESTRICTION_DBUS_REPLY_ERROR(invocation, //LCOV_EXCL_LINE
352                                                  STC_ERROR_DB_FAILED);
353                 __stc_free_restriction_rule_members(&rule);
354                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
355                 return TRUE;
356         }
357
358         stc_plugin_monitor_remove_rstn(&rule);
359         STC_DBUS_REPLY_ERROR_NONE(invocation);
360         __stc_free_restriction_rule_members(&rule);
361         __STC_LOG_FUNC_EXIT__;
362         return TRUE;
363 }
364
365 gboolean handle_restriction_set_list(StcRestriction *object,
366                                 GDBusMethodInvocation *invocation,
367                                 GVariant *parameters,
368                                 void *user_data)
369 {
370         __STC_LOG_FUNC_ENTER__;
371         GVariantIter *iter = NULL;
372         GVariantIter *iter_row = NULL;
373         stc_error_e ret = STC_ERROR_NONE;
374
375         stc_set_keep_alive(TRUE);
376
377         g_variant_get(parameters, "aa{sv}", &iter);
378         while (g_variant_iter_next(iter, "a{sv}", &iter_row)) {
379                 table_restrictions_info rule;
380
381                 memset(&rule, 0, sizeof(table_restrictions_info));
382                 __initialize_rstn_rule(&rule);
383
384                 stc_manager_gdbus_dict_foreach(iter_row,
385                         __stc_extract_restriction_rule, &rule);
386
387                 if (__validate_rstn_rule(&rule)) {
388                         ret = table_restrictions_update(&rule);
389                         if (ret != STC_ERROR_NONE) {
390                                 STC_RESTRICTION_DBUS_REPLY_ERROR(invocation, ret); //LCOV_EXCL_LINE
391                                 g_variant_iter_free(iter_row);
392                                 g_variant_iter_free(iter);
393                                 __stc_free_restriction_rule_members(&rule);
394                                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
395                                 return TRUE;
396                         }
397
398                         ret = stc_plugin_monitor_add_rstn(&rule);
399                         if (ret != STC_ERROR_NONE) {
400                                 STC_RESTRICTION_DBUS_REPLY_ERROR(invocation, ret); //LCOV_EXCL_LINE
401                                 g_variant_iter_free(iter_row);
402                                 g_variant_iter_free(iter);
403                                 __stc_free_restriction_rule_members(&rule);
404                                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
405                                 return TRUE;
406                         }
407                 }
408
409                 __stc_free_restriction_rule_members(&rule);
410                 g_variant_iter_free(iter_row);
411         }
412         g_variant_iter_free(iter);
413
414         STC_DBUS_REPLY_ERROR_NONE(invocation);
415         __STC_LOG_FUNC_EXIT__;
416         return TRUE;
417 }
418
419 gboolean handle_restriction_unset_list(StcRestriction *object,
420                                    GDBusMethodInvocation *invocation,
421                                    GVariant *parameters,
422                                    void *user_data)
423 {
424         __STC_LOG_FUNC_ENTER__;
425         GVariantIter *iter = NULL;
426         GVariantIter *iter_row = NULL;
427         stc_error_e ret = STC_ERROR_NONE;
428
429         stc_set_keep_alive(TRUE);
430
431         g_variant_get(parameters, "aa{sv}", &iter);
432         while (g_variant_iter_next(iter, "a{sv}", &iter_row)) {
433                 table_restrictions_info rule;
434
435                 memset(&rule, 0, sizeof(table_restrictions_info));
436                 __initialize_rstn_rule(&rule);
437
438                 stc_manager_gdbus_dict_foreach(iter_row,
439                         __stc_extract_restriction_rule, &rule);
440
441                 if (__validate_rstn_rule(&rule)) {
442                         ret = table_restrictions_delete(rule.app_id, rule.iftype,
443                                                 rule.ifname, rule.subscriber_id, rule.roaming);
444                         if (ret != STC_ERROR_NONE) {
445                                 STC_RESTRICTION_DBUS_REPLY_ERROR(invocation, ret); //LCOV_EXCL_LINE
446                                 g_variant_iter_free(iter_row);
447                                 g_variant_iter_free(iter);
448                                 __stc_free_restriction_rule_members(&rule);
449                                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
450                                 return TRUE;
451                         }
452
453                         ret = stc_plugin_monitor_remove_rstn(&rule);
454                         if (ret != STC_ERROR_NONE) {
455                                 STC_RESTRICTION_DBUS_REPLY_ERROR(invocation, ret); //LCOV_EXCL_LINE
456                                 g_variant_iter_free(iter_row);
457                                 g_variant_iter_free(iter);
458                                 __stc_free_restriction_rule_members(&rule);
459                                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
460                                 return TRUE;
461                         }
462                 }
463
464                 __stc_free_restriction_rule_members(&rule);
465                 g_variant_iter_free(iter_row);
466         }
467         g_variant_iter_free(iter);
468
469         STC_DBUS_REPLY_ERROR_NONE(invocation);
470         __STC_LOG_FUNC_EXIT__;
471         return TRUE;
472 }
473
474 gboolean handle_restriction_get(StcRestriction *object,
475                                 GDBusMethodInvocation *invocation,
476                                 const gchar *app_id,
477                                 void *user_data)
478 {
479         __STC_LOG_FUNC_ENTER__;
480         GVariantBuilder *builder = NULL;
481         GVariant *return_parameters = NULL;
482         stc_error_e ret;
483
484         stc_set_keep_alive(TRUE);
485
486         builder = g_variant_builder_new(G_VARIANT_TYPE("a{sv}"));
487
488         ret = table_restrictions_per_app(app_id,
489                                          __table_restrictions_per_app_cb,
490                                          builder);
491         if (ret < STC_ERROR_NONE) {
492                 g_variant_builder_unref(builder); //LCOV_EXCL_LINE
493                 STC_RESTRICTION_DBUS_REPLY_ERROR(invocation, ret); //LCOV_EXCL_LINE
494                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
495                 return TRUE; //LCOV_EXCL_LINE
496         }
497
498         return_parameters = g_variant_new("(ia{sv})", STC_ERROR_NONE, builder);
499         g_variant_builder_unref(builder);
500
501         DEBUG_GDBUS_VARIANT("Return parameters: ", return_parameters);
502         STC_DBUS_REPLY(invocation, return_parameters);
503         __STC_LOG_FUNC_EXIT__;
504         return TRUE;
505 }
506
507 gboolean handle_restriction_get_all(StcRestriction *object,
508                                     GDBusMethodInvocation *invocation,
509                                     void *user_data)
510 {
511         __STC_LOG_FUNC_ENTER__;
512         GVariantBuilder *builder = NULL;
513         GVariant *return_parameters = NULL;
514         stc_error_e ret;
515
516         stc_set_keep_alive(TRUE);
517
518         builder = g_variant_builder_new(G_VARIANT_TYPE("aa{sv}"));
519
520         ret = table_restrictions_foreach(__table_restrictions_foreach_app_cb,
521                                          builder);
522         if (ret < STC_ERROR_NONE) {
523                 g_variant_builder_unref(builder); //LCOV_EXCL_LINE
524                 STC_RESTRICTION_DBUS_REPLY_ERROR(invocation, ret); //LCOV_EXCL_LINE
525                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
526                 return TRUE; //LCOV_EXCL_LINE
527         }
528
529         return_parameters = g_variant_new("(iaa{sv})", STC_ERROR_NONE, builder);
530         g_variant_builder_unref(builder);
531
532         DEBUG_GDBUS_VARIANT("Return parameters: ", return_parameters);
533         STC_DBUS_REPLY(invocation, return_parameters);
534         __STC_LOG_FUNC_EXIT__;
535         return TRUE;
536 }
537
538 gboolean handle_restriction_get_type(StcRestriction *object,
539                                       GDBusMethodInvocation *invocation,
540                                       const gchar *app_id,
541                                       int iftype,
542                                       void *user_data)
543 {
544         __STC_LOG_FUNC_ENTER__;
545         GVariant *return_parameters = NULL;
546         stc_rstn_type_e type = STC_RSTN_TYPE_UNKNOWN;
547         stc_error_e ret;
548
549         stc_set_keep_alive(TRUE);
550
551         ret = table_restrictions_get_restriction_type(app_id, iftype, &type);
552         if (ret < STC_ERROR_NONE) {
553                 STC_RESTRICTION_DBUS_REPLY_ERROR(invocation, ret); //LCOV_EXCL_LINE
554                 __STC_LOG_FUNC_EXIT__; //LCOV_EXCL_LINE
555                 return TRUE;
556         }
557
558         return_parameters = g_variant_new("(ii)", STC_ERROR_NONE, type);
559         STC_DBUS_REPLY(invocation, return_parameters);
560         __STC_LOG_FUNC_EXIT__;
561         return TRUE;
562 }