-
Notifications
You must be signed in to change notification settings - Fork 108
/
kwmc.cpp
106 lines (85 loc) · 2.07 KB
/
kwmc.cpp
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
#include <iostream>
#include <string>
#include <libproc.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <netdb.h>
#include <unistd.h>
#define KwmDaemonPort 3020
int KwmcSockFD;
void Fatal(const std::string &err)
{
std::cout << err << std::endl;
exit(1);
}
std::string ReadFromSocket(int SockFD)
{
std::string Message;
char Cur;
while(recv(SockFD, &Cur, 1, 0))
Message += Cur;
return Message;
}
void WriteToSocket(std::string Msg)
{
Msg += "\n";
send(KwmcSockFD, Msg.c_str(), Msg.size(), 0);
std::string Response = ReadFromSocket(KwmcSockFD);
if(!Response.empty())
std::cout << Response << std::endl;
shutdown(KwmcSockFD, SHUT_RDWR);
close(KwmcSockFD);
}
void KwmcForwardMessageThroughSocket(int argc, char **argv)
{
std::string Msg;
for(int i = 1; i < argc; ++i)
{
Msg += argv[i];
if(i < argc - 1)
Msg += " ";
}
WriteToSocket(Msg);
}
void KwmcConnectToDaemon()
{
struct sockaddr_in srv_addr;
struct hostent *server;
if((KwmcSockFD = socket(PF_INET, SOCK_STREAM, 0)) == -1)
Fatal("Could not create socket!");
server = gethostbyname("localhost");
srv_addr.sin_family = AF_INET;
srv_addr.sin_port = htons(KwmDaemonPort);
std::memcpy(&srv_addr.sin_addr.s_addr, server->h_addr, server->h_length);
std::memset(&srv_addr.sin_zero, '\0', 8);
if(connect(KwmcSockFD, (struct sockaddr*) &srv_addr, sizeof(struct sockaddr)) == -1)
Fatal("Connection failed!");
}
void KwmcInterpreter()
{
while(true)
{
std::string Msg;
std::getline(std::cin, Msg);
if(Msg == "/quit" || Msg == "/q")
break;
KwmcConnectToDaemon();
WriteToSocket(Msg);
}
}
int main(int argc, char **argv)
{
if(argc >= 2)
{
std::string Command = argv[1];
if(Command == "interpret")
KwmcInterpreter();
else
{
KwmcConnectToDaemon();
KwmcForwardMessageThroughSocket(argc, argv);
}
}
return 0;
}