Add support for dumping filesystem state for debugging purposes.
[bluesky.git] / nfs3 / rpc.c
index d47943c..468c58c 100644 (file)
@@ -121,13 +121,73 @@ struct rpc_fail_reply {
     uint32_t accept_stat;
 };
 
+/* Routines for XDR-encoding to a growable string. */
+static bool_t xdr_string_putlong(XDR *xdrs, const long *lp)
+{
+    GString *str = (GString *)xdrs->x_private;
+    uint32_t data = htonl(*lp);
+    g_string_set_size(str, str->len + 4);
+    memcpy(str->str + str->len - 4, &data, 4);
+    return TRUE;
+}
+
+static bool_t xdr_string_putbytes(XDR *xdrs, const char *addr, u_int len)
+{
+    GString *str = (GString *)xdrs->x_private;
+    g_string_set_size(str, str->len + len);
+    memcpy(str->str + str->len - len, addr, len);
+    return TRUE;
+}
+
+static u_int xdr_string_getpos(const XDR *xdrs)
+{
+    GString *str = (GString *)xdrs->x_private;
+    return str->len;
+}
+
+static bool_t xdr_string_putint32(XDR *xdrs, const int32_t *ip)
+{
+    GString *str = (GString *)xdrs->x_private;
+    uint32_t data = htonl(*ip);
+    g_string_set_size(str, str->len + 4);
+    memcpy(str->str + str->len - 4, &data, 4);
+    return TRUE;
+}
+
+static int32_t *xdr_string_inline(XDR *xdrs, u_int len)
+{
+    GString *str = (GString *)xdrs->x_private;
+    g_string_set_size(str, str->len + len);
+    return (int32_t *)(str->str + str->len - len);
+}
+
+static void xdr_string_destroy(XDR *xdrs)
+{
+}
+
+static struct xdr_ops xdr_string_ops = {
+    .x_putlong = xdr_string_putlong,
+    .x_putbytes = xdr_string_putbytes,
+    .x_getpostn = xdr_string_getpos,
+    .x_putint32 = xdr_string_putint32,
+    .x_inline = xdr_string_inline,
+    .x_destroy = xdr_string_destroy,
+};
+
+static void xdr_string_create(XDR *xdrs, GString *string, enum xdr_op op)
+{
+    xdrs->x_op = op;
+    xdrs->x_ops = &xdr_string_ops;
+    xdrs->x_private = (char *)string;
+    xdrs->x_base = NULL;
+    xdrs->x_handy = 0;
+}
+
 static void
 async_rpc_send_failure(RPCRequest *req, enum accept_stat stat)
 {
     struct rpc_fail_reply header;
 
-    fprintf(stderr, "Sending RPC failure status %d\n", stat);
-
     header.xid = htonl(req->xid);
     header.type = htonl(1);     /* REPLY */
     header.stat = htonl(MSG_ACCEPTED);
@@ -140,9 +200,51 @@ async_rpc_send_failure(RPCRequest *req, enum accept_stat stat)
     async_rpc_write(req->connection, (const char *)&header, sizeof(header));
     g_io_channel_flush(req->connection->channel, NULL);
 
+    if (req->args != NULL) {
+        char buf[4];
+        XDR xdr;
+        xdrmem_create(&xdr, buf, sizeof(buf), XDR_FREE);
+        if (!req->xdr_args_free(&xdr, req->args)) {
+            fprintf(stderr, "unable to free arguments");
+        }
+    }
+
     if (req->raw_args != NULL)
         g_string_free(req->raw_args, TRUE);
 
+    g_free(req);
+}
+
+void
+async_rpc_send_reply(RPCRequest *req, void *result)
+{
+    GString *str = g_string_new("");
+    XDR xdr_out;
+    xdr_string_create(&xdr_out, str, XDR_ENCODE);
+    if (!req->xdr_result(&xdr_out, result)) {
+        async_rpc_send_failure(req, SYSTEM_ERR);
+        g_string_free(str, TRUE);
+        return;
+    }
+
+    struct rpc_reply header;
+    header.xid = htonl(req->xid);
+    header.type = htonl(1);     /* REPLY */
+    header.stat = htonl(MSG_ACCEPTED);
+    header.verf_flavor = 0;
+    header.verf_len = 0;
+    header.accept_stat = 0;
+
+    gsize msg_size = str->len;
+    uint32_t fragment = htonl((msg_size + sizeof(header)) | 0x80000000);
+    async_rpc_write(req->connection, (const char *)&fragment, sizeof(fragment));
+    async_rpc_write(req->connection, (const char *)&header, sizeof(header));
+    async_rpc_write(req->connection, str->str, str->len);
+    g_io_channel_flush(req->connection->channel, NULL);
+
+    /* Clean up. */
+    g_string_free(str, TRUE);
+
     if (req->args != NULL) {
         char buf[4];
         XDR xdr;
@@ -152,6 +254,9 @@ async_rpc_send_failure(RPCRequest *req, enum accept_stat stat)
         }
     }
 
+    if (req->raw_args != NULL)
+        g_string_free(req->raw_args, TRUE);
+
     g_free(req);
 }
 
@@ -188,139 +293,141 @@ nfs_program_3(RPCRequest *req)
     };
     char *result;
     xdrproc_t _xdr_argument, _xdr_result;
-    char *(*local)(char *, struct svc_req *);
+    char *(*local)(char *, RPCRequest *);
+
+    printf("Dispatched NFS RPC message type %d\n", req->req_proc);
 
     switch (req->req_proc) {
     case NFSPROC3_NULL:
         _xdr_argument = (xdrproc_t) xdr_void;
         _xdr_result = (xdrproc_t) xdr_void;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_null_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_null_3_svc;
         break;
 
     case NFSPROC3_GETATTR:
         _xdr_argument = (xdrproc_t) xdr_nfs_fh3;
         _xdr_result = (xdrproc_t) xdr_getattr3res;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_getattr_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_getattr_3_svc;
         break;
 
     case NFSPROC3_SETATTR:
         _xdr_argument = (xdrproc_t) xdr_setattr3args;
         _xdr_result = (xdrproc_t) xdr_wccstat3;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_setattr_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_setattr_3_svc;
         break;
 
     case NFSPROC3_LOOKUP:
         _xdr_argument = (xdrproc_t) xdr_diropargs3;
         _xdr_result = (xdrproc_t) xdr_lookup3res;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_lookup_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_lookup_3_svc;
         break;
 
     case NFSPROC3_ACCESS:
         _xdr_argument = (xdrproc_t) xdr_access3args;
         _xdr_result = (xdrproc_t) xdr_access3res;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_access_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_access_3_svc;
         break;
 
     case NFSPROC3_READLINK:
         _xdr_argument = (xdrproc_t) xdr_nfs_fh3;
         _xdr_result = (xdrproc_t) xdr_readlink3res;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_readlink_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_readlink_3_svc;
         break;
 
     case NFSPROC3_READ:
         _xdr_argument = (xdrproc_t) xdr_read3args;
         _xdr_result = (xdrproc_t) xdr_read3res;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_read_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_read_3_svc;
         break;
 
     case NFSPROC3_WRITE:
         _xdr_argument = (xdrproc_t) xdr_write3args;
         _xdr_result = (xdrproc_t) xdr_write3res;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_write_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_write_3_svc;
         break;
 
     case NFSPROC3_CREATE:
         _xdr_argument = (xdrproc_t) xdr_create3args;
         _xdr_result = (xdrproc_t) xdr_diropres3;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_create_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_create_3_svc;
         break;
 
     case NFSPROC3_MKDIR:
         _xdr_argument = (xdrproc_t) xdr_mkdir3args;
         _xdr_result = (xdrproc_t) xdr_diropres3;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_mkdir_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_mkdir_3_svc;
         break;
 
     case NFSPROC3_SYMLINK:
         _xdr_argument = (xdrproc_t) xdr_symlink3args;
         _xdr_result = (xdrproc_t) xdr_diropres3;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_symlink_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_symlink_3_svc;
         break;
 
     case NFSPROC3_MKNOD:
         _xdr_argument = (xdrproc_t) xdr_mknod3args;
         _xdr_result = (xdrproc_t) xdr_diropres3;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_mknod_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_mknod_3_svc;
         break;
 
     case NFSPROC3_REMOVE:
         _xdr_argument = (xdrproc_t) xdr_diropargs3;
         _xdr_result = (xdrproc_t) xdr_wccstat3;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_remove_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_remove_3_svc;
         break;
 
     case NFSPROC3_RMDIR:
         _xdr_argument = (xdrproc_t) xdr_diropargs3;
         _xdr_result = (xdrproc_t) xdr_wccstat3;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_rmdir_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_rmdir_3_svc;
         break;
 
     case NFSPROC3_RENAME:
         _xdr_argument = (xdrproc_t) xdr_rename3args;
         _xdr_result = (xdrproc_t) xdr_rename3res;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_rename_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_rename_3_svc;
         break;
 
     case NFSPROC3_LINK:
         _xdr_argument = (xdrproc_t) xdr_link3args;
         _xdr_result = (xdrproc_t) xdr_link3res;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_link_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_link_3_svc;
         break;
 
     case NFSPROC3_READDIR:
         _xdr_argument = (xdrproc_t) xdr_readdir3args;
         _xdr_result = (xdrproc_t) xdr_readdir3res;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_readdir_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_readdir_3_svc;
         break;
 
     case NFSPROC3_READDIRPLUS:
         _xdr_argument = (xdrproc_t) xdr_readdirplus3args;
         _xdr_result = (xdrproc_t) xdr_readdirplus3res;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_readdirplus_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_readdirplus_3_svc;
         break;
 
     case NFSPROC3_FSSTAT:
         _xdr_argument = (xdrproc_t) xdr_nfs_fh3;
         _xdr_result = (xdrproc_t) xdr_fsstat3res;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_fsstat_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_fsstat_3_svc;
         break;
 
     case NFSPROC3_FSINFO:
         _xdr_argument = (xdrproc_t) xdr_nfs_fh3;
         _xdr_result = (xdrproc_t) xdr_fsinfo3res;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_fsinfo_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_fsinfo_3_svc;
         break;
 
     case NFSPROC3_PATHCONF:
         _xdr_argument = (xdrproc_t) xdr_nfs_fh3;
         _xdr_result = (xdrproc_t) xdr_pathconf3res;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_pathconf_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_pathconf_3_svc;
         break;
 
     case NFSPROC3_COMMIT:
         _xdr_argument = (xdrproc_t) xdr_commit3args;
         _xdr_result = (xdrproc_t) xdr_commit3res;
-        local = (char *(*)(char *, struct svc_req *)) nfsproc3_commit_3_svc;
+        local = (char *(*)(char *, RPCRequest *)) nfsproc3_commit_3_svc;
         break;
 
     default:
@@ -340,41 +447,11 @@ nfs_program_3(RPCRequest *req)
     }
 
     /* Perform the call. */
-    result = (*local)((char *)req->args, NULL);
-
-    /* Encode result and send reply. */
-    static char reply_buf[MAX_RPC_MSGSIZE];
-    XDR xdr_out;
-    xdrmem_create(&xdr_out, reply_buf, MAX_RPC_MSGSIZE, XDR_ENCODE);
-    if (result != NULL && !_xdr_result(&xdr_out, result)) {
-        async_rpc_send_failure(req, SYSTEM_ERR);
-    }
-
-    struct rpc_reply header;
-    header.xid = htonl(xid);
-    header.type = htonl(1);     /* REPLY */
-    header.stat = htonl(MSG_ACCEPTED);
-    header.verf_flavor = 0;
-    header.verf_len = 0;
-    header.accept_stat = 0;
-
-    gsize msg_size = xdr_out.x_ops->x_getpostn(&xdr_out);
-    printf("Have an RPC reply of size %zd bytes\n", msg_size);
-    uint32_t fragment = htonl((msg_size + sizeof(header)) | 0x80000000);
-    async_rpc_write(connection, (const char *)&fragment, sizeof(fragment));
-    async_rpc_write(connection, (const char *)&header, sizeof(header));
-    async_rpc_write(connection, reply_buf, msg_size);
-    g_io_channel_flush(connection->channel, NULL);
-
-    /* Clean up. */
-    xdr_in.x_op = XDR_FREE;
-    if (!_xdr_argument(&xdr_in, (caddr_t)req->args)) {
-        fprintf (stderr, "%s", "unable to free arguments");
-        exit (1);
-    }
-    g_free(req->args);
+    req->xdr_result = _xdr_result;
+    result = (*local)((char *)req->args, req);
 
     bluesky_flushd_invoke(fs);
+    bluesky_debug_dump(fs);
 
     return;
 }
@@ -463,8 +540,6 @@ static gboolean async_rpc_dispatch(RPCConnection *rpc)
     if (buf - msg->str > msg->len)
         return FALSE;
 
-    printf("Dispatching RPC procedure %d...\n", proc);
-
     req->raw_args = msg;
     req->raw_args_header_bytes = buf - msg->str;
     req->req_proc = ntohl(header->proc);
@@ -560,7 +635,6 @@ static gboolean async_rpc_do_read(GIOChannel *channel,
             rpc->frag_len = ntohl(rpc->frag_len);
             g_string_set_size(rpc->msgbuf, rpc->msgbuf->len - 4);
             rpc->frag_hdr_bytes = 0;
-            g_print("RPC fragment header: %08x\n", rpc->frag_len);
         }
     } else {
         /* We were reading in the fragment body. */
@@ -569,7 +643,6 @@ static gboolean async_rpc_do_read(GIOChannel *channel,
         if (rpc->frag_len = 0x80000000) {
             /* We have a complete message since this was the last fragment and
              * there are no more bytes in it.  Dispatch the message. */
-            g_print("Complete RPC message: %zd bytes\n", rpc->msgbuf->len);
             if (!async_rpc_dispatch(rpc)) {
                 fprintf(stderr, "Invalid RPC message, closing channel\n");
                 g_io_channel_shutdown(rpc->channel, TRUE, NULL);