Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Use RAII in winutil

Browse Source
This commit is contained in:
Kovid Goyal 2019-06-11 18:49:36 +05:30
parent 48e26a754d
commit 59264be1f0
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
2 changed files with 45 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/calibre/utils/windows/winutil.c
View File

@ -375,10 +375,10 @@ winutil_strftime(PyObject *self, PyObject *args)
}
static char winutil_doc[] = "Defines utility methods to interface with windows.";
extern PyObject *add_to_recent_docs(PyObject *self, PyObject *args);
extern PyObject *file_association(PyObject *self, PyObject *args);
extern PyObject *friendly_name(PyObject *self, PyObject *args);
extern PyObject *notify_associations_changed(PyObject *self, PyObject *args);
extern PyObject *winutil_add_to_recent_docs(PyObject *self, PyObject *args);
extern PyObject *winutil_file_association(PyObject *self, PyObject *args);
extern PyObject *winutil_friendly_name(PyObject *self, PyObject *args);
extern PyObject *winutil_notify_associations_changed(PyObject *self, PyObject *args);
static PyMethodDef winutil_methods[] = {
{"special_folder_path", winutil_folder_path, METH_VARARGS,

67
src/calibre/utils/windows/winutilpp.cpp
View File

@ -29,71 +29,80 @@ py_to_wchar(PyObject *obj, wchar_t **output) {
return 1;
}
static inline void
free_wchar_buffer(wchar_t **buf) {
class wchar_raii {
private:
wchar_t **handle;
// copy and assignment not implemented; prevent their use by
// declaring private.
wchar_raii( const wchar_raii & ) ;
wchar_raii & operator=( const wchar_raii & ) ;
public:
wchar_raii(wchar_t **buf) : handle(*buf) {}
~wchar_raii() {
#if PY_MAJOR_VERSION >= 3
PyMem_Free(*buf);
PyMem_Free(*handle);
#endif
*buf = NULL;
*handle = NULL;
}
wchar_t *ptr() { return *handle; }
};
extern "C" {
PyObject *
add_to_recent_docs(PyObject *self, PyObject *args) {
wchar_t *path, *app_id;
if (!PyArg_ParseTuple(args, "O&O&", py_to_wchar, &path, py_to_wchar, &app_id)) return NULL;
if (app_id) {
winutil_add_to_recent_docs(PyObject *self, PyObject *args) {
wchar_t *path_, *app_id_;
if (!PyArg_ParseTuple(args, "O&O&", py_to_wchar, &path_, py_to_wchar, &app_id_)) return NULL;
wchar_raii path(path_), app_id(app_id_);
if (app_id.ptr()) {
CComPtr<IShellItem> item;
HRESULT hr = SHCreateItemFromParsingName(path, NULL, IID_PPV_ARGS(&item));
HRESULT hr = SHCreateItemFromParsingName(path.ptr(), NULL, IID_PPV_ARGS(&item));
if (SUCCEEDED(hr)) {
SHARDAPPIDINFO info;
info.psi = item;
info.pszAppID = app_id;
info.pszAppID = app_id.ptr();
SHAddToRecentDocs(SHARD_APPIDINFO, &info);
}
} else {
SHAddToRecentDocs(SHARD_PATHW, path);
SHAddToRecentDocs(SHARD_PATHW, path.ptr());
}
free_wchar_buffer(&path); free_wchar_buffer(&app_id);
Py_RETURN_NONE;
}
PyObject *
file_association(PyObject *self, PyObject *args) {
wchar_t *ext, buf[2048];
winutil_file_association(PyObject *self, PyObject *args) {
wchar_t *ext_, buf[2048];
DWORD sz = sizeof(buf);
if (!PyArg_ParseTuple(args, "O&", py_to_wchar, &ext)) return NULL;
HRESULT hr = AssocQueryStringW(0, ASSOCSTR_EXECUTABLE, ext, NULL, buf, &sz);
free_wchar_buffer(&ext);
if (!PyArg_ParseTuple(args, "O&", py_to_wchar, &ext_)) return NULL;
wchar_raii ext(ext_);
HRESULT hr = AssocQueryStringW(0, ASSOCSTR_EXECUTABLE, ext.ptr(), NULL, buf, &sz);
if (!SUCCEEDED(hr) || sz < 1) Py_RETURN_NONE;
return Py_BuildValue("u", buf);
}
PyObject *
friendly_name(PyObject *self, PyObject *args) {
wchar_t *exe, *prog_id, buf[2048], *p;
winutil_friendly_name(PyObject *self, PyObject *args) {
wchar_t *exe_, *prog_id_, buf[2048], *p;
DWORD sz = sizeof(buf);
if (!PyArg_ParseTuple(args, "O&O&", py_to_wchar, &prog_id, py_to_wchar, &exe)) return NULL;
if (!PyArg_ParseTuple(args, "O&O&", py_to_wchar, &prog_id_, py_to_wchar, &exe_)) return NULL;
wchar_raii exe(exe_), prog_id(prog_id_);
ASSOCF flags = ASSOCF_REMAPRUNDLL;
if (exe) {
p = exe;
if (exe.ptr()) {
p = exe.ptr();
flags |= ASSOCF_OPEN_BYEXENAME;
} else p = prog_id;
if (!p) {
free_wchar_buffer(&exe); free_wchar_buffer(&prog_id);
Py_RETURN_NONE;
}
} else p = prog_id.ptr();
if (!p) Py_RETURN_NONE;
HRESULT hr = AssocQueryStringW(flags, ASSOCSTR_FRIENDLYAPPNAME, p, NULL, buf, &sz);
free_wchar_buffer(&exe); free_wchar_buffer(&prog_id);
if (!SUCCEEDED(hr) || sz < 1) Py_RETURN_NONE;
return Py_BuildValue("u", buf);
}
PyObject *
notify_associations_changed(PyObject *self, PyObject *args) {
winutil_notify_associations_changed(PyObject *self, PyObject *args) {
SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_DWORD | SHCNF_FLUSH, NULL, NULL);
Py_RETURN_NONE;
}