Ticket #5716: test_client.cpp

File test_client.cpp, 1.6 KB (added by atis.elsts, 12 years ago)
Line 
1#include <unistd.h>
2#include <memory.h>
3#include <netinet/in.h>
4#include <arpa/inet.h>
5#include <errno.h>
6#include <stdio.h>
7#include <stdlib.h>
8
9const unsigned short TEST_PORT = 40000;
10
11void usage() {
12 printf("client [tcp|udp] [4|6] [4|6]\n");
13 exit(1);
14}
15
16int main(int argc, char *argv[]) {
17 int socketType = SOCK_DGRAM;
18 int socketFamily1 = AF_INET;
19 int socketFamily2 = AF_INET;
20
21 if (argc > 1) {
22 if (!strcmp(argv[1], "tcp")) {
23 socketType = SOCK_STREAM;
24 } else if (!strcmp(argv[1], "udp")) {
25 socketType = SOCK_DGRAM;
26 } else {
27 usage();
28 }
29 }
30 if (argc > 2) {
31 switch (atoi(argv[2])) {
32 case 4:
33 socketFamily1 = AF_INET;
34 break;
35 case 6:
36 socketFamily1 = AF_INET6;
37 break;
38 default:
39 usage();
40 }
41 }
42 if (argc > 3) {
43 switch (atoi(argv[3])) {
44 case 4:
45 socketFamily2 = AF_INET;
46 break;
47 case 6:
48 socketFamily2 = AF_INET6;
49 break;
50 default:
51 usage();
52 }
53 }
54
55 int fd = socket(socketFamily1, socketType, 0);
56 if (fd < 0) {
57 perror("socket");
58 return -1;
59 }
60
61 sockaddr_storage saddr;
62 memset(&saddr, 0, sizeof(saddr));
63 saddr.ss_family = socketFamily2;
64 ((sockaddr_in *) &saddr)->sin_port = htons(TEST_PORT);
65 if (connect(fd, (sockaddr *) &saddr, socketFamily2 == AF_INET ?
66 sizeof(sockaddr_in) : sizeof(sockaddr_in6)) < 0) {
67 perror("connect");
68 close(fd);
69 return -1;
70 }
71
72 const char *buffer = "hello world";
73 unsigned len = strlen(buffer);
74 int ret = send(fd, buffer, len, 0);
75 if (ret < len) {
76 if (ret < 0) {
77 perror("send");
78 } else if (ret == 0) {
79 printf("no data sent!\n");
80 } else {
81 printf("not all data sent!\n");
82 }
83 } else {
84 printf("send(): success\n");
85 }
86 close(fd);
87 return 0;
88}