]> git.stg.codes - stg.git/blob - projects/stargazer/plugins/other/smux/utils.cpp
328cea88e6cef016b94a6dd161cbd8d79e73c5c7
[stg.git] / projects / stargazer / plugins / other / smux / utils.cpp
1 #include <sys/select.h>
2 #include <unistd.h> // write
3
4 #include <cstring> // memset
5 #include <cerrno>
6
7 #include "stg/common.h"
8
9 #include "asn1/OpenPDU.h"
10 #include "asn1/ClosePDU.h"
11 #include "asn1/RReqPDU.h"
12 #include "asn1/ber_decoder.h"
13 #include "asn1/der_encoder.h"
14
15 #include "pen.h"
16 #include "utils.h"
17
18 bool WaitPackets(int sd)
19 {
20 fd_set rfds;
21 FD_ZERO(&rfds);
22 FD_SET(sd, &rfds);
23
24 struct timeval tv;
25 tv.tv_sec = 0;
26 tv.tv_usec = 500000;
27
28 int res = select(sd + 1, &rfds, NULL, NULL, &tv);
29 if (res == -1) // Error
30     {
31     if (errno != EINTR)
32         printfd(__FILE__, "Error on select: '%s'\n", strerror(errno));
33     return false;
34     }
35
36 if (res == 0) // Timeout
37     return false;
38
39 return true;
40 }
41
42 bool String2OI(const std::string & str, OBJECT_IDENTIFIER_t * oi)
43 {
44 size_t left = 0, pos = 0, arcPos = 0;
45 int arcs[1024];
46 pos = str.find_first_of('.', left);
47 if (pos == 0)
48     {
49     left = 1;
50     pos = str.find_first_of('.', left);
51     }
52 while (pos != std::string::npos)
53     {
54     int arc = 0;
55     if (str2x(str.substr(left, left - pos), arc))
56         {
57         return false;
58         }
59     arcs[arcPos++] = arc;
60     left = pos + 1;
61     pos = str.find_first_of('.', left);
62     }
63 if (left < str.length())
64     {
65     int arc = 0;
66     if (str2x(str.substr(left, left - pos), arc))
67         {
68         return false;
69         }
70     arcs[arcPos++] = arc;
71     }
72 printfd(__FILE__, "String2OI() - arcPos: %d\n", arcPos);
73 OBJECT_IDENTIFIER_set_arcs(oi, arcs, sizeof(arcs[0]), arcPos);
74 return true;
75 }
76
77 std::string OI2String(OBJECT_IDENTIFIER_t * oi)
78 {
79 std::string res;
80
81 int arcs[1024];
82 int count = OBJECT_IDENTIFIER_get_arcs(oi, arcs, sizeof(arcs[0]), 1024);
83
84 if (count > 1024)
85     return "";
86
87 for (int i = 0; i < count; ++i)
88     {
89     res += ".";
90     std::string arc;
91     strprintf(&arc, "%d", arcs[i]);
92     res += arc;
93     }
94
95 return res;
96 }
97
98 bool SendOpenPDU(int fd)
99 {
100 const char * description = "Stg SMUX Plugin";
101 asn_enc_rval_t error;
102 OpenPDU_t msg;
103
104 memset(&msg, 0, sizeof(msg));
105
106 msg.present = OpenPDU_PR_simple;
107 asn_long2INTEGER(&msg.choice.simple.version, SimpleOpen__version_version_1);
108 if (!String2OI(PEN_PREFIX, &msg.choice.simple.identity))
109     {
110     printfd(__FILE__,
111             "SendOpenPDU() - failed to convert string to OBJECT_IDENTIFIER\n");
112     return false;
113     }
114 OCTET_STRING_fromString(&msg.choice.simple.description, description);
115 OCTET_STRING_fromString(&msg.choice.simple.password, "");
116
117 char buffer[1024];
118 error = der_encode_to_buffer(&asn_DEF_OpenPDU, &msg, buffer, sizeof(buffer));
119
120 if (error.encoded == -1)
121     {
122     printfd(__FILE__, "Could not encode OpenPDU (at %s)\n",
123             error.failed_type ? error.failed_type->name : "unknown");
124     return false;
125     }
126 else
127     {
128     write(fd, buffer, error.encoded);
129     printfd(__FILE__, "OpenPDU encoded successfully to %d bytes\n",
130             error.encoded);
131     }
132 return true;
133 }
134
135 int SendClosePDU(int fd)
136 {
137 ClosePDU_t msg;
138
139 memset(&msg, 0, sizeof(msg));
140
141 asn_long2INTEGER(&msg, ClosePDU_goingDown);
142
143 char buffer[1024];
144 asn_enc_rval_t error;
145 error = der_encode_to_buffer(&asn_DEF_ClosePDU, &msg, buffer, sizeof(buffer));
146
147 if (error.encoded == -1)
148     {
149     printfd(__FILE__, "Could not encode ClosePDU (at %s)\n",
150             error.failed_type ? error.failed_type->name : "unknown");
151     return -1;
152     }
153 else
154     {
155     write(fd, buffer, error.encoded);
156     printfd(__FILE__, "ClosePDU encoded successfully\n");
157     }
158 return 0;
159 }
160
161 int SendRReqPDU(int fd)
162 {
163 int oid[] = {1, 3, 6, 1, 4, 1, 38313, 1};
164 asn_enc_rval_t error;
165 RReqPDU_t msg;
166
167 memset(&msg, 0, sizeof(msg));
168
169 msg.priority = 0;
170 asn_long2INTEGER(&msg.operation, RReqPDU__operation_readOnly);
171 OBJECT_IDENTIFIER_set_arcs(&msg.subtree,
172                            oid,
173                            sizeof(oid[0]),
174                            8);
175
176 char buffer[1024];
177 error = der_encode_to_buffer(&asn_DEF_RReqPDU, &msg, buffer, sizeof(buffer));
178
179 if (error.encoded == -1)
180     {
181     printfd(__FILE__, "Could not encode RReqPDU (at %s)\n",
182             error.failed_type ? error.failed_type->name : "unknown");
183     return -1;
184     }
185 else
186     {
187     write(fd, buffer, error.encoded);
188     printfd(__FILE__, "RReqPDU encoded successfully to %d bytes\n",
189             error.encoded);
190     }
191 return 0;
192 }
193
194 SMUX_PDUs_t * RecvSMUXPDUs(int fd)
195 {
196 char buffer[1024];
197 SMUX_PDUs_t * pdus = NULL;
198
199 memset(buffer, 0, sizeof(buffer));
200
201 size_t length = read(fd, buffer, sizeof(buffer));
202 if (length < 1)
203     return NULL;
204 asn_dec_rval_t error;
205 error = ber_decode(0, &asn_DEF_SMUX_PDUs, (void **)&pdus, buffer, length);
206 if(error.code != RC_OK)
207     {
208     printfd(__FILE__, "Failed to decode PDUs at byte %ld\n",
209             (long)error.consumed);
210     return NULL;
211     }
212 return pdus;
213 }
214
215 int SendGetResponsePDU(int fd, GetResponse_PDU_t * getResponse)
216 {
217 asn_enc_rval_t error;
218
219 char buffer[1024];
220 error = der_encode_to_buffer(&asn_DEF_GetResponse_PDU, getResponse, buffer,
221                              sizeof(buffer));
222
223 if (error.encoded == -1)
224     {
225     printfd(__FILE__, "Could not encode GetResponsePDU (at %s)\n",
226             error.failed_type ? error.failed_type->name : "unknown");
227     return -1;
228     }
229 else
230     {
231     write(fd, buffer, error.encoded);
232     printfd(__FILE__, "GetResponsePDU encoded successfully to %d bytes\n",
233             error.encoded);
234     }
235 return 0;
236 }
237
238 int SendGetResponseErrorPDU(int fd,
239                             const PDU_t * getRequest,
240                             int errorStatus,
241                             int errorIndex)
242 {
243 asn_enc_rval_t error;
244 GetResponse_PDU_t msg;
245
246 memset(&msg, 0, sizeof(msg));
247
248 msg.request_id = getRequest->request_id;
249 asn_long2INTEGER(&msg.error_status, errorStatus);
250 asn_long2INTEGER(&msg.error_index, errorIndex);
251
252 char buffer[1024];
253 error = der_encode_to_buffer(&asn_DEF_GetResponse_PDU, &msg, buffer,
254                              sizeof(buffer));
255
256 if (error.encoded == -1)
257     {
258     printfd(__FILE__, "Could not encode GetResponsePDU for error (at %s)\n",
259             error.failed_type ? error.failed_type->name : "unknown");
260     return -1;
261     }
262 else
263     {
264     write(fd, buffer, error.encoded);
265     printfd(__FILE__,
266             "GetResponsePDU for error encoded successfully to %d bytes\n",
267             error.encoded);
268     }
269 return 0;
270 }