quick-read: implement qr_ftruncate.

Signed-off-by: Anand V. Avati <avati@dev.gluster.com>

1 files changed, 142 insertions, 0 deletions

diff --git a/xlators/performance/quick-read/src/quick-read.c b/xlators/performance/quick-read/src/quick-read.c
index 0e3095e5f..632687c0a 100644
--- a/xlators/performance/quick-read/src/quick-read.c
+++ b/xlators/performance/quick-read/src/quick-read.c
@@ -1838,6 +1838,147 @@ out:
 }
 
 
+int32_t
+qr_ftruncate_cbk (call_frame_t *frame, void *cookie, xlator_t *this,
+                  int32_t op_ret, int32_t op_errno, struct stat *buf)
+{
+        int32_t     ret = 0;
+        uint64_t    value = 0; 
+        qr_file_t  *qr_file = NULL;
+        qr_local_t *local = NULL;
+
+        if (op_ret == -1) {
+                goto out;
+        }
+
+        local = frame->local;
+        if ((local == NULL) || (local->fd == NULL)
+            || (local->fd->inode == NULL)) {
+                op_ret = -1;
+                op_errno = EINVAL;
+                goto out;
+        }
+                
+        ret = inode_ctx_get (local->fd->inode, this, &value);
+        if (ret == 0) {
+                qr_file = (qr_file_t *)(long) value;
+                
+                if (qr_file) {
+                        LOCK (&qr_file->lock);
+                        {
+                                if (qr_file->stbuf.st_size != buf->st_size) {
+                                        dict_unref (qr_file->xattr);
+                                        qr_file->xattr = NULL;
+                                }
+                        }
+                        UNLOCK (&qr_file->lock);
+                }
+        }
+
+out:
+        STACK_UNWIND (frame, op_ret, op_errno, buf);
+        return 0;
+}
+
+
+int32_t
+qr_ftruncate_helper (call_frame_t *frame, xlator_t *this, fd_t *fd,
+                     off_t offset)
+{
+        STACK_WIND (frame, qr_ftruncate_cbk, FIRST_CHILD(this),
+                    FIRST_CHILD(this)->fops->ftruncate, fd, offset);
+        return 0;
+}
+
+
+int32_t
+qr_ftruncate (call_frame_t *frame, xlator_t *this, fd_t *fd, off_t offset)
+{
+        int          flags = 0;
+        uint64_t     value = 0;
+        call_stub_t *stub = NULL;  
+        char        *path = NULL;
+        loc_t        loc = {0, };
+        qr_local_t  *local = NULL;
+        qr_fd_ctx_t *qr_fd_ctx = NULL;
+        int32_t      ret = -1, op_ret = -1, op_errno = -1;
+        char         need_open = 0, can_wind = 0, need_unwind = 0;
+
+        ret = fd_ctx_get (fd, this, &value);
+        if (ret == 0) {
+                qr_fd_ctx = (qr_fd_ctx_t *)(long)value;
+        }
+        
+        local = CALLOC (1, sizeof (*local));
+        if (local == NULL) {
+                op_ret = -1;
+                op_errno = ENOMEM;
+                need_unwind = 1;
+                goto out;
+        }
+
+        local->fd = fd;
+        frame->local = local;
+ 
+        if (qr_fd_ctx) {
+                LOCK (&qr_fd_ctx->lock);
+                {
+                        path = qr_fd_ctx->path;
+                        flags = qr_fd_ctx->flags;
+
+                        if (!(qr_fd_ctx->opened
+                              || qr_fd_ctx->open_in_transit)) {
+                                need_open = 1;
+                                qr_fd_ctx->open_in_transit = 1;
+                        }
+
+                        if (qr_fd_ctx->opened) {
+                                can_wind = 1;
+                        } else {
+                                stub = fop_ftruncate_stub (frame,
+                                                           qr_ftruncate_helper,
+                                                           fd, offset);
+                                if (stub == NULL) {
+                                        op_ret = -1;
+                                        op_errno = ENOMEM;
+                                        need_unwind = 1;
+                                        qr_fd_ctx->open_in_transit = 0;
+                                        goto unlock;
+                                }
+
+                                list_add_tail (&stub->list,
+                                               &qr_fd_ctx->waiting_ops);
+                        }
+                }
+        unlock:
+                UNLOCK (&qr_fd_ctx->lock);
+        } else {
+                can_wind = 1;
+        }
+
+out:
+        if (need_unwind) {
+                STACK_UNWIND (frame, op_ret, op_errno, NULL);
+        } else if (can_wind) {
+                STACK_WIND (frame, qr_ftruncate_cbk, FIRST_CHILD(this),
+                            FIRST_CHILD(this)->fops->ftruncate, fd, offset);
+        } else if (need_open) {
+                op_ret = qr_loc_fill (&loc, fd->inode, path);
+                if (op_ret == -1) {
+                        qr_resume_pending_ops (qr_fd_ctx);
+                        goto out;
+                }
+
+                STACK_WIND (frame, qr_open_cbk, FIRST_CHILD(this),
+                            FIRST_CHILD(this)->fops->open, &loc, flags, fd);
+
+                qr_loc_wipe (&loc);
+        }
+        
+        return 0;
+}
+
+
 int32_t 
 init (xlator_t *this)
 {
@@ -1923,6 +2064,7 @@ struct xlator_fops fops = {
         .fentrylk    = qr_fentrylk,
         .finodelk    = qr_finodelk,
         .fsync       = qr_fsync,
+        .ftruncate   = qr_ftruncate,
 };