1

     2

#include <stdlib.h>

     3

     4

#include "dirbuf.h"

     5

#include "lib/log.h"

     6

#include "lib/math.h"

     7

     8

int dirbuf_init (struct dirbuf *buf, size_t req_size, off_t req_off) {

     9

    buf->buf = NULL;

    10

    buf->len = req_size;

    11

    buf->off = 0;

    12

    buf->req_off = req_off;

    13

    14

    DEBUG("\tdirbuf.init: req_size=%zu", req_size);

    15

    16

    // allocate the mem

    17

    if ((buf->buf = malloc(buf->len)) == NULL)

    18

        ERROR("malloc");

    19

    20

    // ok

    21

    return 0;

    22

    23

error:

    24

    return -1;

    25

}

    26

    27

size_t dirbuf_estimate (struct dirbuf *buf, size_t min_namelen) {

    28

    char namebuf[DIRBUF_NAME_MAX];

    29

    int i;

    30

    31

    // build a dummy string of the right length

    32

    for (i = 0; i < min_namelen && i < DIRBUF_NAME_MAX - 1; i++)

    33

        namebuf[i] = 'x';

    34

    35

    namebuf[i] = '\0';

    36

    37

    return buf->len / (fuse_add_direntry(NULL, NULL, 0, namebuf, NULL, 0));

    38

}

    39

    40

int dirbuf_add (fuse_req_t req, struct dirbuf *buf, off_t ent_off, off_t next_off, const char *ent_name, fuse_ino_t ent_ino, mode_t ent_mode) {

    41

    struct stat stbuf;

    42

    size_t ent_size;

    43

    44

    DEBUG("\tdirbuf.add: req_off=%zu, buf->len=%zu, buf->off=%zu, ent_off=%zu, next_off=%zu, ent_name=`%s`, ent_ino=%lu, ent_mode=%07o",

    45

        buf->req_off, buf->len, buf->off, ent_off, next_off, ent_name, ent_ino, ent_mode);

    46

    47

    // skip entries as needed

    48

    if (ent_off < buf->req_off) 

    49

        return 0;

    50

    51

    // set ino

    52

    stbuf.st_ino = ent_ino;

    53

    stbuf.st_mode = ent_mode;

    54

    55

    // try and add the dirent, and see if it fits

    56

    if ((ent_size = fuse_add_direntry(req, buf->buf + buf->off, buf->len - buf->off, ent_name, &stbuf, next_off)) > (buf->len - buf->off)) {

    57

        // 'tis full

    58

        return 1;

    59

    60

    } else {

    61

        // it fit

    62

        buf->off += ent_size;

    63

    }

    64

    65

    // success

    66

    return 0;

    67

}

    68

    69

int dirbuf_done (fuse_req_t req, struct dirbuf *buf) {

    70

    int err;

    71

    72

    // send the reply, return the error later

    73

    err = -fuse_reply_buf(req, buf->buf, buf->off);

    74

    75

    DEBUG("\tdirbuf.done: size=%zu/%zu, err=%d", buf->off, buf->len, err);

    76

    77

    // free the dirbuf

    78

    dirbuf_release(buf);

    79

    80

    // return the error code

    81

    return err;

    82

}

    83

    84

void dirbuf_release (struct dirbuf *buf) {

    85

    free(buf->buf); buf->buf = NULL;

    86

}

    87