1 /* Copyright (c) Mark Harmstone 2016-17
2  *
3  * This file is part of WinBtrfs.
4  *
5  * WinBtrfs is free software: you can redistribute it and/or modify
6  * it under the terms of the GNU Lesser General Public Licence as published by
7  * the Free Software Foundation, either version 3 of the Licence, or
8  * (at your option) any later version.
9  *
10  * WinBtrfs is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU Lesser General Public Licence for more details.
14  *
15  * You should have received a copy of the GNU Lesser General Public Licence
16  * along with WinBtrfs.  If not, see <http://www.gnu.org/licenses/>. */
17 
18 #pragma once
19 
20 #include <shlobj.h>
21 
22 extern LONG objs_loaded;
23 
24 class BtrfsIconOverlay : public IShellIconOverlayIdentifier {
25 public:
26     BtrfsIconOverlay() {
27         refcount = 0;
28         InterlockedIncrement(&objs_loaded);
29     }
30 
31     virtual ~BtrfsIconOverlay() {
32         InterlockedDecrement(&objs_loaded);
33     }
34 
35     // IUnknown
36 
37     HRESULT __stdcall QueryInterface(REFIID riid, void **ppObj);
38 
39     ULONG __stdcall AddRef() {
40         return InterlockedIncrement(&refcount);
41     }
42 
43     ULONG __stdcall Release() {
44         LONG rc = InterlockedDecrement(&refcount);
45 
46         if (rc == 0)
47             delete this;
48 
49         return rc;
50     }
51 
52     // IShellIconOverlayIdentifier
53 
54     virtual HRESULT __stdcall GetOverlayInfo(PWSTR pwszIconFile, int cchMax, int* pIndex, DWORD* pdwFlags);
55     virtual HRESULT __stdcall GetPriority(int *pPriority);
56     virtual HRESULT __stdcall IsMemberOf(PCWSTR pwszPath, DWORD dwAttrib);
57 
58 private:
59     LONG refcount;
60 };
61