dlls/qmgr/enum_jobs.c

   1 /*
   2  * Queue Manager (BITS) Job Enumerator
   3  *
   4  * Copyright 2007 Google (Roy Shea)
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "qmgr.h"
  22 #include "wine/debug.h"
  23
  24 WINE_DEFAULT_DEBUG_CHANNEL(qmgr);
  25
  26 typedef struct
  27 {
  28     IEnumBackgroundCopyJobs IEnumBackgroundCopyJobs_iface;
  29     LONG ref;
  30     IBackgroundCopyJob4 **jobs;
  31     ULONG numJobs;
  32     ULONG indexJobs;
  33 } EnumBackgroundCopyJobsImpl;
  34
  35 static inline EnumBackgroundCopyJobsImpl *impl_from_IEnumBackgroundCopyJobs(IEnumBackgroundCopyJobs *iface)
  36 {
  37     return CONTAINING_RECORD(iface, EnumBackgroundCopyJobsImpl, IEnumBackgroundCopyJobs_iface);
  38 }
  39
  40 static HRESULT WINAPI EnumBackgroundCopyJobs_QueryInterface(IEnumBackgroundCopyJobs *iface,
  41         REFIID riid, void **ppv)
  42 {
  43     EnumBackgroundCopyJobsImpl *This = impl_from_IEnumBackgroundCopyJobs(iface);
  44
  45     TRACE("(%p)->(%s, %p)\n", This, debugstr_guid(riid), ppv);
  46
  47     if (IsEqualGUID(riid, &IID_IUnknown) || IsEqualGUID(riid, &IID_IEnumBackgroundCopyJobs))
  48     {
  49         *ppv = iface;
  50         IEnumBackgroundCopyJobs_AddRef(iface);
  51         return S_OK;
  52     }
  53
  54     *ppv = NULL;
  55     return E_NOINTERFACE;
  56 }
  57
  58 static ULONG WINAPI EnumBackgroundCopyJobs_AddRef(IEnumBackgroundCopyJobs *iface)
  59 {
  60     EnumBackgroundCopyJobsImpl *This = impl_from_IEnumBackgroundCopyJobs(iface);
  61     ULONG ref = InterlockedIncrement(&This->ref);
  62
  63     TRACE("(%p)->(%ld)\n", This, ref);
  64
  65     return ref;
  66 }
  67
  68 static ULONG WINAPI EnumBackgroundCopyJobs_Release(IEnumBackgroundCopyJobs *iface)
  69 {
  70     EnumBackgroundCopyJobsImpl *This = impl_from_IEnumBackgroundCopyJobs(iface);
  71     ULONG ref = InterlockedDecrement(&This->ref);
  72     ULONG i;
  73
  74     TRACE("(%p)->(%ld)\n", This, ref);
  75
  76     if (ref == 0) {
  77         for(i = 0; i < This->numJobs; i++)
  78             IBackgroundCopyJob4_Release(This->jobs[i]);
  79         free(This->jobs);
  80         free(This);
  81     }
  82
  83     return ref;
  84 }
  85
  86 static HRESULT WINAPI EnumBackgroundCopyJobs_Next(IEnumBackgroundCopyJobs *iface, ULONG celt,
  87         IBackgroundCopyJob **rgelt, ULONG *pceltFetched)
  88 {
  89     EnumBackgroundCopyJobsImpl *This = impl_from_IEnumBackgroundCopyJobs(iface);
  90     ULONG fetched;
  91     ULONG i;
  92
  93     TRACE("(%p)->(%ld %p %p)\n", This, celt, rgelt, pceltFetched);
  94
  95     fetched = min(celt, This->numJobs - This->indexJobs);
  96     if (pceltFetched)
  97         *pceltFetched = fetched;
  98     else
  99     {
 100         /* We need to initialize this array if the caller doesn't request
 101            the length because length_is will default to celt.  */
 102         for (i = 0; i < celt; ++i)
 103             rgelt[i] = NULL;
 104
 105         /* pceltFetched can only be NULL if celt is 1 */
 106         if (celt != 1)
 107             return E_INVALIDARG;
 108     }
 109
 110     /* Fill in the array of objects */
 111     for (i = 0; i < fetched; ++i)
 112     {
 113         rgelt[i] = (IBackgroundCopyJob *)This->jobs[This->indexJobs++];
 114         IBackgroundCopyJob_AddRef(rgelt[i]);
 115     }
 116
 117     return fetched == celt ? S_OK : S_FALSE;
 118 }
 119
 120 static HRESULT WINAPI EnumBackgroundCopyJobs_Skip(IEnumBackgroundCopyJobs *iface, ULONG celt)
 121 {
 122     EnumBackgroundCopyJobsImpl *This = impl_from_IEnumBackgroundCopyJobs(iface);
 123
 124     TRACE("(%p)->(%ld)\n", This, celt);
 125
 126     if (This->numJobs - This->indexJobs < celt)
 127     {
 128         This->indexJobs = This->numJobs;
 129         return S_FALSE;
 130     }
 131
 132     This->indexJobs += celt;
 133     return S_OK;
 134 }
 135
 136 static HRESULT WINAPI EnumBackgroundCopyJobs_Reset(IEnumBackgroundCopyJobs *iface)
 137 {
 138     EnumBackgroundCopyJobsImpl *This = impl_from_IEnumBackgroundCopyJobs(iface);
 139
 140     TRACE("(%p)\n", This);
 141
 142     This->indexJobs = 0;
 143     return S_OK;
 144 }
 145
 146 static HRESULT WINAPI EnumBackgroundCopyJobs_Clone(IEnumBackgroundCopyJobs *iface,
 147         IEnumBackgroundCopyJobs **ppenum)
 148 {
 149     EnumBackgroundCopyJobsImpl *This = impl_from_IEnumBackgroundCopyJobs(iface);
 150     FIXME("(%p)->(%p): stub\n", This, ppenum);
 151     return E_NOTIMPL;
 152 }
 153
 154 static HRESULT WINAPI EnumBackgroundCopyJobs_GetCount(IEnumBackgroundCopyJobs *iface,
 155     ULONG *puCount)
 156 {
 157     EnumBackgroundCopyJobsImpl *This = impl_from_IEnumBackgroundCopyJobs(iface);
 158
 159     TRACE("(%p)->(%p)\n", This, puCount);
 160
 161     *puCount = This->numJobs;
 162     return S_OK;
 163 }
 164
 165 static const IEnumBackgroundCopyJobsVtbl EnumBackgroundCopyJobsVtbl =
 166 {
 167     EnumBackgroundCopyJobs_QueryInterface,
 168     EnumBackgroundCopyJobs_AddRef,
 169     EnumBackgroundCopyJobs_Release,
 170     EnumBackgroundCopyJobs_Next,
 171     EnumBackgroundCopyJobs_Skip,
 172     EnumBackgroundCopyJobs_Reset,
 173     EnumBackgroundCopyJobs_Clone,
 174     EnumBackgroundCopyJobs_GetCount
 175 };
 176
 177 HRESULT enum_copy_job_create(BackgroundCopyManagerImpl *qmgr, IEnumBackgroundCopyJobs **enumjob)
 178 {
 179     EnumBackgroundCopyJobsImpl *This;
 180     BackgroundCopyJobImpl *job;
 181     ULONG i;
 182
 183     TRACE("%p, %p)\n", qmgr, enumjob);
 184
 185     This = malloc(sizeof(*This));
 186     if (!This)
 187         return E_OUTOFMEMORY;
 188     This->IEnumBackgroundCopyJobs_iface.lpVtbl = &EnumBackgroundCopyJobsVtbl;
 189     This->ref = 1;
 190
 191     /* Create array of jobs */
 192     This->indexJobs = 0;
 193
 194     EnterCriticalSection(&qmgr->cs);
 195     This->numJobs = list_count(&qmgr->jobs);
 196
 197     if (0 < This->numJobs)
 198     {
 199         This->jobs = malloc(This->numJobs * sizeof *This->jobs);
 200         if (!This->jobs)
 201         {
 202             LeaveCriticalSection(&qmgr->cs);
 203             free(This);
 204             return E_OUTOFMEMORY;
 205         }
 206     }
 207     else
 208         This->jobs = NULL;
 209
 210     i = 0;
 211     LIST_FOR_EACH_ENTRY(job, &qmgr->jobs, BackgroundCopyJobImpl, entryFromQmgr)
 212     {
 213         IBackgroundCopyJob4_AddRef(&job->IBackgroundCopyJob4_iface);
 214         This->jobs[i++] = &job->IBackgroundCopyJob4_iface;
 215     }
 216     LeaveCriticalSection(&qmgr->cs);
 217
 218     *enumjob = &This->IEnumBackgroundCopyJobs_iface;
 219     return S_OK;
 220 }