| 0,0 → 1,294 |
|---|
| #include <sys/socket.h> |
| #include <menuet/net.h> |
| |
| #include "tp.h" |
| |
| extern tp_obj tp_dict(TP); |
| extern tp_obj tp_method(TP,tp_obj self,tp_obj v(TP)); |
| extern tp_obj tp_fnc(TP,tp_obj v(TP)); |
| |
| #define GET_SOCKET_DESCRIPTOR(_obj, _sock) do{ \ |
| if (!tp_has(tp, _obj, tp_string("socket"))) \ |
| tp_raise(tp_None, "Socket not open", tp_None); \ |
| _sock = (__u32)tp_get(tp, _obj, tp_string("socket")).number.val;\ |
| } while(0) |
| |
| /* Socket close method. |
| * |
| * Example: |
| * s.close() # s must be a socket object created by socket. |
| * |
| * Raises exception if socket was not opened. Otherwise returns True. |
| */ |
| tp_obj kolibri_close_socket(TP) |
| { |
| tp_obj self = TP_TYPE(TP_DICT); |
| __u32 socktype; |
| __u32 s; |
| |
| GET_SOCKET_DESCRIPTOR(self, s); |
| |
| con_printf("cp5\n"); |
| socktype = (__u32)tp_get(tp, self, tp_string("type")).number.val; |
| con_printf("cp6\n"); |
| if (!tp_has(tp, self, tp_string("socket"))) |
| { |
| con_printf("cp7\n"); |
| tp_raise(tp_None, "Socket not open", tp_None); |
| } |
| s = (__u32)tp_get(tp, self, tp_string("socket")).number.val; |
| con_printf("cp0, s=%d\n", s); |
| if (socktype == SOCK_STREAM) |
| { |
| con_printf("cp1\n"); |
| __menuet__close_TCP_socket(s); |
| con_printf("cp2\n"); |
| } |
| else if (socktype == SOCK_DGRAM) |
| { |
| __menuet__close_UDP_socket(s); |
| } |
| return tp_True; |
| } |
| |
| /* Socket send method. |
| * |
| * Example: |
| * data="<html><head><title>Preved!!!</title></head><body>Example.</body></html>" |
| * s.send(data) |
| * or: |
| * s.send(data, 20) # Send just 20 bytes |
| */ |
| tp_obj kolibri_send(TP) |
| { |
| tp_obj self = TP_TYPE(TP_DICT); |
| tp_obj data_obj = TP_TYPE(TP_STRING); |
| __u32 datalen = TP_DEFAULT(tp_False).number.val; |
| __u32 socktype = (__u32)tp_get(tp, self, tp_string("type")).number.val; |
| __u32 s; |
| |
| GET_SOCKET_DESCRIPTOR(self, s); |
| |
| if (datalen < 0 || datalen > data_obj.string.len) |
| datalen = data_obj.string.len; |
| if (socktype == SOCK_STREAM) |
| __menuet__write_TCP_socket(s, datalen, (void *)data_obj.string.val); |
| else if (socktype == SOCK_DGRAM) |
| __menuet__write_UDP_socket(s, datalen, (void *)data_obj.string.val); |
| return tp_None; |
| } |
| |
| /* Socket recv method. |
| * |
| * data="<html><head><title>Preved!!!</title></head><body>Example.</body></html>" |
| * s.recv(data) |
| * or: |
| * s.recv(data, 20) # Send just 20 bytes |
| */ |
| tp_obj kolibri_recv(TP) |
| { |
| tp_obj self = TP_TYPE(TP_DICT); |
| tp_obj data_obj = TP_TYPE(TP_STRING); |
| __u32 datalen = TP_DEFAULT(tp_False).number.val; |
| __u32 s; |
| __u8 c; |
| __u8 *buf, *p; |
| __u32 buf_size; |
| __u32 bytes_read = 0; |
| int i; |
| |
| GET_SOCKET_DESCRIPTOR(self, s); |
| |
| if (datalen) |
| buf_size = datalen; |
| else |
| buf_size = 2048; |
| if (!(buf = malloc(datalen))) |
| tp_raise(tp_None, "Cannot allocate buffer for received data", tp_None); |
| p = buf; |
| while (__menuet__read_socket(s, &c) && bytes_read < buf_size) |
| { |
| *p++ = c; |
| bytes_read++; |
| if (bytes_read >= buf_size && !datalen) |
| { |
| buf_size += 1024; |
| buf = realloc(buf, buf_size); |
| } |
| } |
| return tp_string_n(buf, bytes_read); |
| } |
| |
| void inet_pton(TP, const char *buf, int len, __u32 *addr) |
| { |
| char *p = (char *)buf; |
| int i = 0; |
| __u32 val = 0; |
| *addr = 0; |
| while (*p && p < buf + len && i < 4) |
| { |
| con_printf("char %d", *p); |
| if (*p == '.') |
| { |
| if (val > 255) |
| tp_raise(, "ValueError: number > 255 in IP address", tp_None); |
| *addr += (val << ((i++) << 3)); |
| con_printf("num %d\n", val); |
| val = 0; |
| } |
| else |
| { |
| if (*p < '0' || *p > '9') |
| tp_raise(, "ValueError: bad char in IP address, digit expected", tp_None); |
| val = val * 10 + *p - '0'; |
| } |
| p++; |
| } |
| } |
| |
| /* Converter from string presentation to binary address. */ |
| tp_obj kolibri_inet_pton(TP) |
| { |
| tp_obj obj; |
| __u32 addr; |
| con_printf("A1\n"); |
| obj = TP_TYPE(TP_STRING); |
| con_printf("A2\n"); |
| inet_pton(tp, (char *)obj.string.val, (int)obj.string.len, &addr); |
| con_printf("A3\n"); |
| return tp_number(addr); |
| } |
| |
| /* Socket bind method. |
| * |
| * In KolibriOS it just sets local address and port. |
| * |
| * Example: |
| * s.bind('10.10.1.2', 6000) #Connects to 10.10.1.2:6000 |
| */ |
| tp_obj kolibri_bind(TP) |
| { |
| tp_obj self = TP_TYPE(TP_DICT); |
| tp_obj local_addr_obj = TP_OBJ(); |
| __u32 local_port = (__u32)TP_TYPE(TP_NUMBER).number.val; |
| __u32 local_addr; |
| |
| if (local_addr_obj.type == TP_NUMBER) |
| local_addr = local_addr_obj.number.val; |
| else if (local_addr_obj.type == TP_STRING) |
| inet_pton(tp, (const char *)local_addr_obj.string.val, local_addr_obj.string.len, &local_addr); |
| |
| tp_set(tp, self, tp_string("local_addr"), tp_number(local_addr)); |
| tp_set(tp, self, tp_string("local_port"), tp_number(local_port)); |
| return tp_None; |
| } |
| |
| /* Socket connect method. |
| * |
| * Example: |
| * s.connect('10.10.1.1', 7000) #Connects to 10.10.1.1:7000 |
| */ |
| tp_obj kolibri_connect(TP) |
| { |
| tp_obj self = TP_TYPE(TP_DICT); |
| tp_obj remote_addr_obj = TP_OBJ(); |
| __u32 remote_addr; |
| __u32 remote_port = (__u32)TP_TYPE(TP_NUMBER).number.val; |
| __u32 local_port = tp_get(tp, self, tp_string("local_port")).number.val; |
| __u32 socktype = (__u32)tp_get(tp, self, tp_string("type")).number.val; |
| int s = -1; /* Socket descriptor */ |
| |
| |
| if (remote_addr_obj.type == TP_NUMBER) |
| remote_addr = remote_addr_obj.number.val; |
| else if (remote_addr_obj.type == TP_STRING) |
| inet_pton(tp, (const char *)remote_addr_obj.string.val, remote_addr_obj.string.len, &remote_addr); |
| |
| if (socktype == SOCK_STREAM) |
| s = __menuet__open_TCP_socket(local_port, remote_port, remote_addr, 1); |
| else if (socktype == SOCK_DGRAM) |
| s = __menuet__open_UDP_socket(local_port, remote_port, remote_addr); |
| if (s >= 0) |
| { |
| tp_set(tp, self, tp_string("socket"), tp_number(s)); |
| return tp_True; |
| } |
| else |
| return tp_False; |
| } |
| |
| /* Socket listen method. |
| * |
| * Example: |
| * s.listen('10.10.1.1', 5000) |
| */ |
| tp_obj kolibri_listen(TP) |
| { |
| tp_obj self = TP_TYPE(TP_DICT); |
| tp_obj remote_addr_obj = TP_OBJ(); |
| __u32 remote_addr; |
| __u32 remote_port = (__u32)TP_TYPE(TP_NUMBER).number.val; |
| __u32 local_port = tp_get(tp, self, tp_string("local_port")).number.val; |
| __u32 socktype = (__u32)tp_get(tp, self, tp_string("type")).number.val; |
| int s = -1; /* Socket descriptor */ |
| |
| if (socktype != SOCK_STREAM) |
| tp_raise(tp_None, "IOError: attempt to listen on non-TCP socket", tp_None); |
| |
| if (remote_addr_obj.type == TP_NUMBER) |
| remote_addr = remote_addr_obj.number.val; |
| else if (remote_addr_obj.type == TP_STRING) |
| inet_pton(tp, (const char *)remote_addr_obj.string.val, remote_addr_obj.string.len, &remote_addr); |
| |
| if ((s = __menuet__open_TCP_socket(local_port, remote_port, remote_addr, 0)) >= 0) |
| { |
| tp_set(tp, self, tp_string("socket"), tp_number(s)); |
| return tp_True; |
| } |
| else |
| return tp_False; |
| } |
| |
| |
| /* Exported function. |
| * |
| * Example: |
| * |
| * s = socket(socket.AF_INET, socket.SOCK_DGRAM) |
| * |
| * Returns socket object. |
| */ |
| tp_obj kolibri_socket(TP) |
| { |
| tp_obj s; |
| tp_obj sockfamily = TP_TYPE(TP_NUMBER); |
| tp_obj socktype = TP_TYPE(TP_NUMBER); |
| |
| if (abs(sockfamily.number.val - AF_INET) > 0.000001 || |
| (abs(socktype.number.val - SOCK_STREAM) > 0.000001 && |
| abs(socktype.number.val - SOCK_DGRAM) > 0.000001)) |
| return tp_None; |
| s = tp_dict(tp); |
| tp_set(tp, s, tp_string("family"), sockfamily); |
| tp_set(tp, s, tp_string("type"), socktype); |
| tp_set(tp, s, tp_string("bind"), tp_method(tp, s, kolibri_bind)); |
| tp_set(tp, s, tp_string("connect"), tp_method(tp, s, kolibri_connect)); |
| tp_set(tp, s, tp_string("send"), tp_method(tp, s, kolibri_send)); |
| tp_set(tp, s, tp_string("recv"), tp_method(tp, s, kolibri_recv)); |
| tp_set(tp, s, tp_string("close"), tp_method(tp, s, kolibri_close_socket)); |
| if (abs(socktype.number.val - SOCK_STREAM) < 0.000001) |
| tp_set(tp, s, tp_string("listen"), tp_method(tp, s, kolibri_listen)); |
| return s; |
| } |
| |
| tp_obj kolibri_socket_module(TP) |
| { |
| tp_obj socket_mod = tp_dict(tp); |
| |
| tp_set(tp, socket_mod, tp_string("AF_INET"), tp_number(AF_INET)); |
| tp_set(tp, socket_mod, tp_string("SOCK_STREAM"), tp_number(SOCK_STREAM)); |
| tp_set(tp, socket_mod, tp_string("SOCK_DGRAM"), tp_number(SOCK_DGRAM)); |
| tp_set(tp, socket_mod, tp_string("socket"), tp_fnc(tp, kolibri_socket)); |
| tp_set(tp, socket_mod, tp_string("inet_pton"), tp_fnc(tp, kolibri_inet_pton)); |
| return socket_mod; |
| } |