X-Git-Url: https://gerrit.automotivelinux.org/gerrit/gitweb?a=blobdiff_plain;f=src%2Fafb-stub-ws.c;h=ce68b441cc736fc40b980f891e6a544a827ae4b3;hb=8406a51db98b563a4562f26de8338efc0478aff2;hp=351b332010d6f4ad1510ecf46779a44beb42e6af;hpb=53c7b44e96e6015a4da4f88f22e98b13cfa2e056;p=src%2Fapp-framework-binder.git

diff --git a/src/afb-stub-ws.c b/src/afb-stub-ws.c
index 351b3320..ce68b441 100644
--- a/src/afb-stub-ws.c
+++ b/src/afb-stub-ws.c
@@ -144,7 +144,7 @@ struct server_describe
 	uint32_t descid;
 };
 
-/******************* client description part for server *****************************/
+/******************* stub description for client or servers ******************/
 
 struct afb_stub_ws
 {
@@ -181,6 +181,9 @@ struct afb_stub_ws
 	/* apiset */
 	struct afb_apiset *apiset;
 
+	/* on hangup callback */
+	void (*on_hangup)(struct afb_stub_ws *);
+
 	/* the api name */
 	char apiname[1];
 };
@@ -1205,6 +1208,8 @@ static void server_on_hangup(void *closure)
 	if (stubws->fd >= 0) {
 		close(stubws->fd);
 		stubws->fd = -1;
+		if (stubws->on_hangup)
+			stubws->on_hangup(stubws);
 	}
 
 	/* release the client */
@@ -1272,19 +1277,7 @@ static struct afb_stub_ws *afb_stub_ws_create(int fd, const char *apiname, struc
 
 struct afb_stub_ws *afb_stub_ws_create_client(int fd, const char *apiname, struct afb_apiset *apiset)
 {
-	struct afb_api afb_api;
-	struct afb_stub_ws *stubws;
-
-	stubws = afb_stub_ws_create(fd, apiname, apiset, &stub_ws_client_ws_itf);
-	if (stubws) {
-		afb_api.closure = stubws;
-		afb_api.itf = &ws_api_itf;
-		if (afb_apiset_add(apiset, stubws->apiname, afb_api) >= 0)
-			return stubws;
-		afb_stub_ws_unref(stubws);
-	}
-	return NULL;
-
+	return afb_stub_ws_create(fd, apiname, apiset, &stub_ws_client_ws_itf);
 }
 
 struct afb_stub_ws *afb_stub_ws_create_server(int fd, const char *apiname, struct afb_apiset *apiset)
@@ -1327,3 +1320,28 @@ void afb_stub_ws_addref(struct afb_stub_ws *stubws)
 	__atomic_add_fetch(&stubws->refcount, 1, __ATOMIC_RELAXED);
 }
 
+void afb_stub_ws_on_hangup(struct afb_stub_ws *stubws, void (*on_hangup)(struct afb_stub_ws*))
+{
+	stubws->on_hangup = on_hangup;
+}
+
+const char *afb_stub_ws_name(struct afb_stub_ws *stubws)
+{
+	return stubws->apiname;
+}
+
+struct afb_api afb_stub_ws_client_api(struct afb_stub_ws *stubws)
+{
+	struct afb_api api;
+
+	assert(!stubws->listener); /* check client */
+	api.closure = stubws;
+	api.itf = &ws_api_itf;
+	return api;
+}
+
+int afb_stub_ws_client_add(struct afb_stub_ws *stubws, struct afb_apiset *apiset)
+{
+	return afb_apiset_add(apiset, stubws->apiname, afb_stub_ws_client_api(stubws));
+}
+