1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
#include "ipc.h"
ipc_sender *setup_ipc_sender(str addr, int backlog){
ipc_sender *is = calloc(1, sizeof(ipc_sender));
is->addr = dup_str(addr);
is->ssocket = socket(AF_UNIX, SOCK_STREAM, 0);
if(is->ssocket == -1){
log_error("%s: socket: %s", __FUNCTION__, strerror(errno));
goto error;
}
struct sockaddr_un sockaddr = { .sun_family = AF_UNIX };
memcpy(sockaddr.sun_path, is->addr.ptr, is->addr.len);
sockaddr.sun_path[is->addr.len] = '\0';
unlink(is->addr.ptr);
if(bind(is->ssocket, (struct sockaddr*)&sockaddr, sizeof(sockaddr)) == -1){
log_error("%s: bind: %s", __FUNCTION__, strerror(errno));
goto error;
}
if(listen(is->ssocket, backlog) == -1){
log_error("%s: listen: %s", __FUNCTION__, strerror(errno));
goto error;
}
if(0){
error:
destroy_ipc_sender(&is);
}
return is;
}
void destroy_ipc_sender(ipc_sender **is){
if(*is != NULL){
close((*is)->ssocket);
(*is)->ssocket = -1;
unlink((*is)->addr.ptr);
free_str(&(*is)->addr);
free(*is);
*is = NULL;
}
}
ipc_listener *setup_ipc_listener(str saddr){
ipc_listener *il = calloc(1, sizeof(ipc_listener));
il->saddr = dup_str(saddr);
il->csocket = socket(AF_UNIX, SOCK_STREAM, 0);
if(il->csocket == -1){
log_error("%s: socket: %s", __FUNCTION__, strerror(errno));
goto error;
}
struct sockaddr_un socksaddr = { .sun_family = AF_UNIX };
memcpy(socksaddr.sun_path, il->saddr.ptr, il->saddr.len);
socksaddr.sun_path[il->saddr.len] = '\0';
if(connect(il->csocket, (struct sockaddr *)&socksaddr, sizeof(socksaddr)) == -1){
log_error("%s: connect: %s", __FUNCTION__, strerror(errno));
goto error;
}
if(0){
error:
destroy_ipc_listener(&il);
}
return il;
}
void destroy_ipc_listener(ipc_listener **il){
if(*il != NULL){
close((*il)->csocket);
(*il)->csocket = -1;
free_str(&(*il)->saddr);
free(*il);
*il = NULL;
}
}
int send_ipc_message(int to, ipc_type type, str msg){
if(send(to, &type, sizeof(uint8_t), 0) -1){
log_error("Can't send message type to socket %d: %s", to, strerror(errno));
return 1;
}
if(send(to, &msg.len, sizeof(msg.len), 0) == -1){
log_error("Can't send message length to socket %d: %s", to, strerror(errno));
return 1;
}
if(send(to, msg.ptr, msg.len, 0) == -1){
log_error("Can't send message to socket %d: %s", to, strerror(errno));
return 1;
}
char ack[3];
if(recv(to, ack, 3, 0) == -1){
log_error("Receiving ACK from listener");
return 1;
}
if(strncmp(ack, "ACK", 3) != 0){
log_error("Received '%.3s' from listener instead of 'ACK'", ack);
return 1;
}
return 0;
}
ipc_msg receive_ipc_message(ipc_listener *il){
ipc_msg msg = {0};
if(recv(il->csocket, &msg.type, sizeof(uint8_t), 0) == -1){
log_error("Can't receive message type from socket %d: %s", il->csocket, strerror(errno));
goto end;
}
if(recv(il->csocket, &msg.msg.len, sizeof(msg.msg.len), 0) == -1){
log_error("Can't receive message length from socket %d: %s", il->csocket, strerror(errno));
goto end;
}
msg.msg.cap = msg.msg.len;
msg.msg.ptr = calloc(msg.msg.len, sizeof(char));
if(recv(il->csocket, msg.msg.ptr, msg.msg.len, 0) == -1){
log_error("Can't receive message from socket %d: %s", il->csocket, strerror(errno));
free_ipc_message(&msg);
goto end;
}
end:
if(send(il->csocket, "ACK", slen("ACK"), 0) == -1){
log_error("Sending 'ACK' to sender");
}
return msg;
}
void free_ipc_message(ipc_msg *msg){
msg->type = NONE;
free_str(&msg->msg);
}
|
