1 /*
2 * Copyright (C) 2011 The Android Open Source Project
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 /* NOTICE: This is a clean room re-implementation of libnl */
18
19 #include <malloc.h>
20 #include "netlink-types.h"
21 #include "netlink/handlers.h"
22
23 /* Allocate a new callback handle. */
nl_cb_alloc(enum nl_cb_kind kind)24 struct nl_cb *nl_cb_alloc(enum nl_cb_kind kind)
25 {
26 struct nl_cb *cb;
27
28 cb = (struct nl_cb *) malloc(sizeof(struct nl_cb));
29 if (cb == NULL)
30 goto fail;
31 memset(cb, 0, sizeof(*cb));
32
33 return nl_cb_get(cb);
34 fail:
35 return NULL;
36 }
37
38 /* Clone an existing callback handle */
nl_cb_clone(struct nl_cb * orig)39 struct nl_cb *nl_cb_clone(struct nl_cb *orig)
40 {
41 struct nl_cb *new_cb;
42
43 new_cb = nl_cb_alloc(NL_CB_DEFAULT);
44 if (new_cb == NULL)
45 goto fail;
46
47 /* Copy original and set refcount to 1 */
48 memcpy(new_cb, orig, sizeof(*orig));
49 new_cb->cb_refcnt = 1;
50
51 return new_cb;
52 fail:
53 return NULL;
54 }
55
56 /* Set up a callback. */
nl_cb_set(struct nl_cb * cb,enum nl_cb_type type,enum nl_cb_kind kind,nl_recvmsg_msg_cb_t func,void * arg)57 int nl_cb_set(struct nl_cb *cb, enum nl_cb_type type, enum nl_cb_kind kind, \
58 nl_recvmsg_msg_cb_t func, void *arg)
59 {
60 cb->cb_set[type] = func;
61 cb->cb_args[type] = arg;
62 return 0;
63 }
64
65
66
67 /* Set up an error callback. */
nl_cb_err(struct nl_cb * cb,enum nl_cb_kind kind,nl_recvmsg_err_cb_t func,void * arg)68 int nl_cb_err(struct nl_cb *cb, enum nl_cb_kind kind, \
69 nl_recvmsg_err_cb_t func, void *arg)
70 {
71 cb->cb_err = func;
72 cb->cb_err_arg = arg;
73 return 0;
74
75 }
76
nl_cb_get(struct nl_cb * cb)77 struct nl_cb *nl_cb_get(struct nl_cb *cb)
78 {
79 cb->cb_refcnt++;
80 return cb;
81 }
82
nl_cb_put(struct nl_cb * cb)83 void nl_cb_put(struct nl_cb *cb)
84 {
85 if (!cb)
86 return;
87 cb->cb_refcnt--;
88 if (cb->cb_refcnt <= 0)
89 free(cb);
90 }
91