+static void
+ubus_sub_remove_handler(struct ubus_context *ctx, struct ubus_subscriber *s,
+ uint32_t id)
+{
+ struct ubus_lua_subscriber *sub;
+
+ sub = container_of(s, struct ubus_lua_subscriber, s);
+
+ lua_getglobal(state, "__ubus_cb_subscribe");
+ lua_rawgeti(state, -1, sub->rremove);
+ lua_remove(state, -2);
+
+ if (lua_isfunction(state, -1)) {
+ lua_call(state, 0, 0);
+ } else {
+ lua_pop(state, 1);
+ }
+}
+
+static int
+ubus_sub_notify_handler(struct ubus_context *ctx, struct ubus_object *obj,
+ struct ubus_request_data *req, const char *method,
+ struct blob_attr *msg)
+{
+ struct ubus_subscriber *s;
+ struct ubus_lua_subscriber *sub;
+
+ s = container_of(obj, struct ubus_subscriber, obj);
+ sub = container_of(s, struct ubus_lua_subscriber, s);
+
+ lua_getglobal(state, "__ubus_cb_subscribe");
+ lua_rawgeti(state, -1, sub->rnotify);
+ lua_remove(state, -2);
+
+ if (lua_isfunction(state, -1)) {
+ if( msg ){
+ ubus_lua_parse_blob_array(state, blob_data(msg), blob_len(msg), true);
+ lua_call(state, 1, 0);
+ } else {
+ lua_call(state, 0, 0);
+ }
+ } else {
+ lua_pop(state, 1);
+ }
+
+ return 0;
+}
+
+
+
+static void
+ubus_lua_do_subscribe( struct ubus_context *ctx, lua_State *L, const char* target,
+ int idxnotify, int idxremove )
+{
+ uint32_t id;
+ int status;
+ struct ubus_lua_subscriber *sub;
+
+ if( ( status = ubus_lookup_id( ctx, target, &id ) ) ){
+ lua_pushfstring( L, "Unable find target, status=%d", status );
+ lua_error( L );
+ }
+
+ sub = malloc( sizeof( struct ubus_lua_subscriber ) );
+ memset( sub, 0, sizeof( struct ubus_lua_subscriber ) );
+ if( !sub ){
+ lua_pushstring( L, "Out of memory" );
+ lua_error( L );
+ }
+
+ if( idxnotify ){
+ lua_getglobal(L, "__ubus_cb_subscribe");
+ lua_pushvalue(L, idxnotify);
+ sub->rnotify = luaL_ref(L, -2);
+ lua_pop(L, 1);
+ sub->s.cb = ubus_sub_notify_handler;
+ }
+
+ if( idxremove ){
+ lua_getglobal(L, "__ubus_cb_subscribe");
+ lua_pushvalue(L, idxnotify);
+ sub->rnotify = luaL_ref(L, -2);
+ lua_pop(L, 1);
+ sub->s.remove_cb = ubus_sub_remove_handler;
+ }
+
+ if( ( status = ubus_register_subscriber( ctx, &sub->s ) ) ){
+ lua_pushfstring( L, "Failed to register subscriber, status=%d", status );
+ lua_error( L );
+ }
+
+ if( ( status = ubus_subscribe( ctx, &sub->s, id) ) ){
+ lua_pushfstring( L, "Failed to register subscriber, status=%d", status );
+ lua_error( L );
+ }
+}
+
+static int
+ubus_lua_subscribe(lua_State *L) {
+ int idxnotify, idxremove, stackstart;
+ struct ubus_lua_connection *c;
+ const char* target;
+
+ idxnotify = idxremove = 0;
+ stackstart = lua_gettop( L );
+
+
+ c = luaL_checkudata(L, 1, METANAME);
+ target = luaL_checkstring(L, 2);
+ luaL_checktype(L, 3, LUA_TTABLE);
+
+
+ lua_pushstring( L, "notify");
+ lua_gettable( L, 3 );
+ if( lua_type( L, -1 ) == LUA_TFUNCTION ){
+ idxnotify = lua_gettop( L );
+ } else {
+ lua_pop( L, 1 );
+ }
+
+ lua_pushstring( L, "remove");
+ lua_gettable( L, 3 );
+ if( lua_type( L, -1 ) == LUA_TFUNCTION ){
+ idxremove = lua_gettop( L );
+ } else {
+ lua_pop( L, 1 );
+ }
+
+ if( idxnotify )
+ ubus_lua_do_subscribe( c->ctx, L, target, idxnotify, idxremove );
+
+ if( lua_gettop( L ) > stackstart )
+ lua_pop( L, lua_gettop( L ) - stackstart );
+
+ return 0;
+}
+