terom@28: 
terom@28: #include <stdlib.h>
terom@28: #include <assert.h>
terom@28: 
terom@29: #include "dbfs.h"
terom@29: #include "op_base.h"
terom@28: #include "../dirbuf.h"
terom@29: #include "../lib/log.h"
terom@28: 
terom@28: /*
terom@28:  * Directory related functionality like opendir, readdir, releasedir
terom@28:  */
terom@28: struct dbfs_dirop {
terom@29:     struct dbfs_op base;
terom@28: 
terom@29:     // parent dir inodes
terom@29:     uint32_t parent;
terom@28:     
terom@28:     // for readdir
terom@28:     struct dirbuf dirbuf;
terom@28: };
terom@28: 
terom@28: /*
terom@29:  * Release the dirbuf.
terom@28:  */
terom@29: static void _dbfs_dirop_free (struct dbfs_op *op_base) {
terom@29:     struct dbfs_dirop *dirop = (struct dbfs_dirop *) op_base;
terom@29: 
terom@28:     // just release the dirbuf
terom@28:     dirbuf_release(&dirop->dirbuf);
terom@28: }
terom@28: 
terom@28: /*
terom@28:  * Handle the results for the initial attribute lookup for the dir itself during opendir ops.
terom@28:  */
terom@29: static void dbfs_opendir_res (const struct evsql_result_info *res, void *arg) {
terom@28:     struct dbfs_dirop *dirop = arg;
terom@28:     int err;
terom@28:     
terom@29:     assert(dirop->base.req);
terom@29:     assert(dirop->base.trans); // query callbacks don't get called if the trans fails
terom@29:     assert(!dirop->base.open);
terom@28:    
terom@28:     // check the results
terom@28:     if ((err = _dbfs_check_res(res, 1, 2)))
terom@28:         SERROR(err = (err ==  1 ? ENOENT : EIO));
terom@28: 
terom@28:     const char *type;
terom@28: 
terom@28:     // extract the data
terom@28:     if (0
terom@28:         ||  evsql_result_uint32(res, 0, 0, &dirop->parent,  1 ) // file_tree.parent
terom@28:         ||  evsql_result_string(res, 0, 1, &type,           0 ) // inodes.type
terom@28:     )
terom@28:         SERROR(err = EIO);
terom@28: 
terom@28:     // is it a dir?
terom@28:     if (_dbfs_mode(type) != S_IFDIR)
terom@28:         EERROR(err = ENOTDIR, "wrong type: %s", type);
terom@28:     
terom@36:     INFO("\t[dbfs.opendir %p:%p] -> ino=%lu, parent=%lu, type=%s", dirop, dirop->base.req, (unsigned long int) dirop->base.ino, (unsigned long int) dirop->parent, type);
terom@28:     
terom@29:     // open_fn done, do the open_reply
terom@29:     if ((err = dbfs_op_open_reply(&dirop->base)))
terom@29:         goto error;
terom@28: 
terom@28:     // success, fallthrough for evsql_result_free
terom@28:     err = 0;
terom@28: 
terom@28: error:
terom@28:     if (err)
terom@28:         // fail it
terom@29:         dbfs_op_fail(&dirop->base, err);
terom@28:     
terom@28:     // free
terom@28:     evsql_result_free(res);
terom@28: }
terom@28: 
terom@28: /*
terom@28:  * The opendir transaction is ready for use. Query for the given dir's info
terom@28:  */
terom@29: static void dbfs_dirop_open (struct dbfs_op *op_base) {
terom@29:     struct dbfs_dirop *dirop = (struct dbfs_dirop *) op_base;
terom@29:     struct dbfs *ctx = fuse_req_userdata(dirop->base.req);
terom@28:     int err;
terom@28:     
terom@29:     assert(dirop->base.trans); 
terom@29:     assert(dirop->base.req);
terom@29:     assert(!dirop->base.open);
terom@28: 
terom@36:     INFO("\t[dbfs.opendir %p:%p] -> trans=%p", dirop, dirop->base.req, dirop->base.trans);
terom@28:     
terom@28:     // first fetch info about the dir itself
terom@28:     const char *sql =
terom@28:         "SELECT"
terom@28:         " file_tree.parent, inodes.type"
terom@33:         " FROM file_tree LEFT OUTER JOIN inodes ON (file_tree.ino = inodes.ino)"
terom@33:         " WHERE file_tree.ino = $1::int4";
terom@28: 
terom@28:     static struct evsql_query_params params = EVSQL_PARAMS(EVSQL_FMT_BINARY) {
terom@28:         EVSQL_PARAM ( UINT32 ),
terom@28: 
terom@28:         EVSQL_PARAMS_END
terom@28:     };
terom@28: 
terom@28:     // build params
terom@28:     if (0
terom@29:         ||  evsql_param_uint32(&params, 0, dirop->base.ino)
terom@28:     )
terom@28:         SERROR(err = EIO);
terom@28:         
terom@28:     // query
terom@29:     if (evsql_query_params(ctx->db, dirop->base.trans, sql, &params, dbfs_opendir_res, dirop) == NULL)
terom@28:         SERROR(err = EIO);
terom@28: 
terom@28:     // ok, wait for the info results
terom@28:     return;
terom@28: 
terom@28: error:
terom@28:     // fail it
terom@29:     dbfs_op_fail(&dirop->base, err);
terom@28: }
terom@28: 
terom@28: /*
terom@29:  * Handle opendir(), this means starting a new op.
terom@28:  */
terom@28: void dbfs_opendir (struct fuse_req *req, fuse_ino_t ino, struct fuse_file_info *fi) {
terom@28:     struct dbfs *ctx = fuse_req_userdata(req);
terom@28:     struct dbfs_dirop *dirop = NULL;
terom@28:     int err;
terom@28:     
terom@28:     // allocate it
terom@28:     if ((dirop = calloc(1, sizeof(*dirop))) == NULL && (err = EIO))
terom@28:         ERROR("calloc");
terom@28: 
terom@29:     // do the op_open
terom@29:     if ((err = dbfs_op_open(ctx, &dirop->base, req, ino, fi, _dbfs_dirop_free, dbfs_dirop_open)))
terom@29:         ERROR("dbfs_op_open");
terom@29: 
terom@28:     INFO("[dbfs.opendir %p:%p] ino=%lu, fi=%p", dirop, req, ino, fi);
terom@28:     
terom@28:     // wait
terom@28:     return;
terom@28: 
terom@28: error:
terom@28:     if (dirop) {
terom@28:         // we can fail normally
terom@29:         dbfs_op_fail(&dirop->base, err);
terom@28: 
terom@28:     } else {
terom@28:         // must error out manually as we couldn't alloc the context
terom@28:         if ((err = fuse_reply_err(req, err)))
terom@28:             EWARNING(err, "fuse_reply_err");
terom@28:     }
terom@28: }
terom@28: 
terom@28: /*
terom@28:  * Got the list of files for our readdir() request.
terom@28:  *
terom@28:  * Fill up the dirbuf, and then send the reply.
terom@28:  *
terom@28:  */
terom@29: static void dbfs_readdir_res (const struct evsql_result_info *res, void *arg) {
terom@28:     struct dbfs_dirop *dirop = arg;
terom@28:     int err;
terom@28:     size_t row;
terom@28:     
terom@29:     assert(dirop->base.req);
terom@29:     assert(dirop->base.trans); // query callbacks don't get called if the trans fails
terom@29:     assert(dirop->base.open);
terom@28:     
terom@28:     // check the results
terom@28:     if ((err = _dbfs_check_res(res, 0, 4)) < 0)
terom@28:         SERROR(err = EIO);
terom@28:         
terom@36:     INFO("\t[dbfs.readdir %p:%p] -> files: res_rows=%zu", dirop, dirop->base.req, evsql_result_rows(res));
terom@28:         
terom@28:     // iterate over the rows
terom@28:     for (row = 0; row < evsql_result_rows(res); row++) {
terom@28:         uint32_t off, ino;
terom@28:         const char *name, *type;
terom@28: 
terom@28:         // extract the data
terom@28:         if (0
terom@28:             ||  evsql_result_uint32(res, row, 0, &off,          0 ) // file_tree.offset
terom@28:             ||  evsql_result_string(res, row, 1, &name,         0 ) // file_tree.name
terom@28:             ||  evsql_result_uint32(res, row, 2, &ino,          0 ) // inodes.ino
terom@28:             ||  evsql_result_string(res, row, 3, &type,         0 ) // inodes.type
terom@28:         )
terom@28:             SERROR(err = EIO);
terom@28:         
terom@28:         INFO("\t%zu: off=%lu+2, name=%s, ino=%lu, type=%s", row, (long unsigned int) off, name, (long unsigned int) ino, type);
terom@28: 
terom@28:         // add to the dirbuf
terom@28:         // offsets are just offset + 2
terom@29:         if ((err = dirbuf_add(dirop->base.req, &dirop->dirbuf, off + 2, off + 3, name, ino, _dbfs_mode(type))) < 0 && (err = EIO))
terom@28:             ERROR("failed to add dirent for inode=%lu", (long unsigned int) ino);
terom@28:         
terom@28:         // stop if it's full
terom@28:         if (err > 0)
terom@28:             break;
terom@28:     }
terom@28: 
terom@28:     // send it
terom@29:     if ((err = dirbuf_done(dirop->base.req, &dirop->dirbuf)))
terom@28:         EERROR(err, "failed to send buf");
terom@29:     
terom@29:     // handled the req
terom@29:     if ((err = dbfs_op_req_done(&dirop->base)))
terom@29:         goto error;
terom@28: 
terom@28:     // good, fallthrough
terom@28:     err = 0;
terom@28: 
terom@28: error:
terom@28:     if (err)
terom@29:         dbfs_op_fail(&dirop->base, err);
terom@28: 
terom@28:     // free
terom@28:     evsql_result_free(res);
terom@28: }
terom@28: 
terom@28: /*
terom@28:  * Handle a readdir request. This will execute a SQL query inside the transaction to get the files at the given offset,
terom@29:  * and dbfs_readdir_res will handle the results.
terom@28:  *
terom@28:  * If trans failed earlier, detect that and return an error.
terom@28:  */
terom@28: void dbfs_readdir (struct fuse_req *req, fuse_ino_t ino, size_t size, off_t off, struct fuse_file_info *fi) {
terom@28:     struct dbfs *ctx = fuse_req_userdata(req);
terom@30:     struct dbfs_dirop *dirop;
terom@28:     int err;
terom@29: 
terom@29:     // get the op
terom@29:     if ((dirop = (struct dbfs_dirop *) dbfs_op_req(req, ino, fi)) == NULL)
terom@30:         return;
terom@28:     
terom@29:     INFO("[dbfs.readdir %p:%p] ino=%lu, size=%zu, off=%zu, fi=%p : trans=%p", dirop, req, ino, size, off, fi, dirop->base.trans);
terom@28: 
terom@28:     // create the dirbuf
terom@28:     if (dirbuf_init(&dirop->dirbuf, size, off))
terom@28:         SERROR(err = EIO);
terom@28: 
terom@28:     // add . and ..
terom@28:     // we set the next offset to 2, because all dirent offsets will be larger than that
terom@28:     // assume that these two should *always* fit
terom@28:     if ((err = (0
terom@29:         ||  dirbuf_add(req, &dirop->dirbuf, 0, 1, ".",   dirop->base.ino,    S_IFDIR )
terom@28:         ||  dirbuf_add(req, &dirop->dirbuf, 1, 2, "..",  
terom@29:                         dirop->parent ? dirop->parent : dirop->base.ino,     S_IFDIR )
terom@28:     )) && (err = EIO))
terom@28:         ERROR("failed to add . and .. dirents");
terom@28: 
terom@28:     // select all relevant file entries
terom@28:     const char *sql = 
terom@28:         "SELECT"
terom@28:         " file_tree.\"offset\", file_tree.name, inodes.ino, inodes.type"
terom@33:         " FROM file_tree LEFT OUTER JOIN inodes ON (file_tree.ino = inodes.ino)"
terom@28:         " WHERE file_tree.parent = $1::int4 AND file_tree.\"offset\" >= $2::int4"
terom@34:         " ORDER BY file_tree.\"offset\""
terom@28:         " LIMIT $3::int4";
terom@28: 
terom@28:     static struct evsql_query_params params = EVSQL_PARAMS(EVSQL_FMT_BINARY) {
terom@28:         EVSQL_PARAM ( UINT32 ),
terom@28:         EVSQL_PARAM ( UINT32 ),
terom@28:         EVSQL_PARAM ( UINT32 ),
terom@28: 
terom@28:         EVSQL_PARAMS_END
terom@28:     };
terom@28: 
terom@28:     // adjust offset to take . and .. into account
terom@28:     if (off > 2)
terom@28:         off -= 2;
terom@28:     
terom@28:     // build params
terom@28:     if (0
terom@29:         ||  evsql_param_uint32(&params, 0, ino)
terom@28:         ||  evsql_param_uint32(&params, 1, off)
terom@28:         ||  evsql_param_uint32(&params, 2, dirbuf_estimate(&dirop->dirbuf, 0))
terom@28:     )
terom@28:         SERROR(err = EIO);
terom@28: 
terom@28:     // query
terom@29:     if (evsql_query_params(ctx->db, dirop->base.trans, sql, &params, dbfs_readdir_res, dirop) == NULL)
terom@28:         SERROR(err = EIO);
terom@28: 
terom@28:     // good, wait
terom@28:     return;
terom@28: 
terom@28: error:
terom@29:     dbfs_op_fail(&dirop->base, err);
terom@28: }
terom@28: 
terom@28: /*
terom@28:  * "For every [succesfull] opendir call there will be exactly one releasedir call."
terom@28:  *
terom@28:  * The dirop may be in a failed state.
terom@28:  */
terom@28: void dbfs_releasedir (struct fuse_req *req, fuse_ino_t ino, struct fuse_file_info *fi) {
terom@29:     // just passthrough to dbfs_op
terom@29:     dbfs_op_release(req, ino, fi);
terom@28: }
terom@28: