WebSVN – HelenOS – Diff – /trunk/uspace/srv/ns/ns.c

 #include <ipc/ipc.h>
 #include <ipc/ns.h>
 #include <ipc/services.h>
 #include <stdio.h>
+#include <bool.h>
 #include <unistd.h>
 #include <stdlib.h>
 #include <errno.h>
 #include <assert.h>
 #include <libadt/list.h>
 #include <libadt/hash_table.h>
 #include <sysinfo.h>
+#include <loader/loader.h>
 #include <ddi.h>
 #include <as.h>
 #define NAME    "ns"
 static int register_service(ipcarg_t service, ipcarg_t phone, ipc_call_t *call);
 static int connect_to_service(ipcarg_t service, ipc_call_t *call,
     ipc_callid_t callid);
+void register_clonable(ipcarg_t service, ipcarg_t phone, ipc_call_t *call,
+    ipc_callid_t callid);
+void connect_to_clonable(ipcarg_t service, ipc_call_t *call,
+    ipc_callid_t callid);
 /* Static functions implementing NS hash table operations. */
 static hash_index_t ns_hash(unsigned long *key);
 static int ns_compare(unsigned long *key, hash_count_t keys, link_t *item);
 static void ns_remove(link_t *item);
 } hashed_service_t;
 static void *clockaddr = NULL;
 static void *klogaddr = NULL;
+/** Request for connection to a clonable service. */
+typedef struct {
+    link_t link;
+    ipcarg_t service;
+    ipc_call_t *call;
+    ipc_callid_t callid;
+} cs_req_t;
+/** List of clonable-service connection requests. */
+static link_t cs_req;
+/** Return true if @a service is clonable. */
+static bool service_clonable(int service)
+{
+    return service == SERVICE_LOAD;
+}
 static void get_as_area(ipc_callid_t callid, ipc_call_t *call, char *name,
     void **addr)
+{
     void *ph_addr;
     if (!hash_table_create(&ns_hash_table, NS_HASH_TABLE_CHAINS, 3,
         &ns_hash_table_ops)) {
         printf(NAME ": No memory available\n");
         return ENOMEM;
+    }
+    list_initialize(&cs_req);
     printf(NAME ": Accepting connections\n");
     while (1) {
         callid = ipc_wait_for_call(&call);
         switch (IPC_GET_METHOD(call)) {
             break;
         case IPC_M_CONNECT_TO_ME:
             /*
              * Server requests service registration.
              */
+            if (service_clonable(IPC_GET_ARG1(call))) {
+                register_clonable(IPC_GET_ARG1(call),
+                    IPC_GET_ARG5(call), &call, callid);
+                continue;
+            } else {
-            retval = register_service(IPC_GET_ARG1(call),
+                retval = register_service(IPC_GET_ARG1(call),
-                IPC_GET_ARG5(call), &call);
+                    IPC_GET_ARG5(call), &call);
+            }
             break;
         case IPC_M_CONNECT_ME_TO:
             /*
              * Client requests to be connected to a service.
              */
+            if (service_clonable(IPC_GET_ARG1(call))) {
+                connect_to_clonable(IPC_GET_ARG1(call),
+                    &call, callid);
+                continue;
+            } else {
-            retval = connect_to_service(IPC_GET_ARG1(call), &call,
+                retval = connect_to_service(IPC_GET_ARG1(call),
-                callid);
+                    &call, callid);
+            }
             break;
         default:
             retval = ENOENT;
             break;
+        }
     return 0;
+}
 /** Register service.
+ *
- * @param service Service to be registered.
+ * @param service   Service to be registered.
- * @param phone Phone to be used for connections to the service.
+ * @param phone     Phone to be used for connections to the service.
- * @param call Pointer to call structure.
+ * @param call      Pointer to call structure.
+ *
- * @return Zero on success or a value from @ref errno.h.
+ * @return      Zero on success or a value from @ref errno.h.
  */
 int register_service(ipcarg_t service, ipcarg_t phone, ipc_call_t *call)
+{
     unsigned long keys[3] = {
         service,
         call->in_phone_hash,
     };
     hashed_service_t *hs;
     if (hash_table_find(&ns_hash_table, keys)) {
         return EEXISTS;
+    }
     hs = (hashed_service_t *) malloc(sizeof(hashed_service_t));
     return 0;
+}
 /** Connect client to service.
+ *
- * @param service Service to be connected to.
+ * @param service   Service to be connected to.
- * @param call Pointer to call structure.
+ * @param call      Pointer to call structure.
- * @param callid Call ID of the request.
+ * @param callid    Call ID of the request.
+ *
  * @return Zero on success or a value from @ref errno.h.
  */
 int connect_to_service(ipcarg_t service, ipc_call_t *call, ipc_callid_t callid)
+{
     unsigned long keys[3] = { service, 0, 0 };
     link_t *hlp;
     hashed_service_t *hs;
     hlp = hash_table_find(&ns_hash_table, keys);
     if (!hlp) {
         return ENOENT;
+    }
     hs = hash_table_get_instance(hlp, hashed_service_t, link);
     return ipc_forward_fast(callid, hs->phone, IPC_GET_ARG2(*call),
         IPC_GET_ARG3(*call), 0, IPC_FF_NONE);
+}
+/** Register clonable service.
+ *
+ * @param service   Service to be registered.
+ * @param phone     Phone to be used for connections to the service.
+ * @param call      Pointer to call structure.
+ */
+void register_clonable(ipcarg_t service, ipcarg_t phone, ipc_call_t *call,
+    ipc_callid_t callid)
+{
+    int rc;
+    cs_req_t *csr;
+    if (list_empty(&cs_req)) {
+        /* There was no pending connection request. */
+        printf(NAME ": Unexpected clonable server.\n");
+        ipc_answer_0(callid, EBUSY);
+        return;
+    }
+    csr = list_get_instance(cs_req.next, cs_req_t, link);
+    list_remove(&csr->link);
+    /* Currently we can only handle a single type of clonable service. */
+    assert(csr->service == SERVICE_LOAD);
+    ipc_answer_0(callid, EOK);
+    rc = ipc_forward_fast(csr->callid, phone, IPC_GET_ARG2(*csr->call),
+        IPC_GET_ARG3(*csr->call), 0, IPC_FF_NONE);
+    free(csr);
+}
+/** Connect client to clonable service.
+ *
+ * @param service   Service to be connected to.
+ * @param call      Pointer to call structure.
+ * @param callid    Call ID of the request.
+ *
+ * @return      Zero on success or a value from @ref errno.h.
+ */
+void connect_to_clonable(ipcarg_t service, ipc_call_t *call,
+    ipc_callid_t callid)
+{
+    int rc;
+    cs_req_t *csr;
+    assert(service == SERVICE_LOAD);
+    csr = malloc(sizeof(cs_req_t));
+    if (csr == NULL) {
+        ipc_answer_0(callid, ENOMEM);
+        return;
+    }
+    /* Spawn a loader. */
+    rc = loader_spawn("loader");
+    if (rc < 0) {
+        free(csr);
+        ipc_answer_0(callid, rc);
+        return;
+    }
+    csr->service = service;
+    csr->call = call;
+    csr->callid = callid;
+    /*
+     * We can forward the call only after the server we spawned connects
+     * to us. Meanwhile we might need to service more connection requests.
+     * Thus we store the call in a queue.
+     */
+    list_append(&csr->link, &cs_req);
+}
 /** Compute hash index into NS hash table.
+ *
  * @param key Pointer keys. However, only the first key (i.e. service number)
  *        is used to compute the hash index.
  * @return Hash index corresponding to key[0].

Subversion Repositories HelenOS

(root)/trunk/uspace/srv/ns/ns.c – Rev 3083 → 3896