ipc/ipc.h at master · numpad/ipc · GitHub

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
#ifndef IPC_H
#define IPC_H

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/un.h>

/*
 * Inter Process Communication server
 */
typedef struct {
	unsigned int s, s2;
	struct sockaddr_un local, remote;
	unsigned int msglen;
} ipc_server;

/* create unix domain socket inter process communication server */
ipc_server ipc_create(const char *path) {
	ipc_server srv;
	srv.s = socket(AF_UNIX, SOCK_STREAM, 0);

	/* bind */
	srv.local.sun_family = AF_UNIX;
	strcpy(srv.local.sun_path, path);
	unlink(srv.local.sun_path);
	srv.msglen = strlen(srv.local.sun_path) + sizeof(srv.local.sun_family);

	bind(srv.s, (struct sockaddr *)&srv.local, srv.msglen);

	/* listen */
	listen(srv.s, 5);
	return srv;
}

/* accept ipc client */
void ipc_server_accept(ipc_server *srv) {
	srv->msglen = sizeof(srv->remote);
	srv->s2 = accept(srv->s, (struct sockaddr *)&(srv->remote), &srv->msglen);
}

/* read from ipc server */
int ipc_server_read(ipc_server *srv, char *buf, int buflen) {
	srv->msglen = recv(srv->s2, buf, buflen, 0);
	return srv->msglen > 0;
}

/* write to ipc client */
void ipc_server_send(ipc_server *srv, char *buf, int buflen) {
	send(srv->s2, buf, buflen, 0);
}

/* close srv */
void ipc_server_close(ipc_server srv) {
	close(srv.s2);
}

/*
 * Inter Process Communication client
 */

typedef enum {
	IPC_OK = 0,
	IPC_UNKNOWN,
	IPC_SOCKET_ERROR,
	IPC_CONNECT_FAILED
} ipc_status;

typedef struct {
	int s, t, len;
	struct sockaddr_un remote;

	/* status */
	ipc_status error;
} ipc_client;

ipc_client ipc_connect(const char *path) {
	ipc_client cli;
	cli.error = IPC_UNKNOWN;

	cli.s = socket(AF_UNIX, SOCK_STREAM, 0);
	if (cli.s == -1) {
		cli.error = IPC_SOCKET_ERROR;
		return cli;
	}

	cli.remote.sun_family = AF_UNIX;
	strcpy(cli.remote.sun_path, path);
	cli.len = strlen(cli.remote.sun_path) + sizeof(cli.remote.sun_family);
	if (connect(cli.s, (struct sockaddr *)&cli.remote, cli.len) == -1) {
		cli.error = IPC_CONNECT_FAILED;
		return cli;
	}

	cli.error = IPC_OK;
	return cli;
}

/* sends ipc data */
int ipc_client_send(ipc_client *cli, char *buf, int buflen) {
	return send(cli->s, buf, buflen, 0);
}

/* read ipc data */
int ipc_client_read(ipc_client *cli, char *buf, int buflen) {
	cli->t = recv(cli->s, buf, buflen, 0);
	return cli->t > 0;
}

/* close ipc socket */
void ipc_client_close(ipc_client cli) {
	close(cli.s);
}

#endif