0
|
1 #include <stdio.h> |
|
2 #include <stdlib.h> |
|
3 #include <string.h> |
|
4 #include <sys/socket.h> |
|
5 #include <unistd.h> |
|
6 #include "utils.h" |
|
7 |
|
8 #include "socket.h" |
|
9 #include <signal.h> |
|
10 |
21
|
11 #include "screen.h" // FIXME to be removed |
|
12 |
0
|
13 /* Desc: create socket connection |
|
14 * |
|
15 * In : servername, port |
|
16 * Out : socket (or -1 on error) |
|
17 * |
|
18 * Note: - |
|
19 */ |
|
20 int sk_conn(struct sockaddr *name) |
|
21 { |
|
22 int sock; |
|
23 |
|
24 if ((sock = socket(PF_INET, SOCK_STREAM, 0)) < 0) { |
|
25 perror("socket (socket.c:23)"); |
|
26 return -1; |
|
27 } |
|
28 |
|
29 if (connect(sock, (struct sockaddr *) name, sizeof(struct sockaddr)) < 0) { |
|
30 perror("connect (socket.c:29)"); |
|
31 return -1; |
|
32 } |
|
33 |
|
34 return sock; |
|
35 } |
|
36 |
|
37 |
|
38 /* Desc: send data through socket |
|
39 * |
|
40 * In : socket, buffer to send |
|
41 * Out : 0 = fail, 1 = pass |
|
42 * |
|
43 * Note: - |
|
44 */ |
|
45 int sk_send(int sock, char *buffer) |
|
46 { |
19
|
47 //ut_WriteLog("Sending:%s\n", buffer); |
0
|
48 if ((send(sock, buffer, strlen(buffer), 0)) == -1) |
|
49 return 0; |
|
50 else |
|
51 return 1; |
|
52 } |
|
53 |
|
54 /* Desc: receive data through socket |
|
55 * |
|
56 * In : socket |
|
57 * Out : received buffer |
|
58 * |
|
59 * Note: it is up to the caller to free the returned string |
|
60 */ |
|
61 char *sk_recv(int sock) |
|
62 { |
|
63 int i = 1; |
|
64 int tambuffer = 128; |
21
|
65 char mtag[16]; // For tag name |
0
|
66 |
|
67 char *buffer = malloc(tambuffer); |
|
68 char *retval = malloc(tambuffer + 1); |
|
69 |
|
70 memset(retval, 0, tambuffer); |
|
71 memset(buffer, 0, tambuffer + 1); |
|
72 |
|
73 while (1) { |
|
74 char *p1; |
21
|
75 int n = recv(sock, buffer, tambuffer, 0); |
|
76 if (n == -1) { |
|
77 // Error |
|
78 free(buffer); |
|
79 retval[0] = 0; |
|
80 return retval; |
|
81 } |
0
|
82 |
|
83 if (i == 1) { |
|
84 char *p2; |
|
85 strcpy(retval, buffer); |
|
86 p1 = retval+1; |
|
87 p2 = mtag; |
|
88 while (('a' <= *p1) && (*p1 <= 'z') && (p2-mtag < 14)) |
|
89 *p2++ = *p1++; |
|
90 *p2++ = '>'; *p2++ = 0; |
|
91 //fprintf(stderr, "TAG=\"%s\"\n", mtag); |
|
92 } else { |
21
|
93 scr_LogPrint("Realloc %d [%d]", i-1, n); |
|
94 if (!n) |
|
95 break; |
0
|
96 retval = realloc(retval, (tambuffer * i) + 1); |
|
97 strncat(retval, buffer, tambuffer + 1); |
|
98 } |
|
99 i++; |
|
100 p1 = retval + strlen(retval) - strlen(mtag); |
|
101 //fprintf(stderr, "buffer:[%s]\n", buffer); |
|
102 //fprintf(stderr, "End RET=[%s]\n", p1); |
|
103 if (!strcmp(p1, mtag)) |
|
104 break; |
|
105 for (p1 = retval; *p1 && (*p1 != '>'); p1++); |
|
106 if ((*p1 == '>') && (*(p1-1) == '/')) |
|
107 break; |
|
108 memset(buffer, 0, tambuffer); |
|
109 } |
|
110 free(buffer); |
|
111 ut_WriteLog("Received:%s\n", retval); |
|
112 return retval; |
|
113 } |