X-Git-Url: https://gerrit.automotivelinux.org/gerrit/gitweb?a=blobdiff_plain;f=pws%2Fpws.cpp;fp=pws%2Fpws.cpp;h=78796cb06417db99e82ca105552076bc16581d81;hb=cd39ff2c9987904617c53685a75f5fe89513ce81;hp=0000000000000000000000000000000000000000;hpb=85a1f5f0305fda98d7077bab78070456bcd680ac;p=apps%2Fonscreenapp.git

diff --git a/pws/pws.cpp b/pws/pws.cpp
new file mode 100644
index 0000000..78796cb
--- /dev/null
+++ b/pws/pws.cpp
@@ -0,0 +1,501 @@
+/*
+ * Copyright (c) 2020 Collabora Ltd.
+ *
+ * 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 <cstdio>
+#include <cstdlib>
+
+#include <systemd/sd-event.h>
+#include <json-c/json.h>
+#include <assert.h>
+#include "pws.h"
+
+#if !defined(JSON_C_TO_STRING_NOSLASHESCAPE)
+#define JSON_C_TO_STRING_NOSLASHESCAPE 0
+#endif
+
+#define PWS_DEBUG
+
+static void
+idle(struct pws_data_source *pws_d_source)
+{
+	struct sd_event *loop = pws_d_source->loop;
+
+	for (;;) {
+		if (!pws_d_source->callcount)
+			break;
+
+		sd_event_run(loop, TIMEOUT_SD_LOOP);
+	}
+}
+
+static void
+dec_callcount(struct pws_data_source *pws_d_source)
+{
+	if (!pws_d_source)
+		return;
+
+	pws_d_source->callcount--;
+}
+
+static void
+inc_callcount(struct pws_data_source *pws_d_source)
+{
+	if (!pws_d_source)
+		return;
+
+	pws_d_source->callcount++;
+}
+
+static void
+on_pws_reply(void *closure, void *request, struct json_object *result,
+	     const char *error, const char *info)
+{
+	struct pws_data_source *pws_d_source =
+		static_cast<struct pws_data_source *>(closure);
+
+	assert(pws_d_source != NULL);
+
+#ifdef PWS_DEBUG
+	fprintf(stdout, "ON-REPLY %s: %s %s\n%s\n", (char*) request, error,
+			info ?: "",
+			json_object_to_json_string_ext(result,
+				JSON_C_TO_STRING_PRETTY | JSON_C_TO_STRING_NOSLASHESCAPE));
+
+	fflush(stdout);
+#endif
+
+	/* in case of an error do not set the reply */
+	if (!info && !error) {
+		/* should be cleaned-up after a proper request */
+		assert(pws_d_source->reply_valid == false);
+
+		pws_d_source->reply = result;
+		pws_d_source->reply_valid = true;
+	} else {
+		fprintf(stdout, "ON-REPLY: err: %s, request: %s, info %s\n",
+				error, (char *) request, info);
+	}
+
+	/* necessary when getting the reply to exit idle() */
+	dec_callcount(pws_d_source);
+}
+
+static void
+on_pws_event_create(void *closure, uint16_t event_id, const char *event_name)
+{
+	struct pws_data_source *pws_d_source =
+		static_cast<struct pws_data_source *>(closure);
+	(void) pws_d_source;
+
+#ifdef PWS_DEBUG
+	fprintf(stdout, "ON-EVENT-CREATE: [%d:%s]\n", event_id, event_name);
+	fflush(stdout);
+#endif
+}
+
+static void
+on_pws_event_remove(void *closure, uint16_t event_id)
+{
+	struct pws_data_source *pws_d_source =
+		static_cast<struct pws_data_source *>(closure);
+
+	(void) pws_d_source;
+#ifdef PWS_DEBUG
+	fprintf(stdout, "ON-EVENT-REMOVE: [%d]\n", event_id);
+	fflush(stdout);
+#endif
+}
+
+static void
+on_pws_event_subscribe(void *closure, void *request, uint16_t event_id)
+{
+	struct pws_data_source *pws_d_source =
+		static_cast<struct pws_data_source *>(closure);
+
+	(void) pws_d_source;
+#ifdef PWS_DEBUG
+	fprintf(stdout, "ON-EVENT-SUBSCRIBE %s: [%d]\n", (char*)request, event_id);
+	fflush(stdout);
+#endif
+}
+
+static void
+on_pws_event_unsubscribe(void *closure, void *request, uint16_t event_id)
+{
+	struct pws_data_source *pws_d_source =
+		static_cast<struct pws_data_source *>(closure);
+
+	(void) pws_d_source;
+#ifdef PWS_DEBUG
+	fprintf(stdout, "ON-EVENT-UNSUBSCRIBE %s: [%d]\n", (char*)request, event_id);
+	fflush(stdout);
+#endif
+}
+
+static void
+on_pws_event_push(void *closure, uint16_t event_id, struct json_object *data)
+{
+	struct pws_data_source *pws_d_source =
+		static_cast<struct pws_data_source *>(closure);
+
+	(void) pws_d_source;
+#ifdef PWS_DEBUG
+	fprintf(stdout, "ON-EVENT-PUSH: [%d]\n%s\n",
+			event_id,
+			json_object_to_json_string_ext(data, JSON_C_TO_STRING_NOSLASHESCAPE));
+	fprintf(stdout, "ON-EVENT-PUSH: [%d]\n%s\n",
+			event_id,
+			json_object_to_json_string_ext(data, JSON_C_TO_STRING_PRETTY |
+							     JSON_C_TO_STRING_NOSLASHESCAPE));
+
+	fflush(stdout);
+#endif
+}
+
+static void
+on_pws_event_broadcast(void *closure, const char *event_name,
+		      struct json_object *data,
+		      const afb_proto_ws_uuid_t uuid, uint8_t hop)
+{
+	struct pws_data_source *pws_d_source =
+		static_cast<struct pws_data_source *>(closure);
+
+	(void) pws_d_source;
+	(void) uuid;
+	(void) hop;
+
+#ifdef PWS_DEBUG
+	fprintf(stdout, "ON-EVENT-BROADCAST: [%s]\n%s\n",
+			event_name,
+			json_object_to_json_string_ext(data, JSON_C_TO_STRING_NOSLASHESCAPE));
+	fprintf(stdout, "ON-EVENT-BROADCAST: [%s]\n%s\n",
+			event_name,
+			json_object_to_json_string_ext(data, JSON_C_TO_STRING_PRETTY |
+							     JSON_C_TO_STRING_NOSLASHESCAPE));
+	fflush(stdout);
+#endif
+}
+
+/* called when pws hangsup */
+static void
+on_pws_hangup(void *closure)
+{
+	struct pws_data_source *pws_d_source =
+		static_cast<struct pws_data_source *>(closure);
+
+	(void) pws_d_source;
+#ifdef PWS_DEBUG
+	printf("ON-HANGUP\n");
+	fflush(stdout);
+#endif
+
+	exit(EXIT_FAILURE);
+}
+
+/* makes a call */
+static int
+pws_call(const char *verb, const char *object,
+	 struct pws_data_source *pws_d_source)
+{
+	char *key;
+	struct json_object *o;
+	enum json_tokener_error jerr;
+
+	struct afb_proto_ws *pws = pws_d_source->pws;
+	int rc;
+
+	assert(pws != NULL);
+
+	/* allocates an id for the request */
+	rc = asprintf(&key, "%d:%s", ++pws_d_source->num_id, verb);
+	if (rc < 0)
+		return -1;
+
+	/* echo the command if asked */
+	fprintf(stdout, "SEND-CALL: %s %s\n", verb, object?:"null");
+
+	inc_callcount(pws_d_source);
+
+	if (object == NULL || object[0] == 0) {
+		o = NULL;
+	} else {
+		o = json_tokener_parse_verbose(object, &jerr);
+		if (jerr != json_tokener_success)
+			o = json_object_new_string(object);
+	}
+
+	/* send the request */
+	rc = afb_proto_ws_client_call(pws, verb, o, 0, 0, key, NULL);
+	json_object_put(o);
+	if (rc < 0) {
+		fprintf(stderr, "calling %s(%s) failed: %m\n", verb, object ?: "");
+		dec_callcount(pws_d_source);
+		free(key);
+		return -1;
+	}
+
+	free(key);
+	return 0;
+}
+
+/* the callback interface for pws */
+static struct afb_proto_ws_client_itf pws_itf = {
+	.on_reply = on_pws_reply,
+	.on_event_create = on_pws_event_create,
+	.on_event_remove = on_pws_event_remove,
+	.on_event_subscribe = on_pws_event_subscribe,
+	.on_event_unsubscribe = on_pws_event_unsubscribe,
+	.on_event_push = on_pws_event_push,
+	.on_event_broadcast = on_pws_event_broadcast,
+};
+
+struct pws_data_source *
+pws_data_source_init(const char *connect_to)
+{
+	struct afb_proto_ws *pws;
+	struct sd_event *sd_loop;
+	struct pws_data_source *pws_d_source = nullptr;
+
+	if (!connect_to) {
+		fprintf(stderr, "Failed to get a connect_to\n");
+		return nullptr;
+	}
+
+	if (sd_event_default(&sd_loop) < 0) {
+		fprintf(stderr, "Failed to get a default event\n");
+		return nullptr;
+	}
+
+	pws_d_source =
+		static_cast<struct pws_data_source *>(calloc(1, sizeof(struct pws_data_source)));
+
+	if (!pws_d_source) {
+		fprintf(stderr, "Failed to allocate memory\n");
+		return nullptr;
+	}
+
+	pws = afb_ws_client_connect_api(sd_loop, connect_to,
+					&pws_itf, pws_d_source);
+	if (!pws) {
+		fprintf(stderr, "Failed to create a afb_proto_ws\n");
+		return nullptr;
+	}
+
+	afb_proto_ws_on_hangup(pws, on_pws_hangup);
+
+	pws_d_source->pws = pws;
+	pws_d_source->loop = sd_loop;
+	pws_d_source->callcount = 0;
+	pws_d_source->num_id = 0;
+
+	pws_d_source->reply = nullptr;
+	pws_d_source->reply_valid = false;
+
+	return pws_d_source;
+}
+
+static int
+pws_do_call(struct pws_data_source *pws_d_source,
+	    const char *verb, const char *object)
+{
+	if (!verb || !object || !pws_d_source)
+		return -1;
+
+	if (pws_call(verb, object, pws_d_source) < 0)
+		return -1;
+
+	idle(pws_d_source);
+
+	return 0;
+}
+
+void
+pws_data_source_destroy(struct pws_data_source *pws_d_source)
+{
+	assert(pws_d_source != nullptr);
+
+	sd_event_unref(pws_d_source->loop);
+	free(pws_d_source);
+}
+
+void
+pws_data_source_reply_destroy(struct pws_data_source *pws)
+{
+	assert(pws->reply_valid == true);
+	pws->reply_valid = false;
+	free(pws->reply);
+}
+
+
+int
+pws_start_process(struct pws_data_source *pws, const char *afm_name)
+{
+	int rid = -1;
+
+	fprintf(stdout, "pws_start_process() with afm_name %s\n", afm_name);
+	if (pws_do_call(pws, "start", afm_name) < 0)
+		return -1;
+
+	if (pws->reply_valid) {
+		struct json_object *result = pws->reply;
+
+		rid = json_object_get_int(result);
+		fprintf(stdout, "ON-REPLY %s\n",
+				json_object_to_json_string_ext(result,
+					JSON_C_TO_STRING_PRETTY | JSON_C_TO_STRING_NOSLASHESCAPE));
+
+		pws_data_source_reply_destroy(pws);
+	}
+
+	return rid;
+}
+
+bool
+pws_stop_process(struct pws_data_source *pws, const char *afm_name)
+{
+	bool term = false;
+	if (pws_do_call(pws, "terminate", afm_name) < 0)
+		return term;
+
+	if (pws->reply_valid) {
+		struct json_object *result = pws->reply;
+
+		term = json_object_get_boolean(result);
+#ifdef PWS_DEBUG
+		fprintf(stdout, "ON-REPLY %s\n",
+				json_object_to_json_string_ext(result,
+				JSON_C_TO_STRING_PRETTY | JSON_C_TO_STRING_NOSLASHESCAPE));
+#endif
+		pws_data_source_reply_destroy(pws);
+	}
+
+	return term;
+}
+
+
+int
+pws_check_process_is_running(struct pws_data_source *pws, const char *afm_name)
+{
+	int rid = -1;
+
+	if (pws_do_call(pws, "state", afm_name) < 0)
+		return rid;
+
+	if (pws->reply_valid) {
+		struct json_object *result = pws->reply;
+		struct json_object *obj;
+
+		json_object_object_get_ex(result, "runid", &obj);
+		if (obj) {
+			rid = json_object_get_int(obj);
+			fprintf(stdout, "Found rid %d\n", rid);
+		}
+#ifdef PWS_DEBUG
+		fprintf(stdout, "ON-REPLY: %s\n",
+				json_object_to_json_string_ext(result,
+				JSON_C_TO_STRING_PRETTY | JSON_C_TO_STRING_NOSLASHESCAPE));
+#endif
+		pws_data_source_reply_destroy(pws);
+	}
+
+	return rid;
+}
+
+size_t
+pws_list_runners(struct pws_data_source *pws)
+{
+	size_t items = 0;
+	if (pws_do_call(pws, "runners", "true") < 0)
+		return items;
+
+	if (pws->reply_valid) {
+		struct json_object *result = pws->reply;
+
+		/* at least one is running */
+		items = json_object_array_length(result);
+
+#ifdef PWS_DEBUG
+		fprintf(stdout, "found %ld items\n", items);
+		fprintf(stdout, "ON-REPLY: %s\n",
+				json_object_to_json_string_ext(result,
+				JSON_C_TO_STRING_PRETTY | JSON_C_TO_STRING_NOSLASHESCAPE));
+#endif
+		pws_data_source_reply_destroy(pws);
+	}
+
+	return items;
+}
+
+size_t
+pws_list_runnables(struct pws_data_source *pws)
+{
+	size_t items = 0;
+	if (pws_do_call(pws, "runnables", "true") < 0)
+		return items;
+
+	if (pws->reply_valid) {
+		struct json_object *result = pws->reply;
+
+		items = json_object_array_length(result);
+#ifdef PWS_DEBUG
+		fprintf(stdout, "found %ld items\n", items);
+		fprintf(stdout, "ON-REPLY: %s\n",
+				json_object_to_json_string_ext(result,
+				JSON_C_TO_STRING_PRETTY | JSON_C_TO_STRING_NOSLASHESCAPE));
+#endif
+		pws_data_source_reply_destroy(pws);
+	}
+
+	return items;
+}
+
+/*
+ * need to free(json_string) once you're done with it
+ *
+ * call pws_data_source_reply_destroy(pws) once you're done with it.
+ */
+size_t
+pws_get_list_runnables(struct pws_data_source *pws,  struct json_object **json)
+{
+	size_t items = 0;
+	if (pws_do_call(pws, "runnables", "true") < 0)
+		return items;
+
+	fprintf(stdout, "pws_get_list_runnables()\n");
+
+	if (pws->reply_valid) {
+		struct json_object *result = pws->reply;
+
+		items = json_object_array_length(result);
+#ifdef PWS_DEBUG
+		fprintf(stdout, "found %ld items\n", items);
+		fprintf(stdout, "ON-REPLY: %s\n",
+				json_object_to_json_string_ext(result,
+				JSON_C_TO_STRING_PRETTY | JSON_C_TO_STRING_NOSLASHESCAPE));
+#endif
+		if (items == 0) {
+			fprintf(stdout, "pws_get_list_runnables() turn items %ld, bails sooner\n", items);
+			*json = NULL;
+			return items;
+		}
+
+		fprintf(stdout, "pws_get_list_runnables() json reply is set\n");
+		*json = result;
+	}
+
+	fprintf(stdout, "pws_get_list_runnables() turn items %ld\n", items);
+	return items;
+}