fql/sys.cc at master · dccmx/fql · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
#include <stdio.h>
#include <stdlib.h>
#include <limits.h>
#include <dirent.h>
#include <sys/stat.h>
#include <pwd.h>
#include <grp.h>
#include <unistd.h>
#include <sys/types.h>
#include <fcntl.h>
#include <libgen.h>
#include <string.h>

#include "sys.h"

static void AddFile(Table *tb, const char *path, const char *name) {
  struct stat st;

  static char filename[2048];

  sprintf(filename, "%s/%s", path, name);

  if(stat(filename, &st) == -1) return;

  Row row;

  for (vector<string>::const_iterator ite = FullHeader().begin(); ite != FullHeader().end(); ++ite) {
    if (*ite == "name") {
      row.push_back(new String(name));
    } else if (*ite == "path") {
      row.push_back(new String(path));
    } else if (*ite == "perms") {
      row.push_back(new Permission(st.st_mode));
    } else if (*ite == "tperms") {
      row.push_back(new TypePermission(st.st_mode));
    } else if (*ite == "nlink") {
      row.push_back(new Int32(st.st_nlink));
    } else if (*ite == "size") {
      row.push_back(new FileSize(st.st_size));
    } else if (*ite == "inode") {
      row.push_back(new Int32(st.st_ino));
    } else if (*ite == "uid") {
      row.push_back(new Int32(st.st_uid));
    } else if (*ite == "gid") {
      row.push_back(new Int32(st.st_gid));
    } else if (*ite == "uname") {
      struct passwd *pw = getpwuid(st.st_uid);
      if (pw) {
          row.push_back(new String(pw->pw_name));
      } else {
          row.push_back(new String(""));
      }
    } else if (*ite == "gname") {
      struct group *grp = getgrgid(st.st_gid);
      if (grp) {
          row.push_back(new String(grp->gr_name));
      } else {
          row.push_back(new String(""));
      }
    } else if (*ite == "atime") {
      row.push_back(new Time(st.st_atime));
    } else if (*ite == "mtime" || *ite == "time") {
      row.push_back(new Time(st.st_mtime));
    } else if (*ite == "ctime") {
      row.push_back(new Time(st.st_ctime));
    } else {
      fprintf(stderr, "unknow attribute: %s\n", ite->c_str());
    }
  }

  tb->Append(row);
}

void ListDir(Table *tb, const string& dir, bool recursive) {
  DIR *dp;
  struct dirent *ent;

  if (!(dp = opendir(dir.c_str()))) return;

  while((ent = readdir(dp)) != NULL) {
    if (strcmp(ent->d_name, ".") == 0) continue;
    if (strcmp(ent->d_name, "..") == 0) continue;
    if (recursive && (ent->d_type & DT_DIR)) {
      string subdir = dir;
      subdir += "/";
      subdir += ent->d_name;
      ListDir(tb, subdir, recursive);
    }
    AddFile(tb, dir.c_str(), ent->d_name);
  }

  closedir(dp);
}