1 files changed, 155 insertions, 0 deletions

diff --git a/lib/libtle_glue/init.c b/lib/libtle_glue/init.c
new file mode 100644
index 0000000..d845ef8
--- /dev/null
+++ b/lib/libtle_glue/init.c
@@ -0,0 +1,155 @@
+/*
+ * Copyright (c) 2018 Ant Financial Services Group.
+ * 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.
+ */
+
+#ifndef _GNU_SOURCE
+#define _GNU_SOURCE
+#endif
+#include <sched.h>
+#include <pthread.h>
+#include <stdlib.h>
+
+#include <rte_compat.h>
+#include <rte_common.h>
+#include <rte_debug.h>
+#include <rte_eal.h>
+
+#include "util.h"
+#include "fd.h"
+#include "ctx.h"
+#include "sym.h"
+#include "log.h"
+#include "internal.h"
+#include "tle_glue.h"
+
+void
+glue_init1(int argc, char **argv)
+{
+	GLUE_LOG(INFO, "init: DPDK and fd table...");
+
+	if (rte_eal_init(argc, argv) < 0)
+		rte_panic("Failed to init DPDK");
+
+	fd_init();
+}
+
+static void __attribute__((constructor(1000)))
+glue_init(void)
+{
+	char *p;
+	int i, err, argc = 0;
+	char **argv = NULL, **argv_to_release = NULL;
+	char *vnic, *params, *no_huge;
+	cpu_set_t cpuset;
+	pthread_t tid = pthread_self();
+
+	symbol_init();
+
+#define DPDK_PARAMS "DPDK_PARAMS"
+	params = getenv(DPDK_PARAMS);
+#define DPDK_NO_HUGE "DPDK_NO_HUGE"
+	no_huge = getenv(DPDK_NO_HUGE);
+#define DPDK_VNIC "DPDK_VNIC"
+	vnic = getenv(DPDK_VNIC);
+
+	if (params == NULL && no_huge == NULL && vnic == NULL)
+		return;
+
+	argv = grow_argv(argv, argc, 1);
+	argv[argc++] = xstrdup("userspace-stack");
+
+	/* Get the main thread affinity */
+	CPU_ZERO(&cpuset);
+	err = pthread_getaffinity_np(tid, sizeof(cpu_set_t), &cpuset);
+	if (!err) {
+		for (i = 0; i < CPU_SETSIZE; i++) {
+			if (CPU_ISSET(i, &cpuset)) {
+				argv = grow_argv(argv, argc, 2);
+				argv[argc++] = xstrdup("-l");
+				argv[argc++] = xasprintf("%d", i);
+				i = CPU_SETSIZE;
+			}
+		}
+	} else {
+		argv = grow_argv(argv, argc, 2);
+		argv[argc++] = xstrdup("-l");
+		argv[argc++] = xasprintf("0");
+	}
+
+	if (params)
+		p = strtok(params, " ");
+	else
+		p = NULL;
+	while (p != NULL) {
+		argv = grow_argv(argv, argc, 1);
+		argv[argc++] = xstrdup(p);
+		p = strtok(NULL, " ");
+	}
+
+	if (no_huge) {
+		argv = grow_argv(argv, argc, 3);
+		argv[argc++] = xstrdup("-m");
+		argv[argc++] = xstrdup("2048");
+		argv[argc++] = xstrdup("--no-huge");
+	}
+
+	if (vnic) {
+		argv = grow_argv(argv, argc, 2);
+		argv[argc++] = xstrdup(vnic);
+		argv[argc++] = xstrdup("--no-pci");
+	}
+
+	argv = grow_argv(argv, argc, 1);
+	argv[argc++] = xstrdup("--");
+
+	argv_to_release = grow_argv(argv_to_release, 0, argc);
+	for (i = 0; i < argc; ++i)
+		argv_to_release[i] = argv[i];
+
+	glue_init1(argc, argv);
+
+	/* Alloc and setup this default ctx for any sockets operations before
+	 * thread/ctx binding which happens when epoll_wait.
+	 */
+	glue_ctx_alloc();
+
+	release_argv(argc, argv_to_release, argv);
+
+	/* Set back the affinity */
+	err = pthread_setaffinity_np(tid, sizeof(cpu_set_t), &cpuset);
+	if (err)
+		GLUE_LOG(ERR, "Failed to set back affinity");
+}
+
+static void __attribute__((destructor))
+glue_uninit(void)
+{
+	struct sock *so;
+	struct glue_ctx *ctx;
+	int i, max = fd_table.fd_base + fd_table.fd_num;
+
+	/* TODO: lets optimize it */
+	for (i = fd_table.fd_base; i < max; i++) {
+		so = fd2sock(i);
+		if (!so || !so->valid)
+			continue;
+		if (IS_TCP(so))
+			tle_tcp_stream_kill(so->s);
+	}
+
+	for (i = 0; i < nb_ctx; ++i) {
+		ctx = glue_ctx_lookup(0, i);
+		while (be_process(ctx)) { /* empty */ };
+	}
+}