static PyObject *t_resourcebundle_setAppData(PyTypeObject *type,
PyObject *args)
{
charsArg packageName, path;
if (!parseArgs(args, "nf", &packageName, &path))
{
HANDLE fd = CreateFile(path, GENERIC_READ, FILE_SHARE_READ,
NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
UErrorCode status = U_ZERO_ERROR;
DWORD dwSize;
HANDLE hMap;
LPVOID data;
if (fd == INVALID_HANDLE_VALUE)
return PyErr_SetFromWindowsErrWithFilename(0, path);
dwSize = GetFileSize(fd, NULL);
if (dwSize == INVALID_FILE_SIZE)
{
PyErr_SetFromWindowsErrWithFilename(0, path);
CloseHandle(fd);
return NULL;
}
hMap = CreateFileMapping(fd, NULL, PAGE_READONLY, 0, dwSize, NULL);
if (!hMap)
{
PyErr_SetFromWindowsErrWithFilename(0, path);
CloseHandle(fd);
return NULL;
}
CloseHandle(fd);
data = MapViewOfFile(hMap, FILE_MAP_READ, 0, 0, 0);
if (!data)
{
PyErr_SetFromWindowsErrWithFilename(0, path);
CloseHandle(hMap);
return NULL;
}
CloseHandle(hMap);
udata_setAppData(packageName, data, &status);
if (U_FAILURE(status))
{
UnmapViewOfFile(data);
return ICUException(status).reportError();
}
Py_RETURN_NONE;
}
return PyErr_SetArgsError(type, "setAppData", args);
}
PyObject *
GetProcessDict(PyObject *self, PyObject *args)
{
HANDLE hProcessSnap;
PROCESSENTRY32 pe32;
PyObject *pyDict, *pyPid, *pyName;
if (!PyArg_ParseTuple(args, ""))
return NULL;
// Take a snapshot of all processes in the system.
hProcessSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
if(hProcessSnap == INVALID_HANDLE_VALUE)
{
PyErr_SetFromWindowsErrWithFilename(0, "CreateToolhelp32Snapshot");
return NULL;
}
// Set the size of the structure before using it.
pe32.dwSize = sizeof(PROCESSENTRY32);
// Retrieve information about the first process,
// and exit if unsuccessful
if(!Process32First(hProcessSnap, &pe32))
{
CloseHandle(hProcessSnap); // clean the snapshot object
PyErr_SetFromWindowsErrWithFilename(0, "Process32First");
return NULL;
}
// create a dictionary for the results
pyDict = PyDict_New();
// Now walk the snapshot of processes
do
{
pyPid = PyInt_FromLong(pe32.th32ProcessID);
pyName = PyUnicode_FromWideChar(pe32.szExeFile, wcslen(pe32.szExeFile));
PyDict_SetItem(pyDict, pyPid, pyName);
Py_DECREF(pyPid);
Py_DECREF(pyName);
}while(Process32Next(hProcessSnap, &pe32));
CloseHandle(hProcessSnap);
return pyDict;
}
PyObject *
GetProcessName(PyObject *self, PyObject *args)
{
HANDLE hProcessSnap;
PROCESSENTRY32 pe32;
DWORD pid;
if (!PyArg_ParseTuple(args, "k", &pid))
return NULL;
// Take a snapshot of all processes in the system.
hProcessSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
if(hProcessSnap == INVALID_HANDLE_VALUE)
{
PyErr_SetFromWindowsErrWithFilename(0, "CreateToolhelp32Snapshot");
return NULL;
}
// Set the size of the structure before using it.
pe32.dwSize = sizeof(PROCESSENTRY32);
// Retrieve information about the first process,
// and exit if unsuccessful
if(!Process32First(hProcessSnap, &pe32))
{
CloseHandle(hProcessSnap); // clean the snapshot object
PyErr_SetFromWindowsErrWithFilename(0, "Process32First");
return NULL;
}
// Now walk the snapshot of processes
do
{
if(pe32.th32ProcessID == pid)
{
CloseHandle(hProcessSnap);
return PyUnicode_FromWideChar(pe32.szExeFile, wcslen(pe32.szExeFile));
}
}while(Process32Next(hProcessSnap, &pe32));
CloseHandle(hProcessSnap);
return Py_BuildValue("s", "<not found>");
}
static PyObject *posixfile(PyObject *self, PyObject *args, PyObject *kwds)
{
static char *kwlist[] = {"name", "mode", "buffering", NULL};
PyObject *file_obj = NULL;
char *name = NULL;
char *mode = "rb";
DWORD access = 0;
DWORD creation;
HANDLE handle;
int fd, flags = 0;
int bufsize = -1;
char m0, m1, m2;
char fpmode[4];
int fppos = 0;
int plus;
FILE *fp;
if (!PyArg_ParseTupleAndKeywords(args, kwds, "et|si:posixfile", kwlist,
Py_FileSystemDefaultEncoding,
&name, &mode, &bufsize))
return NULL;
m0 = mode[0];
m1 = m0 ? mode[1] : '\0';
m2 = m1 ? mode[2] : '\0';
plus = m1 == '+' || m2 == '+';
fpmode[fppos++] = m0;
if (m1 == 'b' || m2 == 'b') {
flags = _O_BINARY;
fpmode[fppos++] = 'b';
}
else
flags = _O_TEXT;
if (plus) {
flags |= _O_RDWR;
access = GENERIC_READ | GENERIC_WRITE;
fpmode[fppos++] = '+';
}
fpmode[fppos++] = '\0';
switch (m0) {
case 'r':
creation = OPEN_EXISTING;
if (!plus) {
flags |= _O_RDONLY;
access = GENERIC_READ;
}
break;
case 'w':
creation = CREATE_ALWAYS;
if (!plus) {
access = GENERIC_WRITE;
flags |= _O_WRONLY;
}
break;
case 'a':
creation = OPEN_ALWAYS;
flags |= _O_APPEND;
if (!plus) {
flags |= _O_WRONLY;
access = GENERIC_WRITE;
}
break;
default:
PyErr_Format(PyExc_ValueError,
"mode string must begin with one of 'r', 'w', "
"or 'a', not '%c'", m0);
goto bail;
}
handle = CreateFile(name, access,
FILE_SHARE_READ | FILE_SHARE_WRITE |
FILE_SHARE_DELETE,
NULL,
creation,
FILE_ATTRIBUTE_NORMAL,
0);
if (handle == INVALID_HANDLE_VALUE) {
PyErr_SetFromWindowsErrWithFilename(GetLastError(), name);
goto bail;
}
fd = _open_osfhandle((intptr_t)handle, flags);
if (fd == -1) {
CloseHandle(handle);
PyErr_SetFromErrnoWithFilename(PyExc_IOError, name);
goto bail;
}
#ifndef IS_PY3K
fp = _fdopen(fd, fpmode);
if (fp == NULL) {
_close(fd);
PyErr_SetFromErrnoWithFilename(PyExc_IOError, name);
goto bail;
}
file_obj = PyFile_FromFile(fp, name, mode, fclose);
if (file_obj == NULL) {
fclose(fp);
goto bail;
}
PyFile_SetBufSize(file_obj, bufsize);
#else
file_obj = PyFile_FromFd(fd, name, mode, bufsize, NULL, NULL, NULL, 1);
if (file_obj == NULL)
goto bail;
#endif
bail:
PyMem_Free(name);
return file_obj;
}
static PyObject *_listdir(char *path, int plen, int wantstat, char *skip)
{
PyObject *rval = NULL; /* initialize - return value */
PyObject *list;
HANDLE fh;
WIN32_FIND_DATAA fd;
char *pattern;
/* build the path + \* pattern string */
pattern = malloc(plen + 3); /* path + \* + \0 */
if (!pattern) {
PyErr_NoMemory();
goto error_nomem;
}
strcpy(pattern, path);
if (plen > 0) {
char c = path[plen-1];
if (c != ':' && c != '/' && c != '\\')
pattern[plen++] = '\\';
}
strcpy(pattern + plen, "*");
fh = FindFirstFileA(pattern, &fd);
if (fh == INVALID_HANDLE_VALUE) {
PyErr_SetFromWindowsErrWithFilename(GetLastError(), path);
goto error_file;
}
list = PyList_New(0);
if (!list)
goto error_list;
do {
PyObject *item;
if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
if (!strcmp(fd.cFileName, ".")
|| !strcmp(fd.cFileName, ".."))
continue;
if (skip && !strcmp(fd.cFileName, skip)) {
rval = PyList_New(0);
goto error;
}
}
item = make_item(&fd, wantstat);
if (!item)
goto error;
if (PyList_Append(list, item)) {
Py_XDECREF(item);
goto error;
}
Py_XDECREF(item);
} while (FindNextFileA(fh, &fd));
if (GetLastError() != ERROR_NO_MORE_FILES) {
PyErr_SetFromWindowsErrWithFilename(GetLastError(), path);
goto error;
}
rval = list;
Py_XINCREF(rval);
error:
Py_XDECREF(list);
error_list:
FindClose(fh);
error_file:
free(pattern);
error_nomem:
return rval;
}
