1 %module capi 2 %{ 3 #include <netlink/netlink.h> 4 #include <netlink/types.h> 5 #include <netlink/socket.h> 6 #include <netlink/msg.h> 7 #include <netlink/object.h> 8 #include <netlink/cache.h> 9 #include <netlink/attr.h> 10 #include <net/if.h> 11 12 #define DEBUG 13 #include "utils.h" 14 %} 15 16 %include <stdint.i> 17 %include <cstring.i> 18 %include <cpointer.i> 19 20 %inline %{ 21 struct nl_dump_params *alloc_dump_params(void) 22 { 23 struct nl_dump_params *dp; 24 if (!(dp = calloc(1, sizeof(*dp)))) 25 return NULL; 26 dp->dp_fd = stdout; 27 return dp; 28 } 29 30 void free_dump_params(struct nl_dump_params *dp) 31 { 32 free(dp); 33 } 34 %}; 35 36 /* <netlink/types.h> */ 37 38 enum nl_dump_type { 39 NL_DUMP_LINE, /**< Dump object briefly on one line */ 40 NL_DUMP_DETAILS, /**< Dump all attributes but no statistics */ 41 NL_DUMP_STATS, /**< Dump all attributes including statistics */ 42 __NL_DUMP_MAX, 43 }; 44 45 struct nl_dump_params 46 { 47 /** 48 * Specifies the type of dump that is requested. 49 */ 50 enum nl_dump_type dp_type; 51 52 /** 53 * Specifies the number of whitespaces to be put in front 54 * of every new line (indentation). 55 */ 56 int dp_prefix; 57 58 /** 59 * Causes the cache index to be printed for each element. 60 */ 61 int dp_print_index; 62 63 /** 64 * Causes each element to be prefixed with the message type. 65 */ 66 int dp_dump_msgtype; 67 68 /** 69 * A callback invoked for output 70 * 71 * Passed arguments are: 72 * - dumping parameters 73 * - string to append to the output 74 */ 75 void (*dp_cb)(struct nl_dump_params *, char *); 76 77 /** 78 * A callback invoked for every new line, can be used to 79 * customize the indentation. 80 * 81 * Passed arguments are: 82 * - dumping parameters 83 * - line number starting from 0 84 */ 85 void (*dp_nl_cb)(struct nl_dump_params *, int); 86 87 /** 88 * User data pointer, can be used to pass data to callbacks. 89 */ 90 void *dp_data; 91 92 /** 93 * File descriptor the dumping output should go to 94 */ 95 FILE * dp_fd; 96 97 /** 98 * Alternatively the output may be redirected into a buffer 99 */ 100 char * dp_buf; 101 102 /** 103 * Length of the buffer dp_buf 104 */ 105 size_t dp_buflen; 106 107 /** 108 * PRIVATE 109 * Set if a dump was performed prior to the actual dump handler. 110 */ 111 int dp_pre_dump; 112 113 /** 114 * PRIVATE 115 * Owned by the current caller 116 */ 117 int dp_ivar; 118 119 unsigned int dp_line; 120 }; 121 122 /* <net/if.h> */ 123 extern unsigned int if_nametoindex(const char *ifname); 124 125 /* <netlink/errno.h> */ 126 extern const char *nl_geterror(int); 127 128 /* <netlink/utils.h> */ 129 130 extern double nl_cancel_down_bytes(unsigned long long, char **); 131 extern double nl_cancel_down_bits(unsigned long long, char **); 132 %cstring_output_maxsize(char *buf, size_t len) 133 extern int nl_rate2str(unsigned long long rate, int type, char *buf, size_t len); 134 extern double nl_cancel_down_us(uint32_t, char **); 135 136 extern long nl_size2int(const char *); 137 %cstring_output_maxsize(char *buf, const size_t len) 138 extern char *nl_size2str(const size_t, char *buf, const size_t len); 139 extern long nl_prob2int(const char *); 140 141 extern int nl_get_user_hz(void); 142 extern uint32_t nl_us2ticks(uint32_t); 143 extern uint32_t nl_ticks2us(uint32_t); 144 extern int nl_str2msec(const char *, uint64_t *); 145 146 %cstring_output_maxsize(char *buf, size_t len) 147 extern char *nl_msec2str(uint64_t, char *buf, size_t len); 148 149 %cstring_output_maxsize(char *buf, size_t len) 150 extern char *nl_llproto2str(int, char *buf, size_t len); 151 extern int nl_str2llproto(const char *); 152 153 %cstring_output_maxsize(char *buf, size_t len) 154 extern char *nl_ether_proto2str(int, char *buf, size_t len); 155 extern int nl_str2ether_proto(const char *); 156 157 %cstring_output_maxsize(char *buf, size_t len) 158 extern char *nl_ip_proto2str(int, char *buf, size_t len); 159 extern int nl_str2ip_proto(const char *); 160 161 extern void nl_new_line(struct nl_dump_params *); 162 extern void nl_dump(struct nl_dump_params *, const char *, ...); 163 extern void nl_dump_line(struct nl_dump_params *, const char *, ...); 164 165 /* <netlink/netlink.h> */ 166 extern struct nl_dump_params *alloc_dump_params(void); 167 extern void free_dump_params(struct nl_dump_params *); 168 169 extern int nl_connect(struct nl_sock *, int); 170 extern void nl_close(struct nl_sock *); 171 172 /* <netlink/socket.h> */ 173 extern struct nl_sock *nl_socket_alloc(void); 174 extern struct nl_sock *nl_socket_alloc_cb(struct nl_cb *); 175 extern void nl_socket_free(struct nl_sock *); 176 177 extern uint32_t nl_socket_get_local_port(const struct nl_sock *); 178 extern void nl_socket_set_local_port(struct nl_sock *, uint32_t); 179 180 extern uint32_t nl_socket_get_peer_port(const struct nl_sock *); 181 extern void nl_socket_set_peer_port(struct nl_sock *, uint32_t); 182 183 extern uint32_t nl_socket_get_peer_groups(const struct nl_sock *sk); 184 extern void nl_socket_set_peer_groups(struct nl_sock *sk, uint32_t groups); 185 186 extern int nl_socket_set_buffer_size(struct nl_sock *, int, int); 187 extern void nl_socket_set_cb(struct nl_sock *, struct nl_cb *); 188 189 extern int nl_send_auto_complete(struct nl_sock *, struct nl_msg *); 190 extern int nl_recvmsgs(struct nl_sock *, struct nl_cb *); 191 192 /* <netlink/msg.h> */ 193 extern int nlmsg_size(int); 194 extern int nlmsg_total_size(int); 195 extern int nlmsg_padlen(int); 196 197 extern void * nlmsg_data(const struct nlmsghdr *); 198 extern int nlmsg_datalen(const struct nlmsghdr *); 199 extern void * nlmsg_tail(const struct nlmsghdr *); 200 201 /* attribute access */ 202 extern struct nlattr * nlmsg_attrdata(const struct nlmsghdr *, int); 203 extern int nlmsg_attrlen(const struct nlmsghdr *, int); 204 205 /* message parsing */ 206 extern int nlmsg_valid_hdr(const struct nlmsghdr *, int); 207 extern int nlmsg_ok(const struct nlmsghdr *, int); 208 extern struct nlmsghdr * nlmsg_next(struct nlmsghdr *, int *); 209 extern int nlmsg_parse(struct nlmsghdr *, int, struct nlattr **, 210 int, struct nla_policy *); 211 extern struct nlattr * nlmsg_find_attr(struct nlmsghdr *, int, int); 212 extern int nlmsg_validate(struct nlmsghdr *, int, int, 213 struct nla_policy *); 214 215 extern struct nl_msg * nlmsg_alloc(void); 216 extern struct nl_msg * nlmsg_alloc_size(size_t); 217 extern struct nl_msg * nlmsg_alloc_simple(int, int); 218 extern void nlmsg_set_default_size(size_t); 219 extern struct nl_msg * nlmsg_inherit(struct nlmsghdr *); 220 extern struct nl_msg * nlmsg_convert(struct nlmsghdr *); 221 extern void * nlmsg_reserve(struct nl_msg *, size_t, int); 222 extern int nlmsg_append(struct nl_msg *, void *, size_t, int); 223 extern int nlmsg_expand(struct nl_msg *, size_t); 224 225 extern struct nlmsghdr * nlmsg_put(struct nl_msg *, uint32_t, uint32_t, 226 int, int, int); 227 extern struct nlmsghdr * nlmsg_hdr(struct nl_msg *); 228 extern void nlmsg_get(struct nl_msg *); 229 extern void nlmsg_free(struct nl_msg *); 230 231 /* attribute modification */ 232 extern void nlmsg_set_proto(struct nl_msg *, int); 233 extern int nlmsg_get_proto(struct nl_msg *); 234 extern size_t nlmsg_get_max_size(struct nl_msg *); 235 extern void nlmsg_set_src(struct nl_msg *, struct sockaddr_nl *); 236 extern struct sockaddr_nl *nlmsg_get_src(struct nl_msg *); 237 extern void nlmsg_set_dst(struct nl_msg *, struct sockaddr_nl *); 238 extern struct sockaddr_nl *nlmsg_get_dst(struct nl_msg *); 239 extern void nlmsg_set_creds(struct nl_msg *, struct ucred *); 240 extern struct ucred * nlmsg_get_creds(struct nl_msg *); 241 242 extern char * nl_nlmsgtype2str(int, char *, size_t); 243 extern int nl_str2nlmsgtype(const char *); 244 245 extern char * nl_nlmsg_flags2str(int, char *, size_t); 246 247 extern int nl_msg_parse(struct nl_msg *, 248 void (*cb)(struct nl_object *, void *), 249 void *); 250 251 extern void nl_msg_dump(struct nl_msg *, FILE *); 252 253 %inline %{ 254 struct nl_object *cast_obj(void *obj) 255 { 256 return (struct nl_object *) obj; 257 } 258 259 struct nl_object *object_alloc_name(const char *name) 260 { 261 struct nl_object *obj; 262 263 if (nl_object_alloc_name(name, &obj) < 0) 264 return NULL; 265 266 return obj; 267 } 268 %}; 269 270 extern struct nl_object *nl_object_alloc(struct nl_object_ops *); 271 extern void nl_object_free(struct nl_object *); 272 extern struct nl_object *nl_object_clone(struct nl_object *); 273 extern void nl_object_get(struct nl_object *); 274 extern void nl_object_put(struct nl_object *); 275 extern int nl_object_shared(struct nl_object *); 276 277 %cstring_output_maxsize(char *buf, size_t len) 278 extern void nl_object_dump_buf(struct nl_object *, char *buf, size_t len); 279 280 extern void nl_object_dump(struct nl_object *, struct nl_dump_params *); 281 282 extern int nl_object_identical(struct nl_object *, struct nl_object *); 283 extern uint32_t nl_object_diff(struct nl_object *, struct nl_object *); 284 extern int nl_object_match_filter(struct nl_object *, struct nl_object *); 285 286 %cstring_output_maxsize(char *buf, size_t len) 287 extern char *nl_object_attrs2str(struct nl_object *, uint32_t, char *buf, size_t len); 288 289 %cstring_output_maxsize(char *buf, size_t len) 290 extern char *nl_object_attr_list(struct nl_object *, char *buf, size_t len); 291 292 extern void nl_object_mark(struct nl_object *); 293 extern void nl_object_unmark(struct nl_object *); 294 extern int nl_object_is_marked(struct nl_object *); 295 296 extern int nl_object_get_refcnt(struct nl_object *); 297 298 /* <netlink/cache.h> */ 299 300 typedef void (*change_func_t)(struct nl_cache *, struct nl_object *, int, void *); 301 302 %inline %{ 303 struct nl_cache *alloc_cache_name(const char *name) 304 { 305 struct nl_cache *c; 306 if (nl_cache_alloc_name(name, &c) < 0) 307 return NULL; 308 return c; 309 } 310 311 struct nl_cache_mngr *alloc_cache_mngr(struct nl_sock *sock, 312 int protocol, int flags) 313 { 314 struct nl_cache_mngr *mngr; 315 316 if (nl_cache_mngr_alloc(sock, protocol, flags, &mngr) < 0) 317 return NULL; 318 319 return mngr; 320 } 321 322 struct nl_cache *cache_mngr_add(struct nl_cache_mngr *mngr, 323 const char *name, change_func_t func, 324 void *arg) 325 { 326 struct nl_cache *cache; 327 328 if (nl_cache_mngr_add(mngr, name, func, arg, &cache) < 0) 329 return NULL; 330 331 return cache; 332 } 333 %} 334 335 /* Access Functions */ 336 extern int nl_cache_nitems(struct nl_cache *); 337 extern int nl_cache_nitems_filter(struct nl_cache *, 338 struct nl_object *); 339 extern struct nl_cache_ops * nl_cache_get_ops(struct nl_cache *); 340 extern struct nl_object * nl_cache_get_first(struct nl_cache *); 341 extern struct nl_object * nl_cache_get_last(struct nl_cache *); 342 extern struct nl_object * nl_cache_get_next(struct nl_object *); 343 extern struct nl_object * nl_cache_get_prev(struct nl_object *); 344 345 extern struct nl_cache * nl_cache_alloc(struct nl_cache_ops *); 346 extern struct nl_cache * nl_cache_subset(struct nl_cache *, 347 struct nl_object *); 348 extern void nl_cache_clear(struct nl_cache *); 349 extern void nl_cache_free(struct nl_cache *); 350 351 /* Cache modification */ 352 extern int nl_cache_add(struct nl_cache *, 353 struct nl_object *); 354 extern int nl_cache_parse_and_add(struct nl_cache *, 355 struct nl_msg *); 356 extern void nl_cache_remove(struct nl_object *); 357 extern int nl_cache_refill(struct nl_sock *, 358 struct nl_cache *); 359 extern int nl_cache_pickup(struct nl_sock *, 360 struct nl_cache *); 361 extern int nl_cache_resync(struct nl_sock *, 362 struct nl_cache *, 363 change_func_t, 364 void *); 365 extern int nl_cache_include(struct nl_cache *, 366 struct nl_object *, 367 change_func_t, 368 void *); 369 extern void nl_cache_set_arg1(struct nl_cache *, int); 370 extern void nl_cache_set_arg2(struct nl_cache *, int); 371 372 /* General */ 373 extern int nl_cache_is_empty(struct nl_cache *); 374 extern struct nl_object * nl_cache_search(struct nl_cache *, 375 struct nl_object *); 376 extern void nl_cache_mark_all(struct nl_cache *); 377 378 /* Dumping */ 379 extern void nl_cache_dump(struct nl_cache *, 380 struct nl_dump_params *); 381 extern void nl_cache_dump_filter(struct nl_cache *, 382 struct nl_dump_params *, 383 struct nl_object *); 384 385 /* Iterators */ 386 extern void nl_cache_foreach(struct nl_cache *, 387 void (*cb)(struct nl_object *, 388 void *), 389 void *arg); 390 extern void nl_cache_foreach_filter(struct nl_cache *, 391 struct nl_object *, 392 void (*cb)(struct 393 nl_object *, 394 void *), 395 void *arg); 396 397 /* --- cache management --- */ 398 399 /* Cache type management */ 400 extern struct nl_cache_ops * nl_cache_ops_lookup(const char *); 401 extern struct nl_cache_ops * nl_cache_ops_associate(int, int); 402 extern struct nl_msgtype * nl_msgtype_lookup(struct nl_cache_ops *, int); 403 extern void nl_cache_ops_foreach(void (*cb)(struct nl_cache_ops *, void *), void *); 404 extern int nl_cache_mngt_register(struct nl_cache_ops *); 405 extern int nl_cache_mngt_unregister(struct nl_cache_ops *); 406 407 /* Global cache provisioning/requiring */ 408 extern void nl_cache_mngt_provide(struct nl_cache *); 409 extern void nl_cache_mngt_unprovide(struct nl_cache *); 410 extern struct nl_cache * nl_cache_mngt_require(const char *); 411 412 struct nl_cache_mngr; 413 414 #define NL_AUTO_PROVIDE 1 415 416 extern int nl_cache_mngr_get_fd(struct nl_cache_mngr *); 417 extern int nl_cache_mngr_poll(struct nl_cache_mngr *, 418 int); 419 extern int nl_cache_mngr_data_ready(struct nl_cache_mngr *); 420 extern void nl_cache_mngr_free(struct nl_cache_mngr *); 421 422 /* <netlink/addr.h> */ 423 %inline %{ 424 struct nl_addr *addr_parse(const char *addr, int guess) 425 { 426 struct nl_addr *result; 427 428 if (nl_addr_parse(addr, guess, &result) < 0) 429 return NULL; 430 431 return result; 432 } 433 %}; 434 435 extern struct nl_addr *nl_addr_alloc(size_t); 436 extern struct nl_addr *nl_addr_alloc_attr(struct nlattr *, int); 437 extern struct nl_addr *nl_addr_build(int, void *, size_t); 438 extern struct nl_addr *nl_addr_clone(struct nl_addr *); 439 440 extern struct nl_addr *nl_addr_get(struct nl_addr *); 441 extern void nl_addr_put(struct nl_addr *); 442 extern int nl_addr_shared(struct nl_addr *); 443 444 extern int nl_addr_cmp(struct nl_addr *, struct nl_addr *); 445 extern int nl_addr_cmp_prefix(struct nl_addr *, struct nl_addr *); 446 extern int nl_addr_iszero(struct nl_addr *); 447 extern int nl_addr_valid(char *, int); 448 extern int nl_addr_guess_family(struct nl_addr *); 449 extern int nl_addr_fill_sockaddr(struct nl_addr *, struct sockaddr *, socklen_t *); 450 extern int nl_addr_info(struct nl_addr *, struct addrinfo **); 451 extern int nl_addr_resolve(struct nl_addr *addr, char *host, size_t hostlen); 452 453 extern void nl_addr_set_family(struct nl_addr *, int); 454 extern int nl_addr_get_family(struct nl_addr *); 455 extern int nl_addr_set_binary_addr(struct nl_addr *, void *, size_t); 456 457 extern void *nl_addr_get_binary_addr(struct nl_addr *); 458 extern unsigned int nl_addr_get_len(struct nl_addr *); 459 extern void nl_addr_set_prefixlen(struct nl_addr *, int); 460 extern unsigned int nl_addr_get_prefixlen(struct nl_addr *); 461 462 %cstring_output_maxsize(char *buf, size_t len) 463 extern char *nl_af2str(int, char *buf, size_t len); 464 extern int nl_str2af(const char *); 465 466 %cstring_output_maxsize(char *buf, size_t len) 467 extern char *nl_addr2str(struct nl_addr *, char *buf, size_t len); 468 469 /* Message Handlers <netlink/handlers.h> */ 470 /** 471 * Callback actions 472 * @ingroup cb 473 */ 474 enum nl_cb_action { 475 /** Proceed with wathever would come next */ 476 NL_OK, 477 /** Skip this message */ 478 NL_SKIP, 479 /** Stop parsing altogether and discard remaining messages */ 480 NL_STOP, 481 }; 482 483 /** 484 * Callback kinds 485 * @ingroup cb 486 */ 487 enum nl_cb_kind { 488 /** Default handlers (quiet) */ 489 NL_CB_DEFAULT, 490 /** Verbose default handlers (error messages printed) */ 491 NL_CB_VERBOSE, 492 /** Debug handlers for debugging */ 493 NL_CB_DEBUG, 494 /** Customized handler specified by the user */ 495 NL_CB_CUSTOM, 496 __NL_CB_KIND_MAX, 497 }; 498 499 #define NL_CB_KIND_MAX (__NL_CB_KIND_MAX - 1) 500 501 /** 502 * Callback types 503 * @ingroup cb 504 */ 505 enum nl_cb_type { 506 /** Message is valid */ 507 NL_CB_VALID, 508 /** Last message in a series of multi part messages received */ 509 NL_CB_FINISH, 510 /** Report received that data was lost */ 511 NL_CB_OVERRUN, 512 /** Message wants to be skipped */ 513 NL_CB_SKIPPED, 514 /** Message is an acknowledge */ 515 NL_CB_ACK, 516 /** Called for every message received */ 517 NL_CB_MSG_IN, 518 /** Called for every message sent out except for nl_sendto() */ 519 NL_CB_MSG_OUT, 520 /** Message is malformed and invalid */ 521 NL_CB_INVALID, 522 /** Called instead of internal sequence number checking */ 523 NL_CB_SEQ_CHECK, 524 /** Sending of an acknowledge message has been requested */ 525 NL_CB_SEND_ACK, 526 /** Flag NLM_F_DUMP_INTR is set in message */ 527 NL_CB_DUMP_INTR, 528 __NL_CB_TYPE_MAX, 529 }; 530 531 #define NL_CB_TYPE_MAX (__NL_CB_TYPE_MAX - 1) 532 533 extern struct nl_cb *nl_cb_alloc(enum nl_cb_kind); 534 extern struct nl_cb *nl_cb_clone(struct nl_cb *); 535 536 struct nlmsgerr { 537 int error; 538 }; 539 540 %{ 541 542 struct pynl_callback { 543 PyObject *cbf; 544 PyObject *cba; 545 }; 546 547 struct pynl_cbinfo { 548 struct nl_cb *cb; 549 struct pynl_callback cbtype[NL_CB_TYPE_MAX+1]; 550 struct pynl_callback cberr; 551 struct list_head list; 552 }; 553 554 LIST_HEAD(callback_list); 555 556 static struct pynl_cbinfo *pynl_find_cbinfo(struct nl_cb *cb, int unlink) 557 { 558 struct list_head *pos, *prev; 559 struct pynl_cbinfo *info; 560 561 list_for_each_safe(pos, prev, &callback_list) { 562 info = container_of(pos, struct pynl_cbinfo, list); 563 if (info->cb == cb) { 564 if (unlink) 565 list_del(pos, prev); 566 pynl_dbg("cb=%p: found=%p\n", cb, info); 567 return info; 568 } 569 } 570 pynl_dbg("cb=%p: not found\n", cb); 571 return NULL; 572 } 573 574 static struct pynl_cbinfo *pynl_get_cbinfo(struct nl_cb *cb, int unlink) 575 { 576 struct pynl_cbinfo *info; 577 578 info = pynl_find_cbinfo(cb, unlink); 579 580 if (info || unlink) { 581 /* found or no need to allocate a new one */ 582 pynl_dbg("cb=%p: done\n", cb); 583 return info; 584 } 585 586 info = calloc(1, sizeof(*info)); 587 info->cb = cb; 588 list_add(&info->list, &callback_list); 589 pynl_dbg("cb=%p: added %p\n", cb, info); 590 return info; 591 } 592 593 static int nl_recv_msg_handler(struct nl_msg *msg, void *arg) 594 { 595 struct pynl_callback *cbd = arg; 596 PyObject *msgobj; 597 PyObject *cbparobj; 598 PyObject *resobj; 599 PyObject *funcobj; 600 int result; 601 602 if (!cbd) { 603 result = NL_STOP; 604 goto done; 605 } 606 msgobj = SWIG_NewPointerObj(SWIG_as_voidptr(msg), 607 SWIGTYPE_p_nl_msg, 0 | 0 ); 608 /* add selfobj if callback is a method */ 609 if (cbd->cbf && PyMethod_Check(cbd->cbf)) { 610 PyObject *selfobj = PyMethod_Self(cbd->cbf); 611 cbparobj = Py_BuildValue("(OOO)", selfobj ? selfobj : cbd->cba, 612 msgobj, cbd->cba); 613 funcobj = PyMethod_Function(cbd->cbf); 614 pynl_dbg("callback %sbounded instance method %p\n", 615 selfobj ? "" : "un", funcobj); 616 } else { 617 cbparobj = Py_BuildValue("(OO)", msgobj, cbd->cba); 618 funcobj = cbd->cbf; 619 pynl_dbg("callback function %p\n", funcobj); 620 } 621 resobj = PyObject_CallObject(funcobj, cbparobj); 622 Py_DECREF(cbparobj); 623 if (resobj && PyInt_Check(resobj)) 624 result = (int)PyInt_AsLong(resobj); 625 else 626 result = NL_STOP; 627 Py_XDECREF(resobj); 628 done: 629 pynl_dbg("result=%d\n", result); 630 return result; 631 } 632 633 static int nl_recv_err_handler(struct sockaddr_nl *nla, struct nlmsgerr *err, 634 void *arg) 635 { 636 struct pynl_callback *cbd = arg; 637 PyObject *errobj; 638 PyObject *cbparobj; 639 PyObject *resobj; 640 PyObject *funcobj; 641 int result; 642 643 if (!cbd) 644 return NL_STOP; 645 errobj = SWIG_NewPointerObj(SWIG_as_voidptr(err), 646 SWIGTYPE_p_nlmsgerr, 0 | 0 ); 647 /* add selfobj if callback is a method */ 648 if (cbd->cbf && PyMethod_Check(cbd->cbf)) { 649 PyObject *selfobj = PyMethod_Self(cbd->cbf); 650 cbparobj = Py_BuildValue("(OOO)", selfobj ? selfobj : cbd->cba, 651 errobj, cbd->cba); 652 funcobj = PyMethod_Function(cbd->cbf); 653 } else { 654 cbparobj = Py_BuildValue("(OO)", errobj, cbd->cba); 655 funcobj = cbd->cbf; 656 } 657 resobj = PyObject_CallObject(funcobj, cbparobj); 658 Py_DECREF(cbparobj); 659 if (resobj && PyInt_Check(resobj)) 660 result = (int)PyInt_AsLong(resobj); 661 else 662 result = NL_STOP; 663 Py_XDECREF(resobj); 664 pynl_dbg("error: err=%d ret=%d\n", err->error, result); 665 return result; 666 } 667 668 %} 669 %inline %{ 670 struct nl_cb *py_nl_cb_clone(struct nl_cb *cb) 671 { 672 struct pynl_cbinfo *info, *clone_info; 673 struct nl_cb *clone; 674 int i; 675 676 clone = nl_cb_clone(cb); 677 info = pynl_find_cbinfo(cb, 0); 678 if (info) { 679 clone_info = pynl_get_cbinfo(clone, 0); 680 /* increase refcnt to callback parameters and copy them */ 681 for (i = 0; info && i <= NL_CB_TYPE_MAX; i++) { 682 Py_XINCREF(info->cbtype[i].cbf); 683 Py_XINCREF(info->cbtype[i].cba); 684 clone_info->cbtype[i].cbf = info->cbtype[i].cbf; 685 clone_info->cbtype[i].cba = info->cbtype[i].cba; 686 } 687 Py_XINCREF(info->cberr.cbf); 688 Py_XINCREF(info->cberr.cba); 689 clone_info->cberr.cbf = info->cberr.cbf; 690 clone_info->cberr.cba = info->cberr.cba; 691 } 692 return clone; 693 } 694 695 void py_nl_cb_put(struct nl_cb *cb) 696 { 697 struct pynl_cbinfo *info; 698 int i; 699 700 /* obtain callback info (and unlink) */ 701 info = pynl_get_cbinfo(cb, 1); 702 pynl_dbg("cb=%p, info=%p\n", cb, info); 703 /* decrease refcnt for callback type handlers */ 704 for (i = 0; info && i <= NL_CB_TYPE_MAX; i++) { 705 Py_XDECREF(info->cbtype[i].cbf); 706 Py_XDECREF(info->cbtype[i].cba); 707 } 708 /* decrease refcnt for error handler and free callback info */ 709 if (info) { 710 Py_XDECREF(info->cberr.cbf); 711 Py_XDECREF(info->cberr.cba); 712 free(info); 713 } 714 nl_cb_put(cb); 715 } 716 717 int py_nl_cb_set(struct nl_cb *cb, enum nl_cb_type t, enum nl_cb_kind k, 718 PyObject *func, PyObject *a) 719 { 720 struct pynl_cbinfo *info; 721 722 /* obtain callback info */ 723 info = pynl_get_cbinfo(cb, 0); 724 725 /* clear existing handlers (if any) */ 726 Py_XDECREF(info->cbtype[t].cbf); 727 Py_XDECREF(info->cbtype[t].cba); 728 info->cbtype[t].cbf = NULL; 729 info->cbtype[t].cba = NULL; 730 pynl_dbg("cb=%p, info=%p, type=%d, kind=%d\n", cb, info, t, k); 731 /* handle custom callback */ 732 if (k == NL_CB_CUSTOM) { 733 Py_XINCREF(func); 734 Py_XINCREF(a); 735 info->cbtype[t].cbf = func; 736 info->cbtype[t].cba = a; 737 return nl_cb_set(cb, t, k, 738 nl_recv_msg_handler, &info->cbtype[t]); 739 } 740 return nl_cb_set(cb, t, k, NULL, NULL); 741 } 742 743 int py_nl_cb_set_all(struct nl_cb *cb, enum nl_cb_kind k, 744 PyObject *func , PyObject *a) 745 { 746 struct pynl_cbinfo *info; 747 int t; 748 749 info = pynl_get_cbinfo(cb, 0); 750 pynl_dbg("cb=%p, info=%p, kind=%d\n", cb, info, k); 751 for (t = 0; t <= NL_CB_TYPE_MAX; t++) { 752 /* (possibly) free existing handler */ 753 Py_XDECREF(info->cbtype[t].cbf); 754 Py_XDECREF(info->cbtype[t].cba); 755 info->cbtype[t].cbf = NULL; 756 info->cbtype[t].cba = NULL; 757 if (k == NL_CB_CUSTOM) { 758 Py_XINCREF(func); 759 Py_XINCREF(a); 760 info->cbtype[t].cbf = func; 761 info->cbtype[t].cba = a; 762 } 763 } 764 if (k == NL_CB_CUSTOM) 765 /* callback argument is same for all so using idx 0 here */ 766 return nl_cb_set_all(cb, k, nl_recv_msg_handler, 767 &info->cbtype[0]); 768 else 769 return nl_cb_set_all(cb, k, NULL, NULL); 770 } 771 772 int py_nl_cb_err(struct nl_cb *cb, enum nl_cb_kind k, 773 PyObject *func, PyObject *a) 774 { 775 struct pynl_cbinfo *info; 776 777 /* obtain callback info */ 778 info = pynl_get_cbinfo(cb, 0); 779 pynl_dbg("cb=%p, info=%p, kind=%d\n", cb, info, k); 780 /* clear existing handlers (if any) */ 781 Py_XDECREF(info->cberr.cbf); 782 Py_XDECREF(info->cberr.cba); 783 info->cberr.cbf = NULL; 784 info->cberr.cba = NULL; 785 786 /* handle custom callback */ 787 if (k == NL_CB_CUSTOM) { 788 Py_XINCREF(func); 789 Py_XINCREF(a); 790 info->cberr.cbf = func; 791 info->cberr.cba = a; 792 return nl_cb_err(cb, k, 793 nl_recv_err_handler, &info->cberr); 794 } 795 return nl_cb_err(cb, k, NULL, NULL); 796 } 797 %} 798 799 /* Attributes <netlink/attr.h> */ 800 /* 801 * This typemap is a bit tricky as it uses arg1, which is knowledge about 802 * the SWIGged wrapper output. 803 */ 804 %typemap(out) void * { 805 $result = PyByteArray_FromStringAndSize($1, nla_len(arg1)); 806 } 807 extern void *nla_data(struct nlattr *); 808 %typemap(out) void *; 809 extern int nla_type(const struct nlattr *); 810 811 /* Integer attribute */ 812 extern uint8_t nla_get_u8(struct nlattr *); 813 extern int nla_put_u8(struct nl_msg *, int, uint8_t); 814 extern uint16_t nla_get_u16(struct nlattr *); 815 extern int nla_put_u16(struct nl_msg *, int, uint16_t); 816 extern uint32_t nla_get_u32(struct nlattr *); 817 extern int nla_put_u32(struct nl_msg *, int, uint32_t); 818 extern uint64_t nla_get_u64(struct nlattr *); 819 extern int nla_put_u64(struct nl_msg *, int, uint64_t); 820 821 /* String attribute */ 822 extern char * nla_get_string(struct nlattr *); 823 extern char * nla_strdup(struct nlattr *); 824 extern int nla_put_string(struct nl_msg *, int, const char *); 825 826 /* Flag attribute */ 827 extern int nla_get_flag(struct nlattr *); 828 extern int nla_put_flag(struct nl_msg *, int); 829 830 /* Msec attribute */ 831 extern unsigned long nla_get_msecs(struct nlattr *); 832 extern int nla_put_msecs(struct nl_msg *, int, unsigned long); 833 834 /* Attribute nesting */ 835 extern int nla_put_nested(struct nl_msg *, int, struct nl_msg *); 836 extern struct nlattr * nla_nest_start(struct nl_msg *, int); 837 extern int nla_nest_end(struct nl_msg *, struct nlattr *); 838 %inline %{ 839 PyObject *py_nla_parse_nested(int max, struct nlattr *nest_attr, PyObject *p) 840 { 841 struct nlattr *tb_msg[max + 1]; 842 struct nla_policy *policy = NULL; 843 void *pol; 844 PyObject *attrs = Py_None; 845 PyObject *k; 846 PyObject *v; 847 PyObject *resobj; 848 int err; 849 int i; 850 851 if (p != Py_None) { 852 PyObject *pobj; 853 854 if (!PyList_Check(p)) { 855 fprintf(stderr, "expected list object\n"); 856 err = -1; 857 goto fail; 858 } 859 pobj = PyList_GetItem(p, 0); 860 err = SWIG_ConvertPtr(pobj, &pol, SWIGTYPE_p_nla_policy, 0 | 0 ); 861 if (!SWIG_IsOK(err)) 862 goto fail; 863 policy = pol; 864 } 865 err = nla_parse_nested(tb_msg, max, nest_attr, policy); 866 if (err < 0) { 867 fprintf(stderr, "Failed to parse response message\n"); 868 } else { 869 attrs = PyDict_New(); 870 for (i = 0; i <= max; i++) 871 if (tb_msg[i]) { 872 k = PyInt_FromLong((long)i); 873 v = SWIG_NewPointerObj(SWIG_as_voidptr(tb_msg[i]), SWIGTYPE_p_nlattr, 0 | 0 ); 874 PyDict_SetItem(attrs, k, v); 875 } 876 } 877 fail: 878 if (attrs == Py_None) 879 Py_INCREF(attrs); 880 resobj = Py_BuildValue("(iO)", err, attrs); 881 return resobj; 882 } 883 884 /* 885 * nla_get_nested() - get list of nested attributes. 886 * 887 * nla_for_each_<nested|attr>() is a macro construct that needs another approach 888 * for Python. Create and return list of nested attributes. 889 */ 890 PyObject *nla_get_nested(struct nlattr *nest_attr) 891 { 892 PyObject *listobj; 893 PyObject *nestattrobj; 894 struct nlattr *pos; 895 int rem; 896 897 listobj = PyList_New(0); 898 nla_for_each_nested(pos, nest_attr, rem) { 899 nestattrobj = SWIG_NewPointerObj(SWIG_as_voidptr(pos), 900 SWIGTYPE_p_nlattr, 0 | 0 ); 901 PyList_Append(listobj, nestattrobj); 902 } 903 return listobj; 904 } 905 %} 906 907 /** 908 * @ingroup attr 909 * Basic attribute data types 910 * 911 * See \ref attr_datatypes for more details. 912 */ 913 enum { 914 NLA_UNSPEC, /**< Unspecified type, binary data chunk */ 915 NLA_U8, /**< 8 bit integer */ 916 NLA_U16, /**< 16 bit integer */ 917 NLA_U32, /**< 32 bit integer */ 918 NLA_U64, /**< 64 bit integer */ 919 NLA_STRING, /**< NUL terminated character string */ 920 NLA_FLAG, /**< Flag */ 921 NLA_MSECS, /**< Micro seconds (64bit) */ 922 NLA_NESTED, /**< Nested attributes */ 923 __NLA_TYPE_MAX, 924 }; 925 926 #define NLA_TYPE_MAX (__NLA_TYPE_MAX - 1) 927 928 /** @} */ 929 930 /** 931 * @ingroup attr 932 * Attribute validation policy. 933 * 934 * See \ref attr_datatypes for more details. 935 */ 936 struct nla_policy { 937 /** Type of attribute or NLA_UNSPEC */ 938 uint16_t type; 939 940 /** Minimal length of payload required */ 941 uint16_t minlen; 942 943 /** Maximal length of payload allowed */ 944 uint16_t maxlen; 945 }; 946 947 %inline %{ 948 PyObject *nla_policy_array(int n_items) 949 { 950 struct nla_policy *policies; 951 PyObject *listobj; 952 PyObject *polobj; 953 int i; 954 955 policies = calloc(n_items, sizeof(*policies)); 956 listobj = PyList_New(n_items); 957 for (i = 0; i < n_items; i++) { 958 polobj = SWIG_NewPointerObj(SWIG_as_voidptr(&policies[i]), 959 SWIGTYPE_p_nla_policy, 0 | 0 ); 960 PyList_SetItem(listobj, i, polobj); 961 } 962 return listobj; 963 } 964 %} 965
