Mercurial > public > mercurial-scm > hg-stable
annotate mercurial/osutil.c @ 5424:005638db9d0c
osutil: fold stat paths together
- simplify st/py_st logic
- use stp to point to stat buffer
- combine stat paths
| author | Matt Mackall <mpm@selenic.com> |
|---|---|
| date | Mon, 08 Oct 2007 18:47:15 -0500 |
| parents | e5f238a8b0d2 |
| children | 830f6e280c90 |
| rev | line source |
|---|---|
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
1 /* |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
2 osutil.c - native operating system services |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
3 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
4 Copyright 2007 Matt Mackall and others |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
5 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
6 This software may be used and distributed according to the terms of |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
7 the GNU General Public License, incorporated herein by reference. |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
8 */ |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
9 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
10 #define _ATFILE_SOURCE |
|
5397
11caa374f497
osutil.c: include Python.h before the other headers
Alexis S. L. Carvalho <alexis@cecm.usp.br>
parents:
5396
diff
changeset
|
11 #include <Python.h> |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
12 #include <alloca.h> |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
13 #include <dirent.h> |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
14 #include <fcntl.h> |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
15 #include <string.h> |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
16 #include <sys/stat.h> |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
17 #include <sys/types.h> |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
18 #include <unistd.h> |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
19 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
20 struct listdir_stat { |
| 5421 | 21 PyObject_HEAD |
| 22 struct stat st; | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
23 }; |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
24 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
25 #define listdir_slot(name) \ |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
26 static PyObject *listdir_stat_##name(PyObject *self, void *x) \ |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
27 { \ |
| 5421 | 28 return PyInt_FromLong(((struct listdir_stat *)self)->st.name); \ |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
29 } |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
30 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
31 listdir_slot(st_dev); |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
32 listdir_slot(st_mode); |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
33 listdir_slot(st_nlink); |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
34 listdir_slot(st_size); |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
35 listdir_slot(st_mtime); |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
36 listdir_slot(st_ctime); |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
37 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
38 static struct PyGetSetDef listdir_stat_getsets[] = { |
| 5421 | 39 {"st_dev", listdir_stat_st_dev, 0, 0, 0}, |
| 40 {"st_mode", listdir_stat_st_mode, 0, 0, 0}, | |
| 41 {"st_nlink", listdir_stat_st_nlink, 0, 0, 0}, | |
| 42 {"st_size", listdir_stat_st_size, 0, 0, 0}, | |
| 43 {"st_mtime", listdir_stat_st_mtime, 0, 0, 0}, | |
| 44 {"st_ctime", listdir_stat_st_ctime, 0, 0, 0}, | |
| 45 {0, 0, 0, 0, 0} | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
46 }; |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
47 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
48 static PyObject *listdir_stat_new(PyTypeObject *t, PyObject *a, PyObject *k) |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
49 { |
| 5421 | 50 return t->tp_alloc(t, 0); |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
51 } |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
52 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
53 static void listdir_stat_dealloc(PyObject *o) |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
54 { |
| 5421 | 55 o->ob_type->tp_free(o); |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
56 } |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
57 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
58 static PyTypeObject listdir_stat_type = { |
| 5421 | 59 PyObject_HEAD_INIT(NULL) |
| 60 0, /*ob_size*/ | |
| 61 "osutil.stat", /*tp_name*/ | |
| 62 sizeof(struct listdir_stat), /*tp_basicsize*/ | |
| 63 0, /*tp_itemsize*/ | |
| 64 (destructor)listdir_stat_dealloc, /*tp_dealloc*/ | |
| 65 0, /*tp_print*/ | |
| 66 0, /*tp_getattr*/ | |
| 67 0, /*tp_setattr*/ | |
| 68 0, /*tp_compare*/ | |
| 69 0, /*tp_repr*/ | |
| 70 0, /*tp_as_number*/ | |
| 71 0, /*tp_as_sequence*/ | |
| 72 0, /*tp_as_mapping*/ | |
| 73 0, /*tp_hash */ | |
| 74 0, /*tp_call*/ | |
| 75 0, /*tp_str*/ | |
| 76 0, /*tp_getattro*/ | |
| 77 0, /*tp_setattro*/ | |
| 78 0, /*tp_as_buffer*/ | |
| 79 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /*tp_flags*/ | |
| 80 "stat objects", /* tp_doc */ | |
| 81 0, /* tp_traverse */ | |
| 82 0, /* tp_clear */ | |
| 83 0, /* tp_richcompare */ | |
| 84 0, /* tp_weaklistoffset */ | |
| 85 0, /* tp_iter */ | |
| 86 0, /* tp_iternext */ | |
| 87 0, /* tp_methods */ | |
| 88 0, /* tp_members */ | |
| 89 listdir_stat_getsets, /* tp_getset */ | |
| 90 0, /* tp_base */ | |
| 91 0, /* tp_dict */ | |
| 92 0, /* tp_descr_get */ | |
| 93 0, /* tp_descr_set */ | |
| 94 0, /* tp_dictoffset */ | |
| 95 0, /* tp_init */ | |
| 96 0, /* tp_alloc */ | |
| 97 listdir_stat_new, /* tp_new */ | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
98 }; |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
99 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
100 static inline int mode_to_kind(int mode) |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
101 { |
| 5421 | 102 if (S_ISREG(mode)) return S_IFREG; |
| 103 if (S_ISDIR(mode)) return S_IFDIR; | |
| 104 if (S_ISLNK(mode)) return S_IFLNK; | |
| 105 if (S_ISBLK(mode)) return S_IFBLK; | |
| 106 if (S_ISCHR(mode)) return S_IFCHR; | |
| 107 if (S_ISFIFO(mode)) return S_IFIFO; | |
| 108 if (S_ISSOCK(mode)) return S_IFSOCK; | |
| 109 return mode; | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
110 } |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
111 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
112 static PyObject *listdir(PyObject *self, PyObject *args, PyObject *kwargs) |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
113 { |
|
5416
ca890c0c3f1f
osutil.c: style fix - delete trailing end-of-line spaces
Giorgos Keramidas <keramida@ceid.upatras.gr>
parents:
5398
diff
changeset
|
114 static char *kwlist[] = { "path", "stat", NULL }; |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
115 PyObject *statobj = NULL; |
| 5421 | 116 DIR *dir = NULL; |
| 117 struct dirent *ent; | |
| 118 PyObject *list = NULL; | |
| 119 PyObject *ctor_args = NULL; | |
| 120 int all_kinds = 1; | |
|
5422
a3ba7ef98c94
osutil: eliminate alloca call
Matt Mackall <mpm@selenic.com>
parents:
5421
diff
changeset
|
121 char full_path[PATH_MAX + 10]; |
| 5421 | 122 int path_len; |
| 123 int do_stat; | |
| 124 char *path; | |
| 125 int ret; | |
| 126 ssize_t size; | |
| 127 ssize_t i; | |
| 128 int dfd; | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
129 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
130 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "s#|O:listdir", kwlist, |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
131 &path, &path_len, &statobj)) |
| 5421 | 132 goto bail; |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
133 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
134 do_stat = statobj && PyObject_IsTrue(statobj); |
|
5416
ca890c0c3f1f
osutil.c: style fix - delete trailing end-of-line spaces
Giorgos Keramidas <keramida@ceid.upatras.gr>
parents:
5398
diff
changeset
|
135 |
| 5421 | 136 dir = opendir(path); |
| 137 if (!dir) { | |
| 138 list = PyErr_SetFromErrnoWithFilename(PyExc_OSError, path); | |
| 139 goto bail; | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
140 } |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
141 |
| 5421 | 142 list = PyList_New(0); |
| 143 if (!list) | |
| 144 goto bail; | |
|
5416
ca890c0c3f1f
osutil.c: style fix - delete trailing end-of-line spaces
Giorgos Keramidas <keramida@ceid.upatras.gr>
parents:
5398
diff
changeset
|
145 |
|
5422
a3ba7ef98c94
osutil: eliminate alloca call
Matt Mackall <mpm@selenic.com>
parents:
5421
diff
changeset
|
146 strncpy(full_path, path, PATH_MAX); |
| 5421 | 147 full_path[path_len] = '/'; |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
148 |
| 5421 | 149 for (ent = readdir(dir); ent; ent = readdir(dir)) { |
| 150 PyObject *name = NULL; | |
| 151 PyObject *py_kind = NULL; | |
| 152 PyObject *val = NULL; | |
| 153 unsigned char d_type; | |
| 154 int kind = -1; | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
155 |
| 5421 | 156 if (!strcmp(ent->d_name, ".") || !strcmp(ent->d_name, "..")) |
| 157 continue; | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
158 |
| 5421 | 159 #ifdef DT_REG |
| 160 if (do_stat) | |
| 161 d_type = 0; | |
| 162 else | |
| 163 d_type = ent->d_type; | |
| 164 #else | |
| 165 d_type = 0; | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
166 #endif |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
167 |
| 5421 | 168 switch (d_type) { |
| 169 #ifdef DT_REG | |
| 170 case DT_REG: kind = S_IFREG; break; | |
| 171 case DT_DIR: kind = S_IFDIR; break; | |
| 172 case DT_LNK: kind = S_IFLNK; break; | |
| 173 case DT_BLK: kind = S_IFBLK; break; | |
| 174 case DT_CHR: kind = S_IFCHR; break; | |
| 175 case DT_FIFO: kind = S_IFIFO; break; | |
| 176 case DT_SOCK: kind = S_IFSOCK; break; | |
| 177 #endif | |
| 178 default: | |
| 179 if (all_kinds) | |
| 180 all_kinds = 0; | |
| 181 break; | |
| 182 } | |
|
5416
ca890c0c3f1f
osutil.c: style fix - delete trailing end-of-line spaces
Giorgos Keramidas <keramida@ceid.upatras.gr>
parents:
5398
diff
changeset
|
183 |
| 5421 | 184 name = PyString_FromString(ent->d_name); |
| 185 if (kind != -1) | |
| 186 py_kind = PyInt_FromLong(kind); | |
| 187 else { | |
| 188 py_kind = Py_None; | |
| 189 Py_INCREF(Py_None); | |
| 190 } | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
191 |
| 5421 | 192 val = PyTuple_New(do_stat ? 3 : 2); |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
193 |
| 5421 | 194 if (!name || !py_kind || !val) { |
| 195 Py_XDECREF(name); | |
| 196 Py_XDECREF(py_kind); | |
| 197 Py_XDECREF(val); | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
198 goto bail; |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
199 } |
|
5416
ca890c0c3f1f
osutil.c: style fix - delete trailing end-of-line spaces
Giorgos Keramidas <keramida@ceid.upatras.gr>
parents:
5398
diff
changeset
|
200 |
| 5421 | 201 PyTuple_SET_ITEM(val, 0, name); |
| 202 PyTuple_SET_ITEM(val, 1, py_kind); | |
| 203 if (do_stat) { | |
| 204 PyTuple_SET_ITEM(val, 2, Py_None); | |
| 205 Py_INCREF(Py_None); | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
206 } |
| 5421 | 207 |
| 208 PyList_Append(list, val); | |
| 209 Py_DECREF(val); | |
| 210 } | |
| 211 | |
| 212 PyList_Sort(list); | |
| 213 size = PyList_Size(list); | |
| 214 #ifdef AT_SYMLINK_NOFOLLOW | |
| 215 dfd = dirfd(dir); | |
| 216 #endif | |
| 217 | |
| 5423 | 218 if (!(do_stat || !all_kinds)) |
| 219 goto done; | |
| 5421 | 220 |
| 5423 | 221 for (i = 0; i < size; i++) { |
| 222 PyObject *elt = PyList_GetItem(list, i); | |
| 223 char *name = PyString_AsString(PyTuple_GET_ITEM(elt, 0)); | |
| 224 PyObject *py_st = NULL; | |
| 225 PyObject *py_kind = PyTuple_GET_ITEM(elt, 1); | |
|
5424
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
226 struct listdir_stat *st; |
|
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
227 struct stat buf; |
|
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
228 struct stat *stp = &buf; |
| 5423 | 229 int kind; |
| 5421 | 230 |
| 5423 | 231 kind = py_kind == Py_None ? -1 : PyInt_AsLong(py_kind); |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
232 |
| 5423 | 233 if (kind != -1 && !do_stat) |
| 234 continue; | |
| 235 | |
| 236 strncat(full_path + path_len + 1, name, PATH_MAX - path_len); | |
| 237 full_path[PATH_MAX] = 0; | |
| 238 | |
| 239 if (do_stat) { | |
| 240 if (!ctor_args) { | |
| 241 ctor_args = PyTuple_New(0); | |
| 242 if (!ctor_args) | |
| 5421 | 243 goto bail; |
| 244 } | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
245 |
|
5424
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
246 py_st = PyObject_CallObject((PyObject *)&listdir_stat_type, |
| 5423 | 247 ctor_args); |
|
5424
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
248 st = (struct listdir_stat *)py_st; |
| 5423 | 249 if (!st) |
| 250 goto bail; | |
|
5424
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
251 stp = &st->st; |
|
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
252 PyTuple_SET_ITEM(elt, 2, py_st); |
|
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
253 } |
|
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
254 |
| 5423 | 255 #ifdef AT_SYMLINK_NOFOLLOW |
|
5424
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
256 ret = fstatat(dfd, name, stp, AT_SYMLINK_NOFOLLOW); |
| 5423 | 257 #else |
|
5424
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
258 ret = lstat(full_path, stp); |
| 5423 | 259 #endif |
|
5424
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
260 if (ret == -1) { |
|
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
261 list = PyErr_SetFromErrnoWithFilename(PyExc_OSError, |
|
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
262 full_path); |
|
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
263 goto bail; |
| 5423 | 264 } |
|
5424
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
265 if (kind == -1) |
|
005638db9d0c
osutil: fold stat paths together
Matt Mackall <mpm@selenic.com>
parents:
5423
diff
changeset
|
266 kind = mode_to_kind(stp->st_mode); |
| 5423 | 267 |
| 268 if (py_kind == Py_None && kind != -1) { | |
| 269 py_kind = PyInt_FromLong(kind); | |
| 270 if (!py_kind) | |
| 271 goto bail; | |
| 272 Py_XDECREF(Py_None); | |
| 273 PyTuple_SET_ITEM(elt, 1, py_kind); | |
| 274 } | |
| 5421 | 275 } |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
276 |
| 5421 | 277 goto done; |
| 278 | |
| 5423 | 279 bail: |
| 280 Py_XDECREF(list); | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
281 |
| 5423 | 282 done: |
| 283 Py_XDECREF(ctor_args); | |
| 284 if (dir) | |
| 285 closedir(dir); | |
| 5421 | 286 return list; |
| 287 } | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
288 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
289 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
290 static char osutil_doc[] = "Native operating system services."; |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
291 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
292 static PyMethodDef methods[] = { |
| 5421 | 293 {"listdir", (PyCFunction)listdir, METH_VARARGS | METH_KEYWORDS, |
| 294 "list a directory\n"}, | |
| 295 {NULL, NULL} | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
296 }; |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
297 |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
298 PyMODINIT_FUNC initosutil(void) |
|
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
299 { |
| 5421 | 300 if (PyType_Ready(&listdir_stat_type) == -1) |
| 301 return; | |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
302 |
| 5421 | 303 Py_InitModule3("osutil", methods, osutil_doc); |
|
5396
5105b119edd2
Add osutil module, containing a listdir function.
Bryan O'Sullivan <bos@serpentine.com>
parents:
diff
changeset
|
304 } |