|
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
#include "check_csgo.h"
using namespace std;
char *servicename = (char*)"CS:GO";
void printVersion()
{
cout << "check_csgo v" << VERSION << endl << endl;
}
void printHelp(bool longVersion)
{
if(longVersion)
{
printVersion();
|
|
|
17
|
cout << "Check CS:GO DS instance." << endl << endl;
|
|
|
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
printHelp(false);
cout << "Options:" << endl;
cout << " -h" << endl;
cout << " Print detailed help screen" << endl;
cout << " -V" << endl;
cout << " Print version information" << endl;
cout << " -H HOSTADDRESS" << endl;
cout << " Host where the Source DS is running" << endl;
cout << " -p" << endl;
cout << " Port where the Source DS is listening. Default is 27015." << endl << endl;
return;
}
cout << "Usage: " << endl << "check_csgo [-hV] -H HOSTADDRESS [-p PORT]" << endl << endl;
}
|
|
|
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
SOURCEDS_PACKET* getSourceDSResponse(int s, sockaddr_in *server)
{
char *data = new char[DATAGRAM_LENGTH];
int dataLength = recvMsg(s, data, DATAGRAM_LENGTH, server);
if(dataLength == -1)
{
return NULL;
}
char *header = (char*) "\xFF\xFF\xFF\xFF";
if(memcmp(data,header,HDR_SIZE))
{
//Invalid response, or non-supported multi-packet response format
return NULL;
}
SOURCEDS_PACKET *response = new SOURCEDS_PACKET;
response->data = data+HDR_SIZE;
response->length = dataLength;
return response;
}
|
|
|
53
54
|
int check_csgo(char *hostname, uint16_t port, SERVERINFO *server_info)
{
|
|
|
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
|
struct sockaddr_in si;
struct hostent *host = NULL;
int s;
int timeout = 10;
setupSocket(port,hostname,host,timeout,&si,&s);
string discard;
char *info_query = (char*) "\xFF\xFF\xFF\xFF\x54Source Engine Query\x00";
sendMsg(s,info_query, strlen(info_query)+1,&si);
SOURCEDS_PACKET *response = getSourceDSResponse(s,&si);
if(response == NULL)
{
return 2;
}
ssize_t offset = 2;
server_info->name = string(response->data+offset,strlen(response->data+offset));
offset += server_info->name.size()+1;
server_info->map = string(response->data+offset,strlen(response->data+offset));
offset += server_info->map.size()+1;
discard = string(response->data+offset,strlen(response->data+offset));
offset += discard.size() + 1;
server_info->game = string(response->data+offset,strlen(response->data+offset));
offset += server_info->game.size()+1;
offset += 2;
server_info->players = (uint8_t) *((response->data)+offset);
|
|
|
82
|
server_info->max_players = (uint8_t) *((response->data)+(++offset));
|
|
|
83
84
85
86
|
delete[] (response->data-HDR_SIZE);
delete response;
|
|
|
87
88
89
90
91
|
if(server_info->players == server_info->max_players)
{
return 1;
}
|
|
|
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
|
return 0;
}
int main(int argc, char **argv)
{
struct itimerval timer;
timer.it_value.tv_sec = 10;
timer.it_value.tv_usec = 0;
timer.it_interval.tv_sec = 0;
timer.it_interval.tv_usec = 0;
setitimer (ITIMER_VIRTUAL, &timer, 0);
struct sigaction sa;
memset (&sa, 0, sizeof (sa));
sa.sa_handler = &timer_handler;
sigaction (SIGVTALRM, &sa, 0);
uint16_t port = 27015;
char *hostname = NULL;
int c;
while ((c = getopt (argc, argv, "H:p:Vh")) != -1)
{
switch(c)
{
case 'H':
hostname = optarg;
break;
case 'p':
|
|
|
121
|
port = (uint16_t) str2int(optarg);
|
|
|
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
break;
case 'V':
printVersion();
return 0;
case 'h':
printHelp(true);
return 0;
case '?':
printHelp(false);
return 3;
}
}
if(hostname == NULL)
{
cout << "No HOSTADDRESS specified. Exiting." << endl;
return 3;
}
SERVERINFO server_info;
int returnCode = check_csgo(hostname,port,&server_info);
cout << servicename;
switch(returnCode)
{
case 0:
cout << " OK";
|
|
|
149
150
|
cout << " - " << hostname << " " << server_info.name << " " << server_info.game << " " << server_info.map
<< " " << (int)server_info.players << "/" << (int)server_info.max_players << " players" << endl ;
|
|
|
151
152
|
break;
|
|
|
153
154
155
156
|
case 1:
cout << " WARNING - Server is full";
break;
|
|
|
157
|
case 2:
|
|
|
158
|
cout << " CRITICAL - No response";
|
|
|
159
160
161
|
break;
}
|
|
|
162
|
|
|
|
163
164
|
return returnCode;
}
|
