Fix sendTo test
[swifty.git] / src / raw / test / test_sw_sendto.c
1 /*
2  * Test sw_sendto "syscall".
3  */
4
5 #include <stdio.h>
6 #include <stdlib.h>
7 #include <string.h>
8 #include <sys/types.h>
9 #include <unistd.h>
10 #include <sys/socket.h>
11 #include <netinet/in.h>
12 #include <errno.h>
13
14 #include "swift_raw.h"
15 #include "swift_types.h"
16
17 #include "debug.h"
18 #include "util.h"
19
20 #include "test_sw.h"
21 #include "test.h"
22
23 static void sendto_dummy(void);
24 static void sendto_invalid_descriptor(void);
25 static void sendto_descriptor_is_not_socket(void);
26 static void sendto_socket_is_not_bound(void);
27 static void sendto_ok(void);
28
29 // Additional function
30 static void fill_sockaddr_sw(struct sockaddr_sw *local_addr, struct sockaddr_sw *remote_addr, char * local_address, char * hash, char * dest_address);
31
32 void sendto_test_suite(void)
33 {
34         start_suite();
35         sendto_dummy();
36
37         sendto_invalid_descriptor();
38         sendto_descriptor_is_not_socket();
39         sendto_socket_is_not_bound();
40         sendto_ok();
41 }
42
43 static void sendto_dummy(void)
44 {
45         test(1 == 1);
46 }
47
48 static void sendto_invalid_descriptor() 
49 {
50         struct sockaddr_sw local_addr;
51         struct sockaddr_sw remote_addr;
52         ssize_t bytes_sent;
53         char buffer[BUFSIZ];
54
55         fill_sockaddr_sw(&local_addr, &remote_addr, "127.0.0.1", "myHash", "127.0.0.1");
56         bytes_sent = sw_sendto(-1, buffer, BUFSIZ, 0, (struct sockaddr *) &remote_addr, sizeof(remote_addr));
57
58         test(bytes_sent < 0 && errno == EBADF);
59 }
60
61 static void sendto_descriptor_is_not_socket(void)
62 {
63         struct sockaddr_sw local_addr;
64         struct sockaddr_sw remote_addr;
65         ssize_t bytes_sent;
66         char buffer[BUFSIZ];
67
68         fill_sockaddr_sw(&local_addr, &remote_addr, "127.0.0.1", "myHash", "127.0.0.1");
69         bytes_sent = sw_sendto(1, buffer, BUFSIZ, 0, (struct sockaddr *) &remote_addr, sizeof(remote_addr));
70
71         test(bytes_sent < 0 && errno == ENOTSOCK);
72 }
73
74 static void sendto_socket_is_not_bound(void)
75 {
76         int sockfd;
77         struct sockaddr_sw local_addr;
78         struct sockaddr_sw remote_addr;
79         ssize_t bytes_sent;
80         char buffer[BUFSIZ];
81         
82         sockfd = sw_socket(PF_INET, SOCK_DGRAM, IPPROTO_SWIFT);
83         DIE(sockfd < 0, "sw_socket");   
84         
85         fill_sockaddr_sw(&local_addr, &remote_addr, "127.0.0.1", "myHash", "127.0.0.1");
86         bytes_sent = sw_sendto(sockfd, buffer, BUFSIZ, 0, (struct sockaddr *) &remote_addr, sizeof(remote_addr));
87
88         test( bytes_sent < 0 && errno == EAFNOSUPPORT );
89 }
90
91 static void sendto_ok(void) 
92 {
93         int sockfd;
94         struct sockaddr_sw local_addr;
95         struct sockaddr_sw remote_addr;
96         ssize_t bytes_sent;
97         char buffer[BUFSIZ];
98         int rc;
99
100         fill_sockaddr_sw(&local_addr, &remote_addr, "127.0.0.1", "myHash", "127.0.0.1");
101
102         sockfd = sw_socket(PF_INET, SOCK_DGRAM, IPPROTO_SWIFT);
103         DIE(sockfd < 0, "sw_socket");
104
105         rc = sw_bind(sockfd, (struct sockaddr *) &local_addr, sizeof(local_addr));
106         DIE(rc < 0, "sw_bind");
107
108         bytes_sent = sw_sendto(sockfd, buffer, BUFSIZ, 0, (struct sockaddr *) &remote_addr, sizeof(remote_addr));
109         test( bytes_sent >= 0 );
110 }
111
112 static void fill_sockaddr_sw(struct sockaddr_sw *local_addr, struct sockaddr_sw *remote_addr, char * local_address, char * hash, char * dest_address) 
113 {
114         local_addr->sin_addr.s_addr = htonl((int)local_address);
115         memcpy(&local_addr->sw_hash, hash, sizeof(struct sw_hash));
116
117         remote_addr->sin_addr.s_addr = htonl((int)dest_address);
118         memcpy(&remote_addr->sw_hash, hash, sizeof(struct sw_hash));
119 }