1 files changed, 123 insertions, 0 deletions
diff --git a/metis/ccnx/forwarder/metis/config/metisControl_SetStrategy.c b/metis/ccnx/forwarder/metis/config/metisControl_SetStrategy.c
new file mode 100644
index 00000000..f7b805ca
--- /dev/null
+++ b/metis/ccnx/forwarder/metis/config/metisControl_SetStrategy.c
@@ -0,0 +1,123 @@
+/*
+ * Copyright (c) 2017 Cisco and/or its affiliates.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+
+#include <config.h>
+
+#include <stdbool.h>
+#include <stdint.h>
+#include <strings.h>
+#include <stdlib.h>
+#include <stdio.h>
+
+#include <LongBow/runtime.h>
+
+#include <parc/algol/parc_Memory.h>
+
+#include <ccnx/api/control/cpi_ManageLinks.h>
+#include <ccnx/api/control/cpi_Forwarding.h>
+
+#include <ccnx/forwarder/metis/core/metis_Forwarder.h>
+#include <ccnx/forwarder/metis/core/metis_Dispatcher.h>
+#include <ccnx/forwarder/metis/config/metisControl_SetDebug.h>
+
+static MetisCommandReturn _metisControlSetStrategy_Execute(MetisCommandParser *parser, MetisCommandOps *ops, PARCList *args);
+static MetisCommandReturn _metisControlSetStrategy_HelpExecute(MetisCommandParser *parser, MetisCommandOps *ops, PARCList *args);
+
+static const char *_commandSetStrategy = "set strategy";
+static const char *_commandSetStrategyHelp = "help set strategy";
+
+// ====================================================
+
+MetisCommandOps *
+metisControlSetStrategy_Create(MetisControlState *state)
+{
+    return metisCommandOps_Create(state, _commandSetStrategy, NULL, _metisControlSetStrategy_Execute, metisCommandOps_Destroy);
+}
+
+MetisCommandOps *
+metisControlSetStrategy_HelpCreate(MetisControlState *state)
+{
+    return metisCommandOps_Create(state, _commandSetStrategyHelp, NULL, _metisControlSetStrategy_HelpExecute, metisCommandOps_Destroy);
+}
+
+// ====================================================
+
+static MetisCommandReturn
+_metisControlSetStrategy_HelpExecute(MetisCommandParser *parser, MetisCommandOps *ops, PARCList *args)
+{
+    printf("set strategy <prefix> <strategy>\n");
+    printf("available strateies:\n");
+    printf("    random\n");
+    printf("    loadbalancer\n");
+    printf("    random_per_dash_segment\n");
+    printf("    loadbalancer_with_delay\n");
+    printf("\n");
+    return MetisCommandReturn_Success;
+}
+
+static MetisCommandReturn
+_metisControlSetStrategy_Execute(MetisCommandParser *parser, MetisCommandOps *ops, PARCList *args)
+{
+    MetisControlState *state = ops->closure;
+
+    if (parcList_Size(args) != 4) {
+        _metisControlSetStrategy_HelpExecute(parser, ops, args);
+        return MetisCommandReturn_Failure;
+    }
+
+    if (((strcmp(parcList_GetAtIndex(args, 0), "set") != 0) || (strcmp(parcList_GetAtIndex(args, 1), "strategy") != 0))) {
+        _metisControlSetStrategy_HelpExecute(parser, ops, args);
+        return MetisCommandReturn_Failure;
+    }
+
+    const char *prefixString = parcList_GetAtIndex(args, 2);
+    const char *strategy = parcList_GetAtIndex(args, 3);
+    CCNxName *prefix = ccnxName_CreateFromCString(prefixString);
+    if (prefix == NULL) {
+        printf("ERROR: could not parse prefix '%s'\n", prefixString);
+        return MetisCommandReturn_Failure;
+    }
+
+    CPIForwardingStrategy *fwdStrategy = cpiForwardingStrategy_Create(prefix, (char *) strategy);
+
+    CCNxControl *setFwdStrategyRequest = ccnxControl_CreateSetStrategyRequest(fwdStrategy);
+
+    cpiForwardingStrategy_Destroy(&fwdStrategy);
+
+    if (metisControlState_GetDebug(state)) {
+        char *str = parcJSON_ToString(ccnxControl_GetJson(setFwdStrategyRequest));
+        printf("request: %s\n", str);
+        parcMemory_Deallocate((void **) &str);
+    }
+
+    CCNxMetaMessage *message = ccnxMetaMessage_CreateFromControl(setFwdStrategyRequest);
+    CCNxMetaMessage *rawResponse = metisControlState_WriteRead(state, message);
+    ccnxMetaMessage_Release(&message);
+
+    ccnxControl_Release(&setFwdStrategyRequest);
+
+    CCNxControl *response = ccnxMetaMessage_GetControl(rawResponse);
+
+    if (metisControlState_GetDebug(state)) {
+        char *str = parcJSON_ToString(ccnxControl_GetJson(response));
+        printf("response: %s\n", str);
+        parcMemory_Deallocate((void **) &str);
+    }
+
+    ccnxMetaMessage_Release(&rawResponse);
+
+    return MetisCommandReturn_Success;
+}