Mercurial > hg
comparison mcabber/src/socket.c @ 24:e88b15cbf2de
[/trunk] Changeset 40 by mikael
* Change structure -> src directory for mcabber source code...
author | mikael |
---|---|
date | Sun, 27 Mar 2005 20:16:02 +0000 |
parents | |
children | 8588f5a4b638 |
comparison
equal
deleted
inserted
replaced
23:d7107507424b | 24:e88b15cbf2de |
---|---|
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 | |
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; | |
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 } | |
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 { | |
93 scr_LogPrint("Realloc %d [%d]", i-1, n); | |
94 if (!n) | |
95 break; | |
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 } |