aboutsummaryrefslogtreecommitdiff
path: root/examples
diff options
context:
space:
mode:
Diffstat (limited to 'examples')
-rw-r--r--examples/common.hpp165
-rw-r--r--examples/driver-protobuf-c-tcp.cpp11
-rw-r--r--examples/driver-protobuf-c.cpp11
-rw-r--r--examples/driver.cpp6
-rw-r--r--examples/userspace_client_protobuf.cpp4
5 files changed, 183 insertions, 14 deletions
diff --git a/examples/common.hpp b/examples/common.hpp
new file mode 100644
index 0000000..5e2b637
--- /dev/null
+++ b/examples/common.hpp
@@ -0,0 +1,165 @@
+#ifndef COMMON_H
+#define COMMON_H 1
+
+#include <ksocket/helper.hpp>
+
+#include "examples/example.pb-c.h"
+
+/* Could be also be done via protobuf */
+enum PduTypes {
+ PDU_SOMETHING_WITH_UINTS = 1,
+ PDU_SOMETHING_MORE = 2,
+ PDU_EVEN_MORE = 3
+};
+
+class SomethingWithUINTsSerializer : virtual public BaseSerializer {
+public:
+ SomethingWithUINTsSerializer(void) {}
+ ~SomethingWithUINTsSerializer(void) {}
+ void SetId(uint32_t id) {
+ swu.has_id = TRUE;
+ swu.id = id;
+ }
+ void SetIpAddress(uint32_t ip_address) {
+ swu.has_ip_address = TRUE;
+ swu.ip_address = ip_address;
+ }
+ void SetPortNum(uint32_t port_num) {
+ swu.has_port_num = TRUE;
+ swu.port_num = port_num;
+ }
+ uint16_t GetPduType(void) override { return PDU_SOMETHING_WITH_UINTS; }
+ size_t GetSerializedSize(void) override {
+ return something_with_uints__get_packed_size(&swu);
+ }
+ size_t Serialize(uint8_t *buf) override {
+ return something_with_uints__pack(&swu, buf);
+ }
+
+ SomethingWithUINTs swu = SOMETHING_WITH_UINTS__INIT;
+};
+
+class SomethingWithUINTsDeserializer : virtual public BaseDeserializer {
+public:
+ SomethingWithUINTsDeserializer(void) {}
+ ~SomethingWithUINTsDeserializer(void) { DeserializeFree(); }
+ bool Deserialize(size_t pdu_len, uint8_t *buf) override {
+ swu = something_with_uints__unpack(NULL, pdu_len, buf);
+ return swu != NULL;
+ }
+ void DeserializeFree(void) override {
+ if (swu != NULL)
+ something_with_uints__free_unpacked(swu, NULL);
+ }
+
+ SomethingWithUINTs *swu = NULL;
+};
+
+class SomethingMoreSerializer : virtual public BaseSerializer,
+ public SomethingWithUINTsSerializer {
+public:
+ SomethingMoreSerializer(void) { sm.uints = &swu; }
+ ~SomethingMoreSerializer(void) {}
+ void SetErrorCode(SomethingMore__Errors error_code) {
+ sm.has_error_code = TRUE;
+ sm.error_code = error_code;
+ }
+ uint16_t GetPduType(void) override { return PDU_SOMETHING_MORE; }
+ size_t GetSerializedSize(void) override {
+ return something_more__get_packed_size(&sm);
+ }
+ size_t Serialize(uint8_t *buf) override {
+ return something_more__pack(&sm, buf);
+ }
+
+ SomethingMore sm = SOMETHING_MORE__INIT;
+};
+
+class SomethingMoreDeserializer : virtual public BaseDeserializer {
+public:
+ SomethingMoreDeserializer(void) {}
+ ~SomethingMoreDeserializer(void) { DeserializeFree(); }
+ bool Deserialize(size_t pdu_len, uint8_t *buf) override {
+ sm = something_more__unpack(NULL, pdu_len, buf);
+ return sm != NULL;
+ }
+ void DeserializeFree(void) override {
+ if (sm != NULL)
+ something_more__free_unpacked(sm, NULL);
+ }
+
+ SomethingMore *sm = NULL;
+};
+
+class EvenMoreSerializer : virtual public BaseSerializer {
+public:
+ explicit EvenMoreSerializer(void) = default;
+ explicit EvenMoreSerializer(EvenMore__SomeEnum enum_value,
+ ProtobufCBinaryDataClass name,
+ ProtobufCBinaryDataClass value)
+ : uints() {
+ em.enum_value = enum_value;
+ em.name = name;
+ em.value = value;
+ }
+ ~EvenMoreSerializer(void) { free(s); }
+ void AddUints(SomethingWithUINTsSerializer *uints) {
+ this->uints.push_back(uints);
+ }
+ void SetS(eastl::string s) {
+ size_t l = s.size();
+ this->s = (char *)malloc(l + 1);
+ memcpy(this->s, s.c_str(), l);
+ this->s[l] = '\0';
+ }
+ uint16_t GetPduType(void) override { return PDU_EVEN_MORE; }
+ size_t GetSerializedSize(void) override {
+ em.s = s;
+ em.n_uints = this->uints.size();
+ if (em.n_uints > 0) {
+ SomethingWithUINTs *out[em.n_uints];
+ ConvertUintsVectorToCArray(out);
+ return even_more__get_packed_size(&em);
+ }
+ return even_more__get_packed_size(&em);
+ }
+ size_t Serialize(uint8_t *buf) override {
+ em.s = s;
+ em.n_uints = this->uints.size();
+ if (em.n_uints > 0) {
+ SomethingWithUINTs *out[em.n_uints];
+ ConvertUintsVectorToCArray(out);
+ return even_more__pack(&em, buf);
+ }
+ return even_more__pack(&em, buf);
+ }
+
+ EvenMore em = EVEN_MORE__INIT;
+ eastl::vector<SomethingWithUINTsSerializer *> uints;
+ char *s = NULL;
+
+ void ConvertUintsVectorToCArray(SomethingWithUINTs **out) {
+ for (size_t i = 0; i < em.n_uints; ++i) {
+ out[i] = &uints[i]->swu;
+ }
+ em.uints = out;
+ }
+};
+
+class EvenMoreDeserializer : virtual public BaseDeserializer {
+public:
+ EvenMoreDeserializer(void) {}
+ ~EvenMoreDeserializer(void) { DeserializeFree(); }
+ bool Deserialize(size_t pdu_len, uint8_t *buf) override {
+ em = even_more__unpack(NULL, pdu_len, buf);
+ return em != NULL;
+ }
+ void DeserializeFree(void) override {
+ if (em != NULL)
+ even_more__free_unpacked(em, NULL);
+ }
+
+ EvenMore *em = NULL;
+};
+
+#endif
diff --git a/examples/driver-protobuf-c-tcp.cpp b/examples/driver-protobuf-c-tcp.cpp
index da2142d..fcf4465 100644
--- a/examples/driver-protobuf-c-tcp.cpp
+++ b/examples/driver-protobuf-c-tcp.cpp
@@ -1,9 +1,10 @@
-#include "berkeley.h"
-#include "ksocket.h"
-#include "examples/example.pb-c.h"
-#include "wsk.h"
+#include <ksocket/berkeley.h>
+#include <ksocket/helper.hpp>
+#include <ksocket/ksocket.h>
+#include <ksocket/wsk.h>
-#include "common.hpp"
+#include "examples/common.hpp"
+#include "examples/example.pb-c.h"
extern "C" {
DRIVER_INITIALIZE DriverEntry;
diff --git a/examples/driver-protobuf-c.cpp b/examples/driver-protobuf-c.cpp
index 942782d..c3c8445 100644
--- a/examples/driver-protobuf-c.cpp
+++ b/examples/driver-protobuf-c.cpp
@@ -1,9 +1,10 @@
-#include "berkeley.h"
-#include "ksocket.h"
-#include "examples/example.pb-c.h"
-#include "wsk.h"
+#include <ksocket/berkeley.h>
+#include <ksocket/helper.hpp>
+#include <ksocket/ksocket.h>
+#include <ksocket/wsk.h>
-#include "common.hpp"
+#include "examples/common.hpp"
+#include "examples/example.pb-c.h"
extern "C" {
DRIVER_INITIALIZE DriverEntry;
diff --git a/examples/driver.cpp b/examples/driver.cpp
index ceae8ff..86e42e9 100644
--- a/examples/driver.cpp
+++ b/examples/driver.cpp
@@ -1,8 +1,8 @@
extern "C" {
-#include "berkeley.h"
-#include "ksocket.h"
-#include "wsk.h"
+#include <ksocket/berkeley.h>
+#include <ksocket/ksocket.h>
+#include <ksocket/wsk.h>
DRIVER_INITIALIZE DriverEntry;
DRIVER_UNLOAD DriverUnload;
diff --git a/examples/userspace_client_protobuf.cpp b/examples/userspace_client_protobuf.cpp
index b8ad711..a387579 100644
--- a/examples/userspace_client_protobuf.cpp
+++ b/examples/userspace_client_protobuf.cpp
@@ -3,7 +3,9 @@
#include <winsock2.h>
#include <ws2tcpip.h>
-#include "common.hpp"
+#include <ksocket/helper.hpp>
+
+#include "examples/common.hpp"
#include "examples/example.pb-c.h"
int main(int argc, char **argv) {
Powered by cgit, Themed by Ted Yin.