24
|
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
|
|
11 #include "screen.h" // FIXME to be removed
|
|
12
|
|
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 {
|
|
47 //ut_WriteLog("Sending:%s\n", buffer);
|
|
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;
|
|
65 char mtag[16]; // For tag name
|
|
66
|
26
|
67 char *buffer = calloc(1, tambuffer);
|
|
68 char *retval = calloc(1, tambuffer + 1);
|
24
|
69
|
|
70 while (1) {
|
|
71 char *p1;
|
|
72 int n = recv(sock, buffer, tambuffer, 0);
|
|
73 if (n == -1) {
|
|
74 // Error
|
|
75 free(buffer);
|
|
76 retval[0] = 0;
|
|
77 return retval;
|
|
78 }
|
|
79
|
|
80 if (i == 1) {
|
|
81 char *p2;
|
26
|
82 strncpy(retval, buffer, tambuffer);
|
24
|
83 p1 = retval+1;
|
|
84 p2 = mtag;
|
|
85 while (('a' <= *p1) && (*p1 <= 'z') && (p2-mtag < 14))
|
|
86 *p2++ = *p1++;
|
|
87 *p2++ = '>'; *p2++ = 0;
|
|
88 //fprintf(stderr, "TAG=\"%s\"\n", mtag);
|
|
89 } else {
|
26
|
90 char *old_retval = retval;
|
24
|
91 scr_LogPrint("Realloc %d [%d]", i-1, n);
|
|
92 if (!n)
|
|
93 break;
|
|
94 retval = realloc(retval, (tambuffer * i) + 1);
|
26
|
95 if (!retval) {
|
|
96 scr_LogPrint("Memory allocation failure!!");
|
|
97 ut_WriteLog("Memory allocation failure in sk_recv()\n", retval);
|
|
98 free(buffer);
|
|
99 return old_retval;
|
|
100 }
|
|
101 strncat(retval, buffer, tambuffer);
|
24
|
102 }
|
|
103 i++;
|
|
104 p1 = retval + strlen(retval) - strlen(mtag);
|
|
105 //fprintf(stderr, "buffer:[%s]\n", buffer);
|
|
106 //fprintf(stderr, "End RET=[%s]\n", p1);
|
|
107 if (!strcmp(p1, mtag))
|
|
108 break;
|
|
109 for (p1 = retval; *p1 && (*p1 != '>'); p1++);
|
|
110 if ((*p1 == '>') && (*(p1-1) == '/'))
|
|
111 break;
|
|
112 memset(buffer, 0, tambuffer);
|
|
113 }
|
|
114 free(buffer);
|
|
115 ut_WriteLog("Received:%s\n", retval);
|
|
116 return retval;
|
|
117 }
|