-#include <sys/select.h>
#include <unistd.h> // write
#include <cstring> // memset
#include "stg/common.h"
-#include "asn1/OpenPDU.h"
-#include "asn1/ClosePDU.h"
-#include "asn1/RReqPDU.h"
-#include "asn1/ber_decoder.h"
-#include "asn1/der_encoder.h"
+#include "stg/OpenPDU.h"
+#include "stg/ClosePDU.h"
+#include "stg/RReqPDU.h"
+#include "stg/ber_decoder.h"
+#include "stg/der_encoder.h"
#include "pen.h"
#include "utils.h"
-bool WaitPackets(int sd)
-{
-fd_set rfds;
-FD_ZERO(&rfds);
-FD_SET(sd, &rfds);
-
-struct timeval tv;
-tv.tv_sec = 0;
-tv.tv_usec = 500000;
-
-int res = select(sd + 1, &rfds, NULL, NULL, &tv);
-if (res == -1) // Error
- {
- if (errno != EINTR)
- printfd(__FILE__, "Error on select: '%s'\n", strerror(errno));
- return false;
- }
-
-if (res == 0) // Timeout
- return false;
-
-return true;
-}
-
bool String2OI(const std::string & str, OBJECT_IDENTIFIER_t * oi)
{
size_t left = 0, pos = 0, arcPos = 0;
}
arcs[arcPos++] = arc;
}
-printfd(__FILE__, "String2OI() - arcPos: %d\n", arcPos);
-OBJECT_IDENTIFIER_set_arcs(oi, arcs, sizeof(arcs[0]), arcPos);
+OBJECT_IDENTIFIER_set_arcs(oi, arcs, sizeof(arcs[0]), static_cast<unsigned int>(arcPos));
return true;
}
char buffer[1024];
error = der_encode_to_buffer(&asn_DEF_OpenPDU, &msg, buffer, sizeof(buffer));
+ASN_STRUCT_FREE_CONTENTS_ONLY(asn_DEF_OpenPDU, &msg);
+
if (error.encoded == -1)
{
printfd(__FILE__, "Could not encode OpenPDU (at %s)\n",
}
else
{
- write(fd, buffer, error.encoded);
- printfd(__FILE__, "OpenPDU encoded successfully to %d bytes\n",
- error.encoded);
+ if (write(fd, buffer, error.encoded) < 0)
+ {
+ printfd(__FILE__, "Failed to send OpenPDU: %s\n", strerror(errno));
+ return false;
+ }
}
return true;
}
-int SendClosePDU(int fd)
+bool SendClosePDU(int fd)
{
ClosePDU_t msg;
asn_enc_rval_t error;
error = der_encode_to_buffer(&asn_DEF_ClosePDU, &msg, buffer, sizeof(buffer));
+ASN_STRUCT_FREE_CONTENTS_ONLY(asn_DEF_ClosePDU, &msg);
+
if (error.encoded == -1)
{
printfd(__FILE__, "Could not encode ClosePDU (at %s)\n",
error.failed_type ? error.failed_type->name : "unknown");
- return -1;
+ return false;
}
else
{
- write(fd, buffer, error.encoded);
- printfd(__FILE__, "ClosePDU encoded successfully\n");
+ if (write(fd, buffer, error.encoded) < 0)
+ {
+ printfd(__FILE__, "Failed to send ClosePDU: %s\n", strerror(errno));
+ return false;
+ }
}
-return 0;
+return true;
}
-int SendRReqPDU(int fd)
+bool SendRReqPDU(int fd)
{
int oid[] = {1, 3, 6, 1, 4, 1, 38313, 1};
asn_enc_rval_t error;
char buffer[1024];
error = der_encode_to_buffer(&asn_DEF_RReqPDU, &msg, buffer, sizeof(buffer));
+ASN_STRUCT_FREE_CONTENTS_ONLY(asn_DEF_RReqPDU, &msg);
+
if (error.encoded == -1)
{
printfd(__FILE__, "Could not encode RReqPDU (at %s)\n",
error.failed_type ? error.failed_type->name : "unknown");
- return -1;
+ return false;
}
else
{
- write(fd, buffer, error.encoded);
- printfd(__FILE__, "RReqPDU encoded successfully to %d bytes\n",
- error.encoded);
+ if (write(fd, buffer, error.encoded) < 0)
+ {
+ printfd(__FILE__, "Failed to send RReqPDU: %s\n", strerror(errno));
+ return false;
+ }
}
-return 0;
+return true;
}
SMUX_PDUs_t * RecvSMUXPDUs(int fd)
return NULL;
asn_dec_rval_t error;
error = ber_decode(0, &asn_DEF_SMUX_PDUs, (void **)&pdus, buffer, length);
+
if(error.code != RC_OK)
{
printfd(__FILE__, "Failed to decode PDUs at byte %ld\n",
return pdus;
}
-std::string OI2String(OBJECT_IDENTIFIER_t * oi)
-{
-std::string res;
-
-int arcs[1024];
-int count = OBJECT_IDENTIFIER_get_arcs(oi, arcs, sizeof(arcs[0]), 1024);
-
-if (count > 1024)
- return "";
-
-for (int i = 0; i < count; ++i)
- {
- res += ".";
- std::string arc;
- strprintf(&arc, "%d", arcs[i]);
- res += arc;
- }
-
-return res;
-}
-
-int SendGetResponsePDU(int fd, GetResponse_PDU_t * getResponse)
+bool SendGetResponsePDU(int fd, GetResponse_PDU_t * getResponse)
{
asn_enc_rval_t error;
{
printfd(__FILE__, "Could not encode GetResponsePDU (at %s)\n",
error.failed_type ? error.failed_type->name : "unknown");
- return -1;
+ return false;
}
else
{
- write(fd, buffer, error.encoded);
- printfd(__FILE__, "GetResponsePDU encoded successfully to %d bytes\n",
- error.encoded);
+ if (write(fd, buffer, error.encoded) < 0)
+ {
+ printfd(__FILE__, "Failed to send GetResponsePDU: %s\n", strerror(errno));
+ return false;
+ }
}
-return 0;
+return true;
}
-int SendGetResponseErrorPDU(int fd,
- const PDU_t * getRequest,
- int errorStatus,
- int errorIndex)
+bool SendGetResponseErrorPDU(int fd,
+ const PDU_t * getRequest,
+ int errorStatus,
+ int errorIndex)
{
asn_enc_rval_t error;
GetResponse_PDU_t msg;
memset(&msg, 0, sizeof(msg));
-msg.request_id = getRequest->request_id;
+long id = 0;
+asn_INTEGER2long(&getRequest->request_id, &id);
+asn_long2INTEGER(&msg.request_id, id);
asn_long2INTEGER(&msg.error_status, errorStatus);
asn_long2INTEGER(&msg.error_index, errorIndex);
error = der_encode_to_buffer(&asn_DEF_GetResponse_PDU, &msg, buffer,
sizeof(buffer));
+ASN_STRUCT_FREE_CONTENTS_ONLY(asn_DEF_GetResponse_PDU, &msg);
+
if (error.encoded == -1)
{
printfd(__FILE__, "Could not encode GetResponsePDU for error (at %s)\n",
error.failed_type ? error.failed_type->name : "unknown");
- return -1;
+ return false;
}
else
{
- write(fd, buffer, error.encoded);
- printfd(__FILE__,
- "GetResponsePDU for error encoded successfully to %d bytes\n",
- error.encoded);
+ if (write(fd, buffer, error.encoded) < 0)
+ {
+ printfd(__FILE__, "Failed to send GetResponseErrorPDU: %s\n", strerror(errno));
+ return false;
+ }
}
-return 0;
+return true;
}