12 #ifdef _WIN32

    13 #include <windows.h>

    14 #else

    46 #endif

    57 #ifdef _WIN32

    58 static PyObject *listdir_stat_st_size(PyObject *self, void *x)

    59 {

    60 	return PyLong_FromLongLong(

    61 		(PY_LONG_LONG)((struct listdir_stat *)self)->st.st_size);

    62 }

    63 #else

    65 #endif

   131 #ifdef _WIN32

   132 

   133 static int to_python_time(const FILETIME *tm)

   134 {

   135 	/* number of seconds between epoch and January 1 1601 */

   136 	const __int64 a0 = (__int64)134774L * (__int64)24L * (__int64)3600L;

   137 	/* conversion factor from 100ns to 1s */

   138 	const __int64 a1 = 10000000;

   139 	/* explicit (int) cast to suspend compiler warnings */

   140 	return (int)((((__int64)tm->dwHighDateTime << 32)

   141 			+ tm->dwLowDateTime) / a1 - a0);

   142 }

   143 

   144 static PyObject *make_item(const WIN32_FIND_DATAA *fd, int wantstat)

   145 {

   146 	PyObject *py_st;

   147 	struct hg_stat *stp;

   148 

   149 	int kind = (fd->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)

   150 		? _S_IFDIR : _S_IFREG;

   151 

   152 	if (!wantstat)

   153 		return Py_BuildValue("si", fd->cFileName, kind);

   154 

   155 	py_st = PyObject_CallObject((PyObject *)&listdir_stat_type, NULL);

   156 	if (!py_st)

   157 		return NULL;

   158 

   159 	stp = &((struct listdir_stat *)py_st)->st;

   160 	/*

   161 	use kind as st_mode

   162 	rwx bits on Win32 are meaningless

   163 	and Hg does not use them anyway

   164 	*/

   165 	stp->st_mode  = kind;

   166 	stp->st_mtime = to_python_time(&fd->ftLastWriteTime);

   167 	stp->st_ctime = to_python_time(&fd->ftCreationTime);

   168 	if (kind == _S_IFREG)

   169 		stp->st_size =	((__int64)fd->nFileSizeHigh << 32)

   170 				+ fd->nFileSizeLow;

   171 	return Py_BuildValue("siN", fd->cFileName,

   172 		kind, py_st);

   173 }

   174 

   175 static PyObject *listdir(PyObject *self, PyObject *args, PyObject *kwargs)

   176 {

   177 	PyObject *rval = NULL; /* initialize - return value */

   178 	PyObject *statobj = NULL; /* initialize - optional arg */

   179 	PyObject *list;

   180 	HANDLE fh;

   181 	WIN32_FIND_DATAA fd;

   182 	char *path, *pattern, *skip = NULL;

   183 	int plen, wantstat;

   184 

   185 	static char *kwlist[] = {"path", "stat", "skip", NULL};

   186 

   187 	if (!PyArg_ParseTupleAndKeywords(args, kwargs, "s#|Os:listdir",

   188 			kwlist, &path, &plen, &statobj, &skip))

   189 		goto error_parse;

   190 

   191 	wantstat = statobj && PyObject_IsTrue(statobj);

   192 

   193 	/* build the path + \* pattern string */

   194 	pattern = malloc(plen+3); /* path + \* + \0 */

   195 	if (!pattern) {

   196 		PyErr_NoMemory();

   197 		goto error_parse;

   198 	}

   199 	strcpy(pattern, path);

   200 

   201 	if (plen > 0) {

   202 		char c = path[plen-1];

   203 		if (c != ':' && c != '/' && c != '\\')

   204 			pattern[plen++] = '\\';

   205 	}

   206 	strcpy(pattern + plen, "*");

   207 

   208 	fh = FindFirstFileA(pattern, &fd);

   209 	if (fh == INVALID_HANDLE_VALUE) {

   210 		PyErr_SetExcFromWindowsErrWithFilename(PyExc_OSError,

   211 							GetLastError(),

   212 							path);

   213 		goto error_file;

   214 	}

   215 

   216 	list = PyList_New(0);

   217 	if (!list)

   218 		goto error_list;

   219 

   220 	do {

   221 		PyObject *item;

   222 

   223 		if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {

   224 			if (!strcmp(fd.cFileName, ".")

   225 			|| !strcmp(fd.cFileName, ".."))

   226 				continue;

   227 

   228 			if (skip && !strcmp(fd.cFileName, skip)) {

   229 				rval = PyList_New(0);

   230 				goto error;

   231 			}

   232 		}

   233 

   234 		item = make_item(&fd, wantstat);

   235 		if (!item)

   236 			goto error;

   237 

   238 		if (PyList_Append(list, item)) {

   239 			Py_XDECREF(item);

   240 			goto error;

   241 		}

   242 

   243 		Py_XDECREF(item);

   244 	} while (FindNextFileA(fh, &fd));

   245 

   246 	if (GetLastError() != ERROR_NO_MORE_FILES) {

   247 		PyErr_SetExcFromWindowsErrWithFilename(PyExc_OSError,

   248 							GetLastError(),

   249 							path);

   250 		goto error;

   251 	}

   252 

   253 	rval = list;

   254 	Py_XINCREF(rval);

   255 error:

   256 	Py_XDECREF(list);

   257 error_list:

   258 	FindClose(fh);

   259 error_file:

   260 	free(pattern);

   261 error_parse:

   262 	return rval;

   263 }

   264 

   265 #else

   266 

   385 #endif /* ndef _WIN32 */

   386