One Level Up Top Level

runtime/lib_net.c - ktap

Global variables defined

Functions defined

Source code

  1. /*
  2. * lib_base.c - base library
  3. *
  4. * This file is part of ktap by Jovi Zhangwei.
  5. *
  6. * Copyright (C) 2012-2013 Jovi Zhangwei <jovi.zhangwei@gmail.com>.
  7. *
  8. * ktap is free software; you can redistribute it and/or modify it
  9. * under the terms and conditions of the GNU General Public License,
  10. * version 2, as published by the Free Software Foundation.
  11. *
  12. * ktap is distributed in the hope it will be useful, but WITHOUT
  13. * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  14. * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  15. * more details.
  16. *
  17. * You should have received a copy of the GNU General Public License along with
  18. * this program; if not, write to the Free Software Foundation, Inc.,
  19. * 51 Franklin St - Fifth Floor, Boston, MA 02110-1301 USA.
  20. */

  22. #include <net/inet_sock.h>
  23. #include "../include/ktap_types.h"
  24. #include "ktap.h"
  25. #include "kp_obj.h"
  26. #include "kp_str.h"
  27. #include "kp_vm.h"

  29. /**
  30. * Return the source IP address for a given sock
  31. */
  32. ‌static int kplib_net_ip_sock_saddr(ktap_state_t *ks)
  33. {
  34.     struct inet_sock *isk;
  35.     int family;

  37.     /* TODO: need to validate the address firstly */

  39.     isk = (struct inet_sock *)kp_arg_checknumber(ks, 1);
  40.     family = isk->sk.__sk_common.skc_family;

  42.     if (family == AF_INET) {
  43.         set_number(ks->top, isk->inet_rcv_saddr);
  44.     } else {
  45.         kp_error(ks, "ip_sock_saddr only support ipv4 now\n");
  46.         set_nil(ks->top);
  47.     }

  49.     incr_top(ks);
  50.     return 1;
  51. }

  53. /**
  54. * Return the destination IP address for a given sock
  55. */
  56. ‌static int kplib_net_ip_sock_daddr(ktap_state_t *ks)
  57. {
  58.     struct inet_sock *isk;
  59.     int family;

  61.     /* TODO: need to validate the address firstly */

  63.     isk = (struct inet_sock *)kp_arg_checknumber(ks, 1);
  64.     family = isk->sk.__sk_common.skc_family;

  66.     if (family == AF_INET) {
  67.         set_number(ks->top, isk->inet_daddr);
  68.     } else {
  69.         kp_error(ks, "ip_sock_daddr only support ipv4 now\n");
  70.         set_nil(ks->top);
  71.     }

  73.     incr_top(ks);
  74.     return 1;

  76. }

  78. /**
  79. * Returns a string representation for an IP address
  80. */
  81. ‌static int kplib_net_format_ip_addr(ktap_state_t *ks)
  82. {
  83.     __be32 ip = (__be32)kp_arg_checknumber(ks, 1);
  84.     ktap_str_t *ts;
  85.     char ipstr[32];

  87.     snprintf(ipstr, 32, "%pI4", &ip);
  88.     ts = kp_str_newz(ks, ipstr);
  89.     if (ts) {
  90.         set_string(ks->top, kp_str_newz(ks, ipstr));
  91.         incr_top(ks);
  92.         return 1;
  93.     } else
  94.         return -1;
  95. }

  97. ‌static const ktap_libfunc_t net_lib_funcs[] = {
  98.     {"ip_sock_saddr", kplib_net_ip_sock_saddr},
  99.     {"ip_sock_daddr", kplib_net_ip_sock_daddr},
  100.     {"format_ip_addr", kplib_net_format_ip_addr},
  101.     {NULL}
  102. };

  104. ‌int kp_lib_init_net(ktap_state_t *ks)
  105. {
  106.     return kp_vm_register_lib(ks, "net", net_lib_funcs);
  107. }

One Level Up Top Level