[HICN-379] Add face priority support in face manager

Change-Id: If4f75d44fc66414a4a70135de7827f5082b97112
Signed-off-by: Jordan Augé <jordan.auge+fdio@cisco.com>

1 files changed, 31 insertions, 0 deletions

diff --git a/hicn-light/src/hicn/io/streamConnection.c b/hicn-light/src/hicn/io/streamConnection.c
index 4e2f9c37e..27ec45d48 100644
--- a/hicn-light/src/hicn/io/streamConnection.c
+++ b/hicn-light/src/hicn/io/streamConnection.c
@@ -66,6 +66,9 @@ typedef struct stream_state {
    * but it is currently not reachable from within the implementation. */
   connection_state_t state;
   connection_state_t admin_state;
+#ifdef WITH_POLICY
+  uint32_t priority;
+#endif /* WITH_POLICY */
 } _StreamState;
 
 // Prototypes
@@ -91,6 +94,10 @@ static connection_state_t _streamConnection_getState(const IoOperations *ops);
 static void _streamConnection_setState(IoOperations *ops, connection_state_t state);
 static connection_state_t _streamConnection_getAdminState(const IoOperations *ops);
 static void _streamConnection_setAdminState(IoOperations *ops, connection_state_t admin_state);
+#ifdef WITH_POLICY
+static uint32_t _streamConnection_getPriority(const IoOperations *ops);
+static void _streamConnection_setPriority(IoOperations *ops, uint32_t priority);
+#endif /* WITH_POLICY */
 static const char * _streamConnection_getInterfaceName(const IoOperations *ops);
 
 /*
@@ -123,6 +130,10 @@ static IoOperations _template = {
     .setState = &_streamConnection_setState,
     .getAdminState = &_streamConnection_getAdminState,
     .setAdminState = &_streamConnection_setAdminState,
+#ifdef WITH_POLICY
+    .getPriority = &_streamConnection_getPriority,
+    .setPriority = &_streamConnection_setPriority,
+#endif /* WITH_POLICY */
     .getInterfaceName = &_streamConnection_getInterfaceName,
 };
 
@@ -147,6 +158,10 @@ IoOperations *streamConnection_AcceptConnection(Forwarder *forwarder, int fd,
   stream->addressPair = pair;
   stream->isClosed = false;
 
+#ifdef WITH_POLICY
+  stream->priority = 0;
+#endif /* WITH_POLICY */
+
   // allocate a connection
   IoOperations *io_ops = parcMemory_AllocateAndClear(sizeof(IoOperations));
   parcAssertNotNull(io_ops, "parcMemory_AllocateAndClear(%zu) returned NULL",
@@ -736,6 +751,22 @@ static void _streamConnection_setAdminState(IoOperations *ops, connection_state_
   stream->admin_state = admin_state;
 }
 
+#ifdef WITH_POLICY
+static uint32_t _streamConnection_getPriority(const IoOperations *ops) {
+  parcAssertNotNull(ops, "Parameter must be non-null");
+  const _StreamState *stream =
+      (const _StreamState *)ioOperations_GetClosure(ops);
+  return stream->priority;
+}
+
+static void _streamConnection_setPriority(IoOperations *ops, uint32_t priority) {
+  parcAssertNotNull(ops, "Parameter must be non-null");
+  _StreamState *stream =
+      (_StreamState *)ioOperations_GetClosure(ops);
+  stream->priority = priority;
+}
+#endif /* WITH_POLICY */
+
 static const char * _streamConnection_getInterfaceName(const IoOperations *ops)
 {
   parcAssertNotNull(ops, "Parameter must be non-null");