summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/client/main.cpp139
-rw-r--r--src/version.h2
-rw-r--r--src/worker/main.cpp166
3 files changed, 307 insertions, 0 deletions
diff --git a/src/client/main.cpp b/src/client/main.cpp
new file mode 100644
index 0000000..dfec059
--- /dev/null
+++ b/src/client/main.cpp
@@ -0,0 +1,139 @@
+/*
+ # Copyright (c) 2011 - OpenSLX Project, Computer Center University of Freiburg
+ #
+ # This program is free software distributed under the GPL version 2.
+ # See http://gpl.openslx.org/
+ #
+ # If you have any feedback please consult http://feedback.openslx.org/ and
+ # send your suggestions, praise, or complaints to feedback@openslx.org
+ #
+ # General information about OpenSLX can be found at http://openslx.org/
+ */
+
+#include <sstream>
+#include <iostream>
+#include <stdio.h>
+#include <libgearman/gearman.h>
+
+#include "../version.h"
+
+using namespace std;
+
+static void usage(char *name)
+{
+ printf("usage: %s -h <host> [-p <port>] -i <image> -d <device>\n", name);
+ printf("\t-h <host>\t gearman job server host\n");
+ printf("\t-p <port>\t gearman job server port\n");
+ printf("\t-i <image>\t remote image file to load\n");
+ printf("\t-d <device>\t local nbd-device to use\n");
+ printf("\t-v\t\t print version and exit\n");
+}
+
+int main(int argc, char* argv[])
+{
+ string host("");
+ in_port_t port = 0;
+ string image("");
+ string device("");
+
+ gearman_return_t ret;
+ gearman_client_st client;
+
+ // parse command line arguments
+ int c;
+ while ((c = getopt(argc, argv, "h:p:i:d:v")) != -1)
+ {
+ switch (c)
+ {
+ case 'h':
+ host = optarg;
+ break;
+ case 'p':
+ port = (in_port_t) atoi(optarg);
+ break;
+ case 'i':
+ image = optarg;
+ break;
+ case 'd':
+ device = optarg;
+ break;
+ case 'v':
+ printf("Version: %s\n", VERSION_STRING);
+ exit(0);
+ default:
+ usage(argv[0]);
+ exit(1);
+ }
+ }
+
+ if (host == "" || image == "" || device == "")
+ {
+ usage(argv[0]);
+ exit(1);
+ }
+
+ // initialize gearman client
+ if (gearman_client_create(&client) == NULL)
+ {
+ cerr << "[ERROR] Memory allocation failure on client creation" << endl;
+ exit(1);
+ }
+ ret = gearman_client_add_server(&client, host.c_str(), port);
+ if (ret != GEARMAN_SUCCESS)
+ {
+ cerr << "[ERROR] " << gearman_client_error(&client) << endl;
+ exit(1);
+ }
+
+ // main loop
+ char *result;
+ size_t result_size;
+ string msg;
+ while (1)
+ {
+ // send request to gearman job server
+ result = (char *) gearman_client_do(&client, "start_nbd_server", NULL,
+ (void*) image.c_str(), (size_t) image.length(), &result_size,
+ &ret);
+
+ if (ret == GEARMAN_WORK_DATA)
+ {
+ msg = result;
+ free(result);
+ continue;
+ }
+ else if (ret == GEARMAN_WORK_STATUS)
+ {
+ uint32_t m, n;
+ gearman_client_do_status(&client, &n, &m);
+ printf("[INFO] Status: %u/%u\n", n, m);
+ continue;
+ }
+ else if (ret == GEARMAN_SUCCESS)
+ {
+ // start nbd-client
+ stringstream command;
+ command << "nbd-client " << host << " " << msg << " " << device;
+ cout << "[DEBUG] " << "Running command: " << command.str() << endl;
+ system(command.str().c_str());
+ free(result);
+ }
+ else if (ret == GEARMAN_WORK_FAIL)
+ {
+ cerr << "[ERROR] " << msg << endl;
+ cerr << "[ERROR] Work failed" << endl;
+ free(result);
+ }
+ else
+ {
+ cerr << "[ERROR] " << gearman_client_error(&client) << endl;
+ free(result);
+ }
+
+ break;
+ }
+
+ // cleanup
+ gearman_client_free(&client);
+ return 0;
+}
diff --git a/src/version.h b/src/version.h
new file mode 100644
index 0000000..43331a7
--- /dev/null
+++ b/src/version.h
@@ -0,0 +1,2 @@
+#define VERSION_STRING "0.1.0"
+#define VERSION_NUMBER 010
diff --git a/src/worker/main.cpp b/src/worker/main.cpp
new file mode 100644
index 0000000..ff97041
--- /dev/null
+++ b/src/worker/main.cpp
@@ -0,0 +1,166 @@
+/*
+ # Copyright (c) 2011 - OpenSLX Project, Computer Center University of Freiburg
+ #
+ # This program is free software distributed under the GPL version 2.
+ # See http://gpl.openslx.org/
+ #
+ # If you have any feedback please consult http://feedback.openslx.org/ and
+ # send your suggestions, praise, or complaints to feedback@openslx.org
+ #
+ # General information about OpenSLX can be found at http://openslx.org/
+ */
+
+#include <map>
+#include <unistd.h>
+#include <iostream>
+#include <sstream>
+#include <stdio.h>
+#include <string.h>
+#include <libgearman/gearman.h>
+
+#include "../version.h"
+
+using namespace std;
+
+int _nbd_server_port;
+map<string, int> _map; // map running nbd-servers: <exported image, port number>
+
+static void *start_nbd_server(gearman_job_st *job, void *context,
+ size_t *result_size, gearman_return_t *ret_ptr)
+{
+ cout << "[DEBUG] Received request from " << gearman_job_handle(job) << endl;
+
+ if (gearman_job_workload_size(job) <= 0)
+ {
+ cerr << "[ERROR] No image given, aborting!" << endl;
+ return NULL;
+ }
+
+ // prepare data
+ char* req = (char*) gearman_job_workload(job);
+ size_t req_size = gearman_job_workload_size(job);
+ string *image = new string(req, req_size);
+ int current_port = 0;
+ cout << "[DEBUG] Requested image: " << *image << endl;
+
+ // test if requested image exists
+ int test = access(image->c_str(), 0);
+ if (test != 0)
+ {
+ cerr << "[ERROR] Requested image not found, aborting!" << endl;
+ char msg[] = "Requested image not found";
+ gearman_job_send_data(job, msg, sizeof(msg));
+ *ret_ptr = GEARMAN_WORK_FAIL;
+ return NULL;
+ }
+
+ if (_map[*image] != 0)
+ {
+ // nbd-server already running
+ cout << "[DEBUG] Requested nbd-server is already running on port "
+ << _map[*image] << endl;
+ current_port = _map[*image];
+ }
+ else
+ {
+ // start new nbd-server
+ cout << "[DEBUG] Starting new nbd-server on port " << _nbd_server_port
+ << endl;
+
+ stringstream command;
+ command << "nbd-server " << _nbd_server_port << " " << *image;
+ cout << "[DEBUG] " << "Running command: " << command.str() << endl;
+ system(command.str().c_str());
+
+ current_port = _nbd_server_port;
+ _map[*image] = _nbd_server_port;
+ _nbd_server_port++;
+ }
+
+ // send port number to client
+ cout << "[DEBUG] " << "Response with port number " << current_port << endl;
+ char buf[32];
+ snprintf(buf, sizeof(buf), "%d", current_port);
+ gearman_job_send_data(job, buf, strlen(buf));
+
+ *ret_ptr = GEARMAN_SUCCESS;
+ return NULL;
+}
+
+static void usage(char *name)
+{
+ printf("usage: %s [-p] [-d] [-v]\n", name);
+ printf("\t-p\t first port number for nbd-servers (default: 5001)\n");
+ printf("\t-d\t start as daemon\n");
+ printf("\t-v\t print version and exit\n");
+}
+
+int main(int argc, char* argv[])
+{
+ gearman_return_t result;
+ gearman_worker_st worker;
+ _nbd_server_port = 5001;
+ bool as_daemon = false;
+
+ // parse command line arguments
+ int c;
+ while ((c = getopt(argc, argv, "p:dv")) != -1)
+ {
+ switch (c)
+ {
+ case 'p':
+ _nbd_server_port = atoi(optarg);
+ break;
+ case 'd':
+ as_daemon = true;
+ break;
+ case 'v':
+ printf("Version: %s\n", VERSION_STRING);
+ exit(0);
+ default:
+ usage(argv[0]);
+ exit(1);
+ }
+ }
+
+ // initialize gearman worker
+ if (gearman_worker_create(&worker) == NULL)
+ {
+ cerr << "[ERROR] Memory allocation failure on worker creation" << endl;
+ exit(1);
+ }
+ result = gearman_worker_add_server(&worker, "127.0.0.1", 0);
+ if (result != GEARMAN_SUCCESS)
+ {
+ cerr << "[ERROR] " << gearman_worker_error(&worker) << endl;
+ exit(1);
+ }
+ result = gearman_worker_add_function(&worker, "start_nbd_server", 0,
+ start_nbd_server, NULL);
+ if (result != GEARMAN_SUCCESS)
+ {
+ cerr << "[ERROR] " << gearman_worker_error(&worker) << endl;
+ exit(1);
+ }
+ cout << "[DEBUG] Worker is ready." << endl;
+ if (as_daemon)
+ {
+ cout << "[DEBUG] Daemonize..." << endl;
+ daemon(0, 0);
+ }
+
+ // main loop
+ while (1)
+ {
+ result = gearman_worker_work(&worker);
+ if (result != GEARMAN_SUCCESS)
+ {
+ cerr << "[ERROR] " << gearman_worker_error(&worker) << endl;
+ break;
+ }
+ }
+
+ // cleanup
+ gearman_worker_free(&worker);
+ cout << "[DEBUG] Worker has been freed." << endl;
+}
generated by cgit v1.2.3-55-g7522 (git 2.39.0) at 2024-04-28 03:24:51 +0200