dll/shellext/shellbtrfs/iconoverlay.h

   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 };