/* * Copyright 1999 Marcus Meissner * Copyright 2002-2003 Michael Günnewig * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ /* TODO: * - IAVIStreaming interface is missing for the IAVIStreamImpl * - IAVIStream_fnFindSample: FIND_INDEX isn't supported. * - IAVIStream_fnReadFormat: formatchanges aren't read in. * - IAVIStream_fnDelete: a stub. * - IAVIStream_fnSetInfo: a stub. * - make thread safe * * KNOWN Bugs: * - native version can hangup when reading a file generated with this DLL. * When index is missing it works, but index seems to be okay. */ #define COBJMACROS #include <assert.h> #include <stdarg.h> #include "windef.h" #include "winbase.h" #include "wingdi.h" #include "winuser.h" #include "winnls.h" #include "winerror.h" #include "mmsystem.h" #include "vfw.h" #include "avifile_private.h" #include "extrachunk.h" #include "wine/debug.h" WINE_DEFAULT_DEBUG_CHANNEL(avifile); #ifndef IDX_PER_BLOCK #define IDX_PER_BLOCK 2730 #endif typedef struct _IAVIFileImpl IAVIFileImpl; typedef struct _IAVIStreamImpl { IAVIStream IAVIStream_iface; LONG ref; IAVIFileImpl *paf; DWORD nStream; /* the n-th stream in file */ AVISTREAMINFOW sInfo; LPVOID lpFormat; DWORD cbFormat; LPVOID lpHandlerData; DWORD cbHandlerData; EXTRACHUNKS extra; LPDWORD lpBuffer; DWORD cbBuffer; /* size of lpBuffer */ DWORD dwCurrentFrame; /* frame/block currently in lpBuffer */ LONG lLastFrame; /* last correct index in idxFrames */ AVIINDEXENTRY *idxFrames; DWORD nIdxFrames; /* upper index limit of idxFrames */ AVIINDEXENTRY *idxFmtChanges; DWORD nIdxFmtChanges; /* upper index limit of idxFmtChanges */ } IAVIStreamImpl; static inline IAVIStreamImpl *impl_from_IAVIStream(IAVIStream *iface) { return CONTAINING_RECORD(iface, IAVIStreamImpl, IAVIStream_iface); } struct _IAVIFileImpl { IUnknown IUnknown_inner; IAVIFile IAVIFile_iface; IPersistFile IPersistFile_iface; IUnknown *outer_unk; LONG ref; AVIFILEINFOW fInfo; IAVIStreamImpl *ppStreams[MAX_AVISTREAMS]; EXTRACHUNKS fileextra; DWORD dwMoviChunkPos; /* some stuff for saving ... */ DWORD dwIdxChunkPos; DWORD dwNextFramePos; DWORD dwInitialFrames; MMCKINFO ckLastRecord; AVIINDEXENTRY *idxRecords; /* won't be updated while loading */ DWORD nIdxRecords; /* current fill level */ DWORD cbIdxRecords; /* size of idxRecords */ /* IPersistFile stuff ... */ HMMIO hmmio; LPWSTR szFileName; UINT uMode; BOOL fDirty; }; static inline IAVIFileImpl *impl_from_IUnknown(IUnknown *iface) { return CONTAINING_RECORD(iface, IAVIFileImpl, IUnknown_inner); } static inline IAVIFileImpl *impl_from_IAVIFile(IAVIFile *iface) { return CONTAINING_RECORD(iface, IAVIFileImpl, IAVIFile_iface); } static inline IAVIFileImpl *impl_from_IPersistFile(IPersistFile *iface) { return CONTAINING_RECORD(iface, IAVIFileImpl, IPersistFile_iface); } /***********************************************************************/ static HRESULT AVIFILE_AddFrame(IAVIStreamImpl *This, DWORD ckid, DWORD size, DWORD offset, DWORD flags); static HRESULT AVIFILE_AddRecord(IAVIFileImpl *This); static DWORD AVIFILE_ComputeMoviStart(IAVIFileImpl *This); static void AVIFILE_ConstructAVIStream(IAVIFileImpl *paf, DWORD nr, const AVISTREAMINFOW *asi); static void AVIFILE_DestructAVIStream(IAVIStreamImpl *This); static HRESULT AVIFILE_LoadFile(IAVIFileImpl *This); static HRESULT AVIFILE_LoadIndex(const IAVIFileImpl *This, DWORD size, DWORD offset); static HRESULT AVIFILE_ParseIndex(const IAVIFileImpl *This, AVIINDEXENTRY *lp, LONG count, DWORD pos, BOOL *bAbsolute); static HRESULT AVIFILE_ReadBlock(IAVIStreamImpl *This, DWORD start, LPVOID buffer, DWORD size); static void AVIFILE_SamplesToBlock(const IAVIStreamImpl *This, LPLONG pos, LPLONG offset); static HRESULT AVIFILE_SaveFile(IAVIFileImpl *This); static HRESULT AVIFILE_SaveIndex(const IAVIFileImpl *This); static ULONG AVIFILE_SearchStream(const IAVIFileImpl *This, DWORD fccType, LONG lSkip); static void AVIFILE_UpdateInfo(IAVIFileImpl *This); static HRESULT AVIFILE_WriteBlock(IAVIStreamImpl *This, DWORD block, FOURCC ckid, DWORD flags, LPCVOID buffer, LONG size); static HRESULT WINAPI IUnknown_fnQueryInterface(IUnknown *iface, REFIID riid, void **ppv) { IAVIFileImpl *This = impl_from_IUnknown(iface); TRACE("(%p,%s,%p)\n", This, debugstr_guid(riid), ppv); if (!ppv) { WARN("invalid parameter\n"); return E_INVALIDARG; } *ppv = NULL; if (IsEqualIID(riid, &IID_IUnknown)) *ppv = &This->IUnknown_inner; else if (IsEqualIID(riid, &IID_IAVIFile)) *ppv = &This->IAVIFile_iface; else if (IsEqualGUID(riid, &IID_IPersistFile)) *ppv = &This->IPersistFile_iface; else { WARN("unknown IID %s\n", debugstr_guid(riid)); return E_NOINTERFACE; } /* Violation of the COM aggregation ref counting rule */ IUnknown_AddRef(&This->IUnknown_inner); return S_OK; } static ULONG WINAPI IUnknown_fnAddRef(IUnknown *iface) { IAVIFileImpl *This = impl_from_IUnknown(iface); ULONG ref = InterlockedIncrement(&This->ref); TRACE("(%p) ref=%d\n", This, ref); return ref; } static ULONG WINAPI IUnknown_fnRelease(IUnknown *iface) { IAVIFileImpl *This = impl_from_IUnknown(iface); ULONG ref = InterlockedDecrement(&This->ref); UINT i; TRACE("(%p) ref=%d\n", This, ref); if (!ref) { if (This->fDirty) AVIFILE_SaveFile(This); for (i = 0; i < This->fInfo.dwStreams; i++) { if (This->ppStreams[i] != NULL) { if (This->ppStreams[i]->ref != 0) ERR(": someone has still %u reference to stream %u (%p)!\n", This->ppStreams[i]->ref, i, This->ppStreams[i]); AVIFILE_DestructAVIStream(This->ppStreams[i]); HeapFree(GetProcessHeap(), 0, This->ppStreams[i]); This->ppStreams[i] = NULL; } } if (This->idxRecords != NULL) { HeapFree(GetProcessHeap(), 0, This->idxRecords); This->idxRecords = NULL; This->nIdxRecords = 0; } if (This->fileextra.lp != NULL) { HeapFree(GetProcessHeap(), 0, This->fileextra.lp); This->fileextra.lp = NULL; This->fileextra.cb = 0; } HeapFree(GetProcessHeap(), 0, This->szFileName); This->szFileName = NULL; if (This->hmmio != NULL) { mmioClose(This->hmmio, 0); This->hmmio = NULL; } HeapFree(GetProcessHeap(), 0, This); } return ref; } static const IUnknownVtbl unk_vtbl = { IUnknown_fnQueryInterface, IUnknown_fnAddRef, IUnknown_fnRelease }; static HRESULT WINAPI IAVIFile_fnQueryInterface(IAVIFile *iface, REFIID riid, void **ppv) { IAVIFileImpl *This = impl_from_IAVIFile(iface); return IUnknown_QueryInterface(This->outer_unk, riid, ppv); } static ULONG WINAPI IAVIFile_fnAddRef(IAVIFile *iface) { IAVIFileImpl *This = impl_from_IAVIFile(iface); return IUnknown_AddRef(This->outer_unk); } static ULONG WINAPI IAVIFile_fnRelease(IAVIFile *iface) { IAVIFileImpl *This = impl_from_IAVIFile(iface); return IUnknown_Release(This->outer_unk); } static HRESULT WINAPI IAVIFile_fnInfo(IAVIFile *iface, AVIFILEINFOW *afi, LONG size) { IAVIFileImpl *This = impl_from_IAVIFile(iface); TRACE("(%p,%p,%d)\n",iface,afi,size); if (afi == NULL) return AVIERR_BADPARAM; if (size < 0) return AVIERR_BADSIZE; AVIFILE_UpdateInfo(This); memcpy(afi, &This->fInfo, min((DWORD)size, sizeof(This->fInfo))); if ((DWORD)size < sizeof(This->fInfo)) return AVIERR_BUFFERTOOSMALL; return AVIERR_OK; } static HRESULT WINAPI IAVIFile_fnGetStream(IAVIFile *iface, IAVIStream **avis, DWORD fccType, LONG lParam) { IAVIFileImpl *This = impl_from_IAVIFile(iface); ULONG nStream; TRACE("(%p,%p,0x%08X,%d)\n", iface, avis, fccType, lParam); if (avis == NULL || lParam < 0) return AVIERR_BADPARAM; nStream = AVIFILE_SearchStream(This, fccType, lParam); /* Does the requested stream exist? */ if (nStream < This->fInfo.dwStreams && This->ppStreams[nStream] != NULL) { *avis = &This->ppStreams[nStream]->IAVIStream_iface; IAVIStream_AddRef(*avis); return AVIERR_OK; } /* Sorry, but the specified stream doesn't exist */ *avis = NULL; return AVIERR_NODATA; } static HRESULT WINAPI IAVIFile_fnCreateStream(IAVIFile *iface, IAVIStream **avis, AVISTREAMINFOW *asi) { IAVIFileImpl *This = impl_from_IAVIFile(iface); DWORD n; TRACE("(%p,%p,%p)\n", iface, avis, asi); /* check parameters */ if (avis == NULL || asi == NULL) return AVIERR_BADPARAM; *avis = NULL; /* Does the user have write permission? */ if ((This->uMode & MMIO_RWMODE) == 0) return AVIERR_READONLY; /* Can we add another stream? */ n = This->fInfo.dwStreams; if (n >= MAX_AVISTREAMS || This->dwMoviChunkPos != 0) { /* already reached max nr of streams * or have already written frames to disk */ return AVIERR_UNSUPPORTED; } /* check AVISTREAMINFO for some really needed things */ if (asi->fccType == 0 || asi->dwScale == 0 || asi->dwRate == 0) return AVIERR_BADFORMAT; /* now it seems to be save to add the stream */ assert(This->ppStreams[n] == NULL); This->ppStreams[n] = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IAVIStreamImpl)); if (This->ppStreams[n] == NULL) return AVIERR_MEMORY; /* initialize the new allocated stream */ AVIFILE_ConstructAVIStream(This, n, asi); This->fInfo.dwStreams++; This->fDirty = TRUE; /* update our AVIFILEINFO structure */ AVIFILE_UpdateInfo(This); /* return it */ *avis = &This->ppStreams[n]->IAVIStream_iface; IAVIStream_AddRef(*avis); return AVIERR_OK; } static HRESULT WINAPI IAVIFile_fnWriteData(IAVIFile *iface, DWORD ckid, void *lpData, LONG size) { IAVIFileImpl *This = impl_from_IAVIFile(iface); TRACE("(%p,0x%08X,%p,%d)\n", iface, ckid, lpData, size); /* check parameters */ if (lpData == NULL) return AVIERR_BADPARAM; if (size < 0) return AVIERR_BADSIZE; /* Do we have write permission? */ if ((This->uMode & MMIO_RWMODE) == 0) return AVIERR_READONLY; This->fDirty = TRUE; return WriteExtraChunk(&This->fileextra, ckid, lpData, size); } static HRESULT WINAPI IAVIFile_fnReadData(IAVIFile *iface, DWORD ckid, void *lpData, LONG *size) { IAVIFileImpl *This = impl_from_IAVIFile(iface); TRACE("(%p,0x%08X,%p,%p)\n", iface, ckid, lpData, size); return ReadExtraChunk(&This->fileextra, ckid, lpData, size); } static HRESULT WINAPI IAVIFile_fnEndRecord(IAVIFile *iface) { IAVIFileImpl *This = impl_from_IAVIFile(iface); TRACE("(%p)\n",iface); if ((This->uMode & MMIO_RWMODE) == 0) return AVIERR_READONLY; This->fDirty = TRUE; /* no frames written to any stream? -- compute start of 'movi'-chunk */ if (This->dwMoviChunkPos == 0) AVIFILE_ComputeMoviStart(This); This->fInfo.dwFlags |= AVIFILEINFO_ISINTERLEAVED; /* already written frames to any stream, ... */ if (This->ckLastRecord.dwFlags & MMIO_DIRTY) { /* close last record */ if (mmioAscend(This->hmmio, &This->ckLastRecord, 0) != 0) return AVIERR_FILEWRITE; AVIFILE_AddRecord(This); if (This->fInfo.dwSuggestedBufferSize < This->ckLastRecord.cksize + 3 * sizeof(DWORD)) This->fInfo.dwSuggestedBufferSize = This->ckLastRecord.cksize + 3 * sizeof(DWORD); } /* write out a new record into file, but don't close it */ This->ckLastRecord.cksize = 0; This->ckLastRecord.fccType = listtypeAVIRECORD; if (mmioSeek(This->hmmio, This->dwNextFramePos, SEEK_SET) == -1) return AVIERR_FILEWRITE; if (mmioCreateChunk(This->hmmio, &This->ckLastRecord, MMIO_CREATELIST) != 0) return AVIERR_FILEWRITE; This->dwNextFramePos += 3 * sizeof(DWORD); return AVIERR_OK; } static HRESULT WINAPI IAVIFile_fnDeleteStream(IAVIFile *iface, DWORD fccType, LONG lParam) { IAVIFileImpl *This = impl_from_IAVIFile(iface); ULONG nStream; TRACE("(%p,0x%08X,%d)\n", iface, fccType, lParam); /* check parameter */ if (lParam < 0) return AVIERR_BADPARAM; /* Have user write permissions? */ if ((This->uMode & MMIO_RWMODE) == 0) return AVIERR_READONLY; nStream = AVIFILE_SearchStream(This, fccType, lParam); /* Does the requested stream exist? */ if (nStream < This->fInfo.dwStreams && This->ppStreams[nStream] != NULL) { /* ... so delete it now */ HeapFree(GetProcessHeap(), 0, This->ppStreams[nStream]); This->fInfo.dwStreams--; if (nStream < This->fInfo.dwStreams) memmove(&This->ppStreams[nStream], &This->ppStreams[nStream + 1], (This->fInfo.dwStreams - nStream) * sizeof(This->ppStreams[0])); This->ppStreams[This->fInfo.dwStreams] = NULL; This->fDirty = TRUE; /* This->fInfo will be updated further when asked for */ return AVIERR_OK; } else return AVIERR_NODATA; } static const struct IAVIFileVtbl avif_vt = { IAVIFile_fnQueryInterface, IAVIFile_fnAddRef, IAVIFile_fnRelease, IAVIFile_fnInfo, IAVIFile_fnGetStream, IAVIFile_fnCreateStream, IAVIFile_fnWriteData, IAVIFile_fnReadData, IAVIFile_fnEndRecord, IAVIFile_fnDeleteStream }; static HRESULT WINAPI IPersistFile_fnQueryInterface(IPersistFile *iface, REFIID riid, void **ppv) { IAVIFileImpl *This = impl_from_IPersistFile(iface); return IUnknown_QueryInterface(This->outer_unk, riid, ppv); } static ULONG WINAPI IPersistFile_fnAddRef(IPersistFile *iface) { IAVIFileImpl *This = impl_from_IPersistFile(iface); return IUnknown_AddRef(This->outer_unk); } static ULONG WINAPI IPersistFile_fnRelease(IPersistFile *iface) { IAVIFileImpl *This = impl_from_IPersistFile(iface); return IUnknown_Release(This->outer_unk); } static HRESULT WINAPI IPersistFile_fnGetClassID(IPersistFile *iface, LPCLSID pClassID) { TRACE("(%p,%p)\n", iface, pClassID); if (pClassID == NULL) return AVIERR_BADPARAM; *pClassID = CLSID_AVIFile; return AVIERR_OK; } static HRESULT WINAPI IPersistFile_fnIsDirty(IPersistFile *iface) { IAVIFileImpl *This = impl_from_IPersistFile(iface); TRACE("(%p)\n", iface); return (This->fDirty ? S_OK : S_FALSE); } static HRESULT WINAPI IPersistFile_fnLoad(IPersistFile *iface, LPCOLESTR pszFileName, DWORD dwMode) { IAVIFileImpl *This = impl_from_IPersistFile(iface); int len; TRACE("(%p,%s,0x%08X)\n", iface, debugstr_w(pszFileName), dwMode); /* check parameter */ if (pszFileName == NULL) return AVIERR_BADPARAM; if (This->hmmio != NULL) return AVIERR_ERROR; /* No reuse of this object for another file! */ /* remember mode and name */ This->uMode = dwMode; len = lstrlenW(pszFileName) + 1; This->szFileName = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR)); if (This->szFileName == NULL) return AVIERR_MEMORY; lstrcpyW(This->szFileName, pszFileName); /* try to open the file */ This->hmmio = mmioOpenW(This->szFileName, NULL, MMIO_ALLOCBUF | dwMode); if (This->hmmio == NULL) { /* mmioOpenW not in native DLLs of Win9x -- try mmioOpenA */ LPSTR szFileName; len = WideCharToMultiByte(CP_ACP, 0, This->szFileName, -1, NULL, 0, NULL, NULL); szFileName = HeapAlloc(GetProcessHeap(), 0, len * sizeof(CHAR)); if (szFileName == NULL) return AVIERR_MEMORY; WideCharToMultiByte(CP_ACP, 0, This->szFileName, -1, szFileName, len, NULL, NULL); This->hmmio = mmioOpenA(szFileName, NULL, MMIO_ALLOCBUF | dwMode); HeapFree(GetProcessHeap(), 0, szFileName); if (This->hmmio == NULL) return AVIERR_FILEOPEN; } /* should we create a new file? */ if (dwMode & OF_CREATE) { memset(& This->fInfo, 0, sizeof(This->fInfo)); This->fInfo.dwFlags = AVIFILEINFO_HASINDEX | AVIFILEINFO_TRUSTCKTYPE; return AVIERR_OK; } else return AVIFILE_LoadFile(This); } static HRESULT WINAPI IPersistFile_fnSave(IPersistFile *iface, LPCOLESTR pszFileName, BOOL fRemember) { TRACE("(%p,%s,%d)\n", iface, debugstr_w(pszFileName), fRemember); /* We write directly to disk, so nothing to do. */ return AVIERR_OK; } static HRESULT WINAPI IPersistFile_fnSaveCompleted(IPersistFile *iface, LPCOLESTR pszFileName) { TRACE("(%p,%s)\n", iface, debugstr_w(pszFileName)); /* We write directly to disk, so nothing to do. */ return AVIERR_OK; } static HRESULT WINAPI IPersistFile_fnGetCurFile(IPersistFile *iface, LPOLESTR *ppszFileName) { IAVIFileImpl *This = impl_from_IPersistFile(iface); TRACE("(%p,%p)\n", iface, ppszFileName); if (ppszFileName == NULL) return AVIERR_BADPARAM; *ppszFileName = NULL; if (This->szFileName != NULL) { int len = lstrlenW(This->szFileName) + 1; *ppszFileName = CoTaskMemAlloc(len * sizeof(WCHAR)); if (*ppszFileName == NULL) return AVIERR_MEMORY; lstrcpyW(*ppszFileName, This->szFileName); } return AVIERR_OK; } static const struct IPersistFileVtbl pf_vt = { IPersistFile_fnQueryInterface, IPersistFile_fnAddRef, IPersistFile_fnRelease, IPersistFile_fnGetClassID, IPersistFile_fnIsDirty, IPersistFile_fnLoad, IPersistFile_fnSave, IPersistFile_fnSaveCompleted, IPersistFile_fnGetCurFile }; HRESULT AVIFILE_CreateAVIFile(IUnknown *pUnkOuter, REFIID riid, void **ppv) { IAVIFileImpl *obj; HRESULT hr; *ppv = NULL; obj = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IAVIFileImpl)); if (!obj) return AVIERR_MEMORY; obj->IUnknown_inner.lpVtbl = &unk_vtbl; obj->IAVIFile_iface.lpVtbl = &avif_vt; obj->IPersistFile_iface.lpVtbl = &pf_vt; obj->ref = 1; if (pUnkOuter) obj->outer_unk = pUnkOuter; else obj->outer_unk = &obj->IUnknown_inner; hr = IUnknown_QueryInterface(&obj->IUnknown_inner, riid, ppv); IUnknown_Release(&obj->IUnknown_inner); return hr; } static HRESULT WINAPI IAVIStream_fnQueryInterface(IAVIStream *iface, REFIID riid, void **ppv) { IAVIStreamImpl *This = impl_from_IAVIStream(iface); TRACE("(%p,%s,%p)\n", This, debugstr_guid(riid), ppv); if (!ppv) { WARN("invalid parameter\n"); return E_INVALIDARG; } *ppv = NULL; if (IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IAVIStream, riid)) { *ppv = iface; IAVIStream_AddRef(iface); return S_OK; } /* FIXME: IAVIStreaming interface */ return E_NOINTERFACE; } static ULONG WINAPI IAVIStream_fnAddRef(IAVIStream *iface) { IAVIStreamImpl *This = impl_from_IAVIStream(iface); ULONG ref = InterlockedIncrement(&This->ref); TRACE("(%p) ref=%d\n", This, ref); /* also add ref to parent, so that it doesn't kill us */ if (This->paf != NULL) IAVIFile_AddRef(&This->paf->IAVIFile_iface); return ref; } static ULONG WINAPI IAVIStream_fnRelease(IAVIStream *iface) { IAVIStreamImpl *This = impl_from_IAVIStream(iface); ULONG ref = InterlockedDecrement(&This->ref); TRACE("(%p) ref=%d\n", This, ref); if (This->paf != NULL) IAVIFile_Release(&This->paf->IAVIFile_iface); return ref; } static HRESULT WINAPI IAVIStream_fnCreate(IAVIStream *iface, LPARAM lParam1, LPARAM lParam2) { TRACE("(%p,0x%08lX,0x%08lX)\n", iface, lParam1, lParam2); /* This IAVIStream interface needs an AVIFile */ return AVIERR_UNSUPPORTED; } static HRESULT WINAPI IAVIStream_fnInfo(IAVIStream *iface, AVISTREAMINFOW *psi, LONG size) { IAVIStreamImpl *This = impl_from_IAVIStream(iface); TRACE("(%p,%p,%d)\n", iface, psi, size); if (psi == NULL) return AVIERR_BADPARAM; if (size < 0) return AVIERR_BADSIZE; memcpy(psi, &This->sInfo, min((DWORD)size, sizeof(This->sInfo))); if ((DWORD)size < sizeof(This->sInfo)) return AVIERR_BUFFERTOOSMALL; return AVIERR_OK; } static LONG WINAPI IAVIStream_fnFindSample(IAVIStream *iface, LONG pos, LONG flags) { IAVIStreamImpl *This = impl_from_IAVIStream(iface); LONG offset = 0; TRACE("(%p,%d,0x%08X)\n",iface,pos,flags); if (flags & FIND_FROM_START) { pos = This->sInfo.dwStart; flags &= ~(FIND_FROM_START|FIND_PREV); flags |= FIND_NEXT; } if (This->sInfo.dwSampleSize != 0) { /* convert samples into block number with offset */ AVIFILE_SamplesToBlock(This, &pos, &offset); } if (flags & FIND_TYPE) { if (flags & FIND_KEY) { while (0 <= pos && pos <= This->lLastFrame) { if (This->idxFrames[pos].dwFlags & AVIIF_KEYFRAME) goto RETURN_FOUND; if (flags & FIND_NEXT) pos++; else pos--; }; } else if (flags & FIND_ANY) { while (0 <= pos && pos <= This->lLastFrame) { if (This->idxFrames[pos].dwChunkLength > 0) goto RETURN_FOUND; if (flags & FIND_NEXT) pos++; else pos--; }; } else if ((flags & FIND_FORMAT) && This->idxFmtChanges != NULL && This->sInfo.fccType == streamtypeVIDEO) { if (flags & FIND_NEXT) { ULONG n; for (n = 0; n < This->sInfo.dwFormatChangeCount; n++) if (This->idxFmtChanges[n].ckid >= pos) { pos = This->idxFmtChanges[n].ckid; goto RETURN_FOUND; } } else { LONG n; for (n = (LONG)This->sInfo.dwFormatChangeCount; n >= 0; n--) { if (This->idxFmtChanges[n].ckid <= pos) { pos = This->idxFmtChanges[n].ckid; goto RETURN_FOUND; } } if (pos > (LONG)This->sInfo.dwStart) return 0; /* format changes always for first frame */ } } return -1; } RETURN_FOUND: if (pos < (LONG)This->sInfo.dwStart) return -1; switch (flags & FIND_RET) { case FIND_LENGTH: /* physical size */ pos = This->idxFrames[pos].dwChunkLength; break; case FIND_OFFSET: /* physical position */ pos = This->idxFrames[pos].dwChunkOffset + 2 * sizeof(DWORD) + offset * This->sInfo.dwSampleSize; break; case FIND_SIZE: /* logical size */ if (This->sInfo.dwSampleSize) pos = This->sInfo.dwSampleSize; else pos = 1; break; case FIND_INDEX: FIXME(": FIND_INDEX flag is not supported!\n"); /* This is an index in the index-table on disc. */ break; }; /* else logical position */ return pos; } static HRESULT WINAPI IAVIStream_fnReadFormat(IAVIStream *iface, LONG pos, void *format, LONG *formatsize) { IAVIStreamImpl *This = impl_from_IAVIStream(iface); TRACE("(%p,%d,%p,%p)\n", iface, pos, format, formatsize); if (formatsize == NULL) return AVIERR_BADPARAM; /* only interested in needed buffersize? */ if (format == NULL || *formatsize <= 0) { *formatsize = This->cbFormat; return AVIERR_OK; } /* copy initial format (only as much as will fit) */ memcpy(format, This->lpFormat, min(*(DWORD*)formatsize, This->cbFormat)); if (*(DWORD*)formatsize < This->cbFormat) { *formatsize = This->cbFormat; return AVIERR_BUFFERTOOSMALL; } /* Could format change? When yes will it change? */ if ((This->sInfo.dwFlags & AVISTREAMINFO_FORMATCHANGES) && pos > This->sInfo.dwStart) { LONG lLastFmt; lLastFmt = IAVIStream_fnFindSample(iface, pos, FIND_FORMAT|FIND_PREV); if (lLastFmt > 0) { FIXME(": need to read formatchange for %d -- unimplemented!\n",lLastFmt); } } *formatsize = This->cbFormat; return AVIERR_OK; } static HRESULT WINAPI IAVIStream_fnSetFormat(IAVIStream *iface, LONG pos, void *format, LONG formatsize) { IAVIStreamImpl *This = impl_from_IAVIStream(iface); BITMAPINFOHEADER *lpbiNew = format; TRACE("(%p,%d,%p,%d)\n", iface, pos, format, formatsize); /* check parameters */ if (format == NULL || formatsize <= 0) return AVIERR_BADPARAM; /* Do we have write permission? */ if ((This->paf->uMode & MMIO_RWMODE) == 0) return AVIERR_READONLY; /* can only set format before frame is written! */ if (This->lLastFrame > pos) return AVIERR_UNSUPPORTED; /* initial format or a formatchange? */ if (This->lpFormat == NULL) { /* initial format */ if (This->paf->dwMoviChunkPos != 0) return AVIERR_ERROR; /* user has used API in wrong sequence! */ This->lpFormat = HeapAlloc(GetProcessHeap(), 0, formatsize); if (This->lpFormat == NULL) return AVIERR_MEMORY; This->cbFormat = formatsize; memcpy(This->lpFormat, format, formatsize); /* update some infos about stream */ if (This->sInfo.fccType == streamtypeVIDEO) { LONG lDim; lDim = This->sInfo.rcFrame.right - This->sInfo.rcFrame.left; if (lDim < lpbiNew->biWidth) This->sInfo.rcFrame.right = This->sInfo.rcFrame.left + lpbiNew->biWidth; lDim = This->sInfo.rcFrame.bottom - This->sInfo.rcFrame.top; if (lDim < lpbiNew->biHeight) This->sInfo.rcFrame.bottom = This->sInfo.rcFrame.top + lpbiNew->biHeight; } else if (This->sInfo.fccType == streamtypeAUDIO) This->sInfo.dwSampleSize = ((LPWAVEFORMATEX)This->lpFormat)->nBlockAlign; return AVIERR_OK; } else { MMCKINFO ck; LPBITMAPINFOHEADER lpbiOld = This->lpFormat; RGBQUAD *rgbNew = (RGBQUAD*)((LPBYTE)lpbiNew + lpbiNew->biSize); AVIPALCHANGE *lppc = NULL; UINT n; /* perhaps format change, check it ... */ if (This->cbFormat != formatsize) return AVIERR_UNSUPPORTED; /* no format change, only the initial one */ if (memcmp(This->lpFormat, format, formatsize) == 0) return AVIERR_OK; /* check that's only the palette, which changes */ if (lpbiOld->biSize != lpbiNew->biSize || lpbiOld->biWidth != lpbiNew->biWidth || lpbiOld->biHeight != lpbiNew->biHeight || lpbiOld->biPlanes != lpbiNew->biPlanes || lpbiOld->biBitCount != lpbiNew->biBitCount || lpbiOld->biCompression != lpbiNew->biCompression || lpbiOld->biClrUsed != lpbiNew->biClrUsed) return AVIERR_UNSUPPORTED; This->sInfo.dwFlags |= AVISTREAMINFO_FORMATCHANGES; /* simply say all colors have changed */ ck.ckid = MAKEAVICKID(cktypePALchange, This->nStream); ck.cksize = 2 * sizeof(WORD) + lpbiOld->biClrUsed * sizeof(PALETTEENTRY); lppc = HeapAlloc(GetProcessHeap(), 0, ck.cksize); if (lppc == NULL) return AVIERR_MEMORY; lppc->bFirstEntry = 0; lppc->bNumEntries = (lpbiOld->biClrUsed < 256 ? lpbiOld->biClrUsed : 0); lppc->wFlags = 0; for (n = 0; n < lpbiOld->biClrUsed; n++) { lppc->peNew[n].peRed = rgbNew[n].rgbRed; lppc->peNew[n].peGreen = rgbNew[n].rgbGreen; lppc->peNew[n].peBlue = rgbNew[n].rgbBlue; lppc->peNew[n].peFlags = 0; } if (mmioSeek(This->paf->hmmio, This->paf->dwNextFramePos, SEEK_SET) == -1 || mmioCreateChunk(This->paf->hmmio, &ck, 0) != S_OK || mmioWrite(This->paf->hmmio, (HPSTR)lppc, ck.cksize) != ck.cksize || mmioAscend(This->paf->hmmio, &ck, 0) != S_OK) { HeapFree(GetProcessHeap(), 0, lppc); return AVIERR_FILEWRITE; } This->paf->dwNextFramePos += ck.cksize + 2 * sizeof(DWORD); HeapFree(GetProcessHeap(), 0, lppc); return AVIFILE_AddFrame(This, cktypePALchange, n, ck.dwDataOffset, 0); } } static HRESULT WINAPI IAVIStream_fnRead(IAVIStream *iface, LONG start, LONG samples, void *buffer, LONG buffersize, LONG *bytesread, LONG *samplesread) { IAVIStreamImpl *This = impl_from_IAVIStream(iface); DWORD size; HRESULT hr; TRACE("(%p,%d,%d,%p,%d,%p,%p)\n", iface, start, samples, buffer, buffersize, bytesread, samplesread); /* clear return parameters if given */ if (bytesread != NULL) *bytesread = 0; if (samplesread != NULL) *samplesread = 0; /* check parameters */ if ((LONG)This->sInfo.dwStart > start) return AVIERR_NODATA; /* couldn't read before start of stream */ if (This->sInfo.dwStart + This->sInfo.dwLength < (DWORD)start) return AVIERR_NODATA; /* start is past end of stream */ /* should we read as much as possible? */ if (samples == -1) { /* User should know how much we have read */ if (bytesread == NULL && samplesread == NULL) return AVIERR_BADPARAM; if (This->sInfo.dwSampleSize != 0) samples = buffersize / This->sInfo.dwSampleSize; else samples = 1; } /* limit to end of stream */ if ((LONG)This->sInfo.dwLength < samples) samples = This->sInfo.dwLength; if ((start - This->sInfo.dwStart) > (This->sInfo.dwLength - samples)) samples = This->sInfo.dwLength - (start - This->sInfo.dwStart); /* nothing to read? Then leave ... */ if (samples == 0) return AVIERR_OK; if (This->sInfo.dwSampleSize != 0) { /* fixed samplesize -- we can read over frame/block boundaries */ LONG block = start; LONG offset = 0; if (!buffer) { if (bytesread) *bytesread = samples*This->sInfo.dwSampleSize; if (samplesread) *samplesread = samples; return AVIERR_OK; } /* convert start sample to block,offset pair */ AVIFILE_SamplesToBlock(This, &block, &offset); /* convert samples to bytes */ samples *= This->sInfo.dwSampleSize; while (samples > 0 && buffersize > 0) { LONG blocksize; if (block != This->dwCurrentFrame) { hr = AVIFILE_ReadBlock(This, block, NULL, 0); if (FAILED(hr)) return hr; } size = min((DWORD)samples, (DWORD)buffersize); blocksize = This->lpBuffer[1]; TRACE("blocksize = %u\n",blocksize); size = min(size, blocksize - offset); memcpy(buffer, ((BYTE*)&This->lpBuffer[2]) + offset, size); block++; offset = 0; buffer = ((LPBYTE)buffer)+size; samples -= size; buffersize -= size; /* fill out return parameters if given */ if (bytesread != NULL) *bytesread += size; if (samplesread != NULL) *samplesread += size / This->sInfo.dwSampleSize; } if (samples == 0) return AVIERR_OK; else return AVIERR_BUFFERTOOSMALL; } else { /* variable samplesize -- we can only read one full frame/block */ if (samples > 1) samples = 1; assert(start <= This->lLastFrame); size = This->idxFrames[start].dwChunkLength; if (buffer != NULL && buffersize >= size) { hr = AVIFILE_ReadBlock(This, start, buffer, size); if (FAILED(hr)) return hr; } else if (buffer != NULL) return AVIERR_BUFFERTOOSMALL; /* fill out return parameters if given */ if (bytesread != NULL) *bytesread = size; if (samplesread != NULL) *samplesread = samples; return AVIERR_OK; } } static HRESULT WINAPI IAVIStream_fnWrite(IAVIStream *iface, LONG start, LONG samples, void *buffer, LONG buffersize, DWORD flags, LONG *sampwritten, LONG *byteswritten) { IAVIStreamImpl *This = impl_from_IAVIStream(iface); FOURCC ckid; HRESULT hr; TRACE("(%p,%d,%d,%p,%d,0x%08X,%p,%p)\n", iface, start, samples, buffer, buffersize, flags, sampwritten, byteswritten); /* clear return parameters if given */ if (sampwritten != NULL) *sampwritten = 0; if (byteswritten != NULL) *byteswritten = 0; /* check parameters */ if (buffer == NULL && (buffersize > 0 || samples > 0)) return AVIERR_BADPARAM; /* Have we write permission? */ if ((This->paf->uMode & MMIO_RWMODE) == 0) return AVIERR_READONLY; switch (This->sInfo.fccType) { case streamtypeAUDIO: ckid = MAKEAVICKID(cktypeWAVEbytes, This->nStream); break; default: if ((flags & AVIIF_KEYFRAME) && buffersize != 0) ckid = MAKEAVICKID(cktypeDIBbits, This->nStream); else ckid = MAKEAVICKID(cktypeDIBcompressed, This->nStream); break; }; /* append to end of stream? */ if (start == -1) { if (This->lLastFrame == -1) start = This->sInfo.dwStart; else start = This->sInfo.dwLength; } else if (This->lLastFrame == -1) This->sInfo.dwStart = start; if (This->sInfo.dwSampleSize != 0) { /* fixed sample size -- audio like */ if (samples * This->sInfo.dwSampleSize != buffersize) return AVIERR_BADPARAM; /* Couldn't skip audio-like data -- User must supply appropriate silence */ if (This->sInfo.dwLength != start) return AVIERR_UNSUPPORTED; /* Convert position to frame/block */ start = This->lLastFrame + 1; if ((This->paf->fInfo.dwFlags & AVIFILEINFO_ISINTERLEAVED) == 0) { FIXME(": not interleaved, could collect audio data!\n"); } } else { /* variable sample size -- video like */ if (samples > 1) return AVIERR_UNSUPPORTED; /* must we fill up with empty frames? */ if (This->lLastFrame != -1) { FOURCC ckid2 = MAKEAVICKID(cktypeDIBcompressed, This->nStream); while (start > This->lLastFrame + 1) { hr = AVIFILE_WriteBlock(This, This->lLastFrame + 1, ckid2, 0, NULL, 0); if (FAILED(hr)) return hr; } } } /* write the block now */ hr = AVIFILE_WriteBlock(This, start, ckid, flags, buffer, buffersize); if (SUCCEEDED(hr)) { /* fill out return parameters if given */ if (sampwritten != NULL) *sampwritten = samples; if (byteswritten != NULL) *byteswritten = buffersize; } return hr; } static HRESULT WINAPI IAVIStream_fnDelete(IAVIStream *iface, LONG start, LONG samples) { IAVIStreamImpl *This = impl_from_IAVIStream(iface); FIXME("(%p,%d,%d): stub\n", iface, start, samples); /* check parameters */ if (start < 0 || samples < 0) return AVIERR_BADPARAM; /* Delete before start of stream? */ if (start + samples < This->sInfo.dwStart) return AVIERR_OK; /* Delete after end of stream? */ if (start > This->sInfo.dwLength) return AVIERR_OK; /* For the rest we need write permissions */ if ((This->paf->uMode & MMIO_RWMODE) == 0) return AVIERR_READONLY; /* 1. overwrite the data with JUNK * * if ISINTERLEAVED { * 2. concat all neighboured JUNK-blocks in this record to one * 3. if this record only contains JUNK and is at end set dwNextFramePos * to start of this record, repeat this. * } else { * 2. concat all neighboured JUNK-blocks. * 3. if the JUNK block is at the end, then set dwNextFramePos to * start of this block. * } */ return AVIERR_UNSUPPORTED; } static HRESULT WINAPI IAVIStream_fnReadData(IAVIStream *iface, DWORD fcc, void *lp, LONG *lpread) { IAVIStreamImpl *This = impl_from_IAVIStream(iface); TRACE("(%p,0x%08X,%p,%p)\n", iface, fcc, lp, lpread); if (fcc == ckidSTREAMHANDLERDATA) { if (This->lpHandlerData != NULL && This->cbHandlerData > 0) { if (lp == NULL || *lpread <= 0) { *lpread = This->cbHandlerData; return AVIERR_OK; } memcpy(lp, This->lpHandlerData, min(This->cbHandlerData, *lpread)); if (*lpread < This->cbHandlerData) return AVIERR_BUFFERTOOSMALL; return AVIERR_OK; } else return AVIERR_NODATA; } else return ReadExtraChunk(&This->extra, fcc, lp, lpread); } static HRESULT WINAPI IAVIStream_fnWriteData(IAVIStream *iface, DWORD fcc, void *lp, LONG size) { IAVIStreamImpl *This = impl_from_IAVIStream(iface); TRACE("(%p,0x%08x,%p,%d)\n", iface, fcc, lp, size); /* check parameters */ if (lp == NULL) return AVIERR_BADPARAM; if (size <= 0) return AVIERR_BADSIZE; /* need write permission */ if ((This->paf->uMode & MMIO_RWMODE) == 0) return AVIERR_READONLY; /* already written something to this file? */ if (This->paf->dwMoviChunkPos != 0) { /* the data will be inserted before the 'movi' chunk, so check for * enough space */ DWORD dwPos = AVIFILE_ComputeMoviStart(This->paf); /* ckid,size => 2 * sizeof(DWORD) */ dwPos += 2 * sizeof(DWORD) + size; if (dwPos >= This->paf->dwMoviChunkPos - 2 * sizeof(DWORD)) return AVIERR_UNSUPPORTED; /* not enough space left */ } This->paf->fDirty = TRUE; if (fcc == ckidSTREAMHANDLERDATA) { if (This->lpHandlerData != NULL) { FIXME(": handler data already set -- overwrite?\n"); return AVIERR_UNSUPPORTED; } This->lpHandlerData = HeapAlloc(GetProcessHeap(), 0, size); if (This->lpHandlerData == NULL) return AVIERR_MEMORY; This->cbHandlerData = size; memcpy(This->lpHandlerData, lp, size); return AVIERR_OK; } else return WriteExtraChunk(&This->extra, fcc, lp, size); } static HRESULT WINAPI IAVIStream_fnSetInfo(IAVIStream *iface, AVISTREAMINFOW *info, LONG infolen) { FIXME("(%p,%p,%d): stub\n", iface, info, infolen); return E_FAIL; } static const struct IAVIStreamVtbl avist_vt = { IAVIStream_fnQueryInterface, IAVIStream_fnAddRef, IAVIStream_fnRelease, IAVIStream_fnCreate, IAVIStream_fnInfo, IAVIStream_fnFindSample, IAVIStream_fnReadFormat, IAVIStream_fnSetFormat, IAVIStream_fnRead, IAVIStream_fnWrite, IAVIStream_fnDelete, IAVIStream_fnReadData, IAVIStream_fnWriteData, IAVIStream_fnSetInfo }; static HRESULT AVIFILE_AddFrame(IAVIStreamImpl *This, DWORD ckid, DWORD size, DWORD offset, DWORD flags) { UINT n; /* pre-conditions */ assert(This != NULL); switch (TWOCCFromFOURCC(ckid)) { case cktypeDIBbits: if (This->paf->fInfo.dwFlags & AVIFILEINFO_TRUSTCKTYPE) flags |= AVIIF_KEYFRAME; break; case cktypeDIBcompressed: if (This->paf->fInfo.dwFlags & AVIFILEINFO_TRUSTCKTYPE) flags &= ~AVIIF_KEYFRAME; break; case cktypePALchange: if (This->sInfo.fccType != streamtypeVIDEO) { ERR(": found palette change in non-video stream!\n"); return AVIERR_BADFORMAT; } if (This->idxFmtChanges == NULL || This->nIdxFmtChanges <= This->sInfo.dwFormatChangeCount) { DWORD new_count = This->nIdxFmtChanges + 16; void *new_buffer; if (This->idxFmtChanges == NULL) { This->idxFmtChanges = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, new_count * sizeof(AVIINDEXENTRY)); if (!This->idxFmtChanges) return AVIERR_MEMORY; } else { new_buffer = HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, This->idxFmtChanges, new_count * sizeof(AVIINDEXENTRY)); if (!new_buffer) return AVIERR_MEMORY; This->idxFmtChanges = new_buffer; } This->nIdxFmtChanges = new_count; } This->sInfo.dwFlags |= AVISTREAMINFO_FORMATCHANGES; n = ++This->sInfo.dwFormatChangeCount; This->idxFmtChanges[n].ckid = This->lLastFrame; This->idxFmtChanges[n].dwFlags = 0; This->idxFmtChanges[n].dwChunkOffset = offset; This->idxFmtChanges[n].dwChunkLength = size; return AVIERR_OK; case cktypeWAVEbytes: if (This->paf->fInfo.dwFlags & AVIFILEINFO_TRUSTCKTYPE) flags |= AVIIF_KEYFRAME; break; default: WARN(": unknown TWOCC 0x%04X found\n", TWOCCFromFOURCC(ckid)); break; }; /* first frame is always a keyframe */ if (This->lLastFrame == -1) flags |= AVIIF_KEYFRAME; if (This->sInfo.dwSuggestedBufferSize < size) This->sInfo.dwSuggestedBufferSize = size; /* get memory for index */ if (This->idxFrames == NULL || This->lLastFrame + 1 >= This->nIdxFrames) { This->nIdxFrames += 512; if (This->idxFrames == NULL) This->idxFrames = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, This->nIdxFrames * sizeof(AVIINDEXENTRY)); else This->idxFrames = HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, This->idxFrames, This->nIdxFrames * sizeof(AVIINDEXENTRY)); if (This->idxFrames == NULL) return AVIERR_MEMORY; } This->lLastFrame++; This->idxFrames[This->lLastFrame].ckid = ckid; This->idxFrames[This->lLastFrame].dwFlags = flags; This->idxFrames[This->lLastFrame].dwChunkOffset = offset; This->idxFrames[This->lLastFrame].dwChunkLength = size; /* update AVISTREAMINFO structure if necessary */ if (This->sInfo.dwLength <= This->lLastFrame) This->sInfo.dwLength = This->lLastFrame + 1; return AVIERR_OK; } static HRESULT AVIFILE_AddRecord(IAVIFileImpl *This) { /* pre-conditions */ assert(This != NULL && This->ppStreams[0] != NULL); if (This->idxRecords == NULL || This->cbIdxRecords / sizeof(AVIINDEXENTRY) <= This->nIdxRecords) { DWORD new_count = This->cbIdxRecords + 1024 * sizeof(AVIINDEXENTRY); void *mem; if (!This->idxRecords) mem = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, new_count); else mem = HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, This->idxRecords, new_count); if (mem) { This->cbIdxRecords = new_count; This->idxRecords = mem; } else { HeapFree(GetProcessHeap(), 0, This->idxRecords); This->idxRecords = NULL; return AVIERR_MEMORY; } } assert(This->nIdxRecords < This->cbIdxRecords/sizeof(AVIINDEXENTRY)); This->idxRecords[This->nIdxRecords].ckid = listtypeAVIRECORD; This->idxRecords[This->nIdxRecords].dwFlags = AVIIF_LIST; This->idxRecords[This->nIdxRecords].dwChunkOffset = This->ckLastRecord.dwDataOffset - 2 * sizeof(DWORD); This->idxRecords[This->nIdxRecords].dwChunkLength = This->ckLastRecord.cksize; This->nIdxRecords++; return AVIERR_OK; } static DWORD AVIFILE_ComputeMoviStart(IAVIFileImpl *This) { DWORD dwPos; DWORD nStream; /* RIFF,hdrl,movi,avih => (3 * 3 + 2) * sizeof(DWORD) = 11 * sizeof(DWORD) */ dwPos = 11 * sizeof(DWORD) + sizeof(MainAVIHeader); for (nStream = 0; nStream < This->fInfo.dwStreams; nStream++) { IAVIStreamImpl *pStream = This->ppStreams[nStream]; /* strl,strh,strf => (3 + 2 * 2) * sizeof(DWORD) = 7 * sizeof(DWORD) */ dwPos += 7 * sizeof(DWORD) + sizeof(AVIStreamHeader); dwPos += ((pStream->cbFormat + 1) & ~1U); if (pStream->lpHandlerData != NULL && pStream->cbHandlerData > 0) dwPos += 2 * sizeof(DWORD) + ((pStream->cbHandlerData + 1) & ~1U); if (pStream->sInfo.szName[0]) dwPos += 2 * sizeof(DWORD) + ((lstrlenW(pStream->sInfo.szName) + 1) & ~1U); } if (This->dwMoviChunkPos == 0) { This->dwNextFramePos = dwPos; /* pad to multiple of AVI_HEADERSIZE only if we are more than 8 bytes away from it */ if (((dwPos + AVI_HEADERSIZE) & ~(AVI_HEADERSIZE - 1)) - dwPos > 2 * sizeof(DWORD)) This->dwNextFramePos = (dwPos + AVI_HEADERSIZE) & ~(AVI_HEADERSIZE - 1); This->dwMoviChunkPos = This->dwNextFramePos - sizeof(DWORD); } return dwPos; } static void AVIFILE_ConstructAVIStream(IAVIFileImpl *paf, DWORD nr, const AVISTREAMINFOW *asi) { IAVIStreamImpl *pstream; /* pre-conditions */ assert(paf != NULL); assert(nr < MAX_AVISTREAMS); assert(paf->ppStreams[nr] != NULL); pstream = paf->ppStreams[nr]; pstream->IAVIStream_iface.lpVtbl = &avist_vt; pstream->ref = 0; pstream->paf = paf; pstream->nStream = nr; pstream->dwCurrentFrame = (DWORD)-1; pstream->lLastFrame = -1; if (asi != NULL) { memcpy(&pstream->sInfo, asi, sizeof(pstream->sInfo)); if (asi->dwLength > 0) { /* pre-allocate mem for frame-index structure */ pstream->idxFrames = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, asi->dwLength * sizeof(AVIINDEXENTRY)); if (pstream->idxFrames != NULL) pstream->nIdxFrames = asi->dwLength; } if (asi->dwFormatChangeCount > 0) { /* pre-allocate mem for formatchange-index structure */ pstream->idxFmtChanges = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, asi->dwFormatChangeCount * sizeof(AVIINDEXENTRY)); if (pstream->idxFmtChanges != NULL) pstream->nIdxFmtChanges = asi->dwFormatChangeCount; } /* These values will be computed */ pstream->sInfo.dwLength = 0; pstream->sInfo.dwSuggestedBufferSize = 0; pstream->sInfo.dwFormatChangeCount = 0; pstream->sInfo.dwEditCount = 1; if (pstream->sInfo.dwSampleSize > 0) SetRectEmpty(&pstream->sInfo.rcFrame); } pstream->sInfo.dwCaps = AVIFILECAPS_CANREAD|AVIFILECAPS_CANWRITE; } static void AVIFILE_DestructAVIStream(IAVIStreamImpl *This) { /* pre-conditions */ assert(This != NULL); This->dwCurrentFrame = (DWORD)-1; This->lLastFrame = -1; This->paf = NULL; if (This->idxFrames != NULL) { HeapFree(GetProcessHeap(), 0, This->idxFrames); This->idxFrames = NULL; This->nIdxFrames = 0; } HeapFree(GetProcessHeap(), 0, This->idxFmtChanges); This->idxFmtChanges = NULL; if (This->lpBuffer != NULL) { HeapFree(GetProcessHeap(), 0, This->lpBuffer); This->lpBuffer = NULL; This->cbBuffer = 0; } if (This->lpHandlerData != NULL) { HeapFree(GetProcessHeap(), 0, This->lpHandlerData); This->lpHandlerData = NULL; This->cbHandlerData = 0; } if (This->extra.lp != NULL) { HeapFree(GetProcessHeap(), 0, This->extra.lp); This->extra.lp = NULL; This->extra.cb = 0; } if (This->lpFormat != NULL) { HeapFree(GetProcessHeap(), 0, This->lpFormat); This->lpFormat = NULL; This->cbFormat = 0; } } static HRESULT AVIFILE_LoadFile(IAVIFileImpl *This) { MainAVIHeader MainAVIHdr; MMCKINFO ckRIFF; MMCKINFO ckLIST1; MMCKINFO ckLIST2; MMCKINFO ck; IAVIStreamImpl *pStream; DWORD nStream; HRESULT hr; if (This->hmmio == NULL) return AVIERR_FILEOPEN; /* initialize stream ptr's */ memset(This->ppStreams, 0, sizeof(This->ppStreams)); /* try to get "RIFF" chunk -- must not be at beginning of file! */ ckRIFF.fccType = formtypeAVI; if (mmioDescend(This->hmmio, &ckRIFF, NULL, MMIO_FINDRIFF) != S_OK) { ERR(": not an AVI!\n"); return AVIERR_FILEREAD; } /* get "LIST" "hdrl" */ ckLIST1.fccType = listtypeAVIHEADER; hr = FindChunkAndKeepExtras(&This->fileextra, This->hmmio, &ckLIST1, &ckRIFF, MMIO_FINDLIST); if (FAILED(hr)) return hr; /* get "avih" chunk */ ck.ckid = ckidAVIMAINHDR; hr = FindChunkAndKeepExtras(&This->fileextra, This->hmmio, &ck, &ckLIST1, MMIO_FINDCHUNK); if (FAILED(hr)) return hr; if (ck.cksize != sizeof(MainAVIHdr)) { ERR(": invalid size of %d for MainAVIHeader!\n", ck.cksize); return AVIERR_BADFORMAT; } if (mmioRead(This->hmmio, (HPSTR)&MainAVIHdr, ck.cksize) != ck.cksize) return AVIERR_FILEREAD; /* check for MAX_AVISTREAMS limit */ if (MainAVIHdr.dwStreams > MAX_AVISTREAMS) { WARN("file contains %u streams, but only supports %d -- change MAX_AVISTREAMS!\n", MainAVIHdr.dwStreams, MAX_AVISTREAMS); return AVIERR_UNSUPPORTED; } /* adjust permissions if copyrighted material in file */ if (MainAVIHdr.dwFlags & AVIFILEINFO_COPYRIGHTED) { This->uMode &= ~MMIO_RWMODE; This->uMode |= MMIO_READ; } /* convert MainAVIHeader into AVIFILINFOW */ memset(&This->fInfo, 0, sizeof(This->fInfo)); This->fInfo.dwRate = MainAVIHdr.dwMicroSecPerFrame; This->fInfo.dwScale = 1000000; This->fInfo.dwMaxBytesPerSec = MainAVIHdr.dwMaxBytesPerSec; This->fInfo.dwFlags = MainAVIHdr.dwFlags; This->fInfo.dwCaps = AVIFILECAPS_CANREAD|AVIFILECAPS_CANWRITE; This->fInfo.dwLength = MainAVIHdr.dwTotalFrames; This->fInfo.dwStreams = MainAVIHdr.dwStreams; This->fInfo.dwSuggestedBufferSize = 0; This->fInfo.dwWidth = MainAVIHdr.dwWidth; This->fInfo.dwHeight = MainAVIHdr.dwHeight; LoadStringW(AVIFILE_hModule, IDS_AVIFILETYPE, This->fInfo.szFileType, ARRAY_SIZE(This->fInfo.szFileType)); /* go back to into header list */ if (mmioAscend(This->hmmio, &ck, 0) != S_OK) return AVIERR_FILEREAD; /* foreach stream exists a "LIST","strl" chunk */ for (nStream = 0; nStream < This->fInfo.dwStreams; nStream++) { /* get next nested chunk in this "LIST","strl" */ if (mmioDescend(This->hmmio, &ckLIST2, &ckLIST1, 0) != S_OK) return AVIERR_FILEREAD; /* nested chunk must be of type "LIST","strl" -- when not normally JUNK */ if (ckLIST2.ckid == FOURCC_LIST && ckLIST2.fccType == listtypeSTREAMHEADER) { pStream = This->ppStreams[nStream] = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IAVIStreamImpl)); if (pStream == NULL) return AVIERR_MEMORY; AVIFILE_ConstructAVIStream(This, nStream, NULL); ck.ckid = 0; while (mmioDescend(This->hmmio, &ck, &ckLIST2, 0) == S_OK) { switch (ck.ckid) { case ckidSTREAMHANDLERDATA: if (pStream->lpHandlerData != NULL) return AVIERR_BADFORMAT; pStream->lpHandlerData = HeapAlloc(GetProcessHeap(), 0, ck.cksize); if (pStream->lpHandlerData == NULL) return AVIERR_MEMORY; pStream->cbHandlerData = ck.cksize; if (mmioRead(This->hmmio, pStream->lpHandlerData, ck.cksize) != ck.cksize) return AVIERR_FILEREAD; break; case ckidSTREAMFORMAT: if (pStream->lpFormat != NULL) return AVIERR_BADFORMAT; if (ck.cksize == 0) break; pStream->lpFormat = HeapAlloc(GetProcessHeap(), 0, ck.cksize); if (pStream->lpFormat == NULL) return AVIERR_MEMORY; pStream->cbFormat = ck.cksize; if (mmioRead(This->hmmio, pStream->lpFormat, ck.cksize) != ck.cksize) return AVIERR_FILEREAD; if (pStream->sInfo.fccType == streamtypeVIDEO) { LPBITMAPINFOHEADER lpbi = pStream->lpFormat; /* some corrections to the video format */ if (lpbi->biClrUsed == 0 && lpbi->biBitCount <= 8) lpbi->biClrUsed = 1u << lpbi->biBitCount; if (lpbi->biCompression == BI_RGB && lpbi->biSizeImage == 0) lpbi->biSizeImage = DIBWIDTHBYTES(*lpbi) * lpbi->biHeight; if (lpbi->biCompression != BI_RGB && lpbi->biBitCount == 8) { if (pStream->sInfo.fccHandler == mmioFOURCC('R','L','E','0') || pStream->sInfo.fccHandler == mmioFOURCC('R','L','E',' ')) lpbi->biCompression = BI_RLE8; } if (lpbi->biCompression == BI_RGB && (pStream->sInfo.fccHandler == 0 || pStream->sInfo.fccHandler == mmioFOURCC('N','O','N','E'))) pStream->sInfo.fccHandler = comptypeDIB; /* init rcFrame if it's empty */ if (IsRectEmpty(&pStream->sInfo.rcFrame)) SetRect(&pStream->sInfo.rcFrame, 0, 0, lpbi->biWidth, lpbi->biHeight); } break; case ckidSTREAMHEADER: { static const WCHAR streamTypeFmt[] = {'%','4','.','4','h','s',0}; static const WCHAR streamNameFmt[] = {'%','s',' ','%','s',' ','#','%','d',0}; AVIStreamHeader streamHdr; WCHAR szType[25]; UINT count; LONG n = ck.cksize; if (ck.cksize > sizeof(streamHdr)) n = sizeof(streamHdr); if (mmioRead(This->hmmio, (HPSTR)&streamHdr, n) != n) return AVIERR_FILEREAD; pStream->sInfo.fccType = streamHdr.fccType; pStream->sInfo.fccHandler = streamHdr.fccHandler; pStream->sInfo.dwFlags = streamHdr.dwFlags; pStream->sInfo.wPriority = streamHdr.wPriority; pStream->sInfo.wLanguage = streamHdr.wLanguage; pStream->sInfo.dwInitialFrames = streamHdr.dwInitialFrames; pStream->sInfo.dwScale = streamHdr.dwScale; pStream->sInfo.dwRate = streamHdr.dwRate; pStream->sInfo.dwStart = streamHdr.dwStart; pStream->sInfo.dwLength = streamHdr.dwLength; pStream->sInfo.dwSuggestedBufferSize = 0; pStream->sInfo.dwQuality = streamHdr.dwQuality; pStream->sInfo.dwSampleSize = streamHdr.dwSampleSize; pStream->sInfo.rcFrame.left = streamHdr.rcFrame.left; pStream->sInfo.rcFrame.top = streamHdr.rcFrame.top; pStream->sInfo.rcFrame.right = streamHdr.rcFrame.right; pStream->sInfo.rcFrame.bottom = streamHdr.rcFrame.bottom; pStream->sInfo.dwEditCount = 0; pStream->sInfo.dwFormatChangeCount = 0; /* generate description for stream like "filename.avi Type #n" */ if (streamHdr.fccType == streamtypeVIDEO) LoadStringW(AVIFILE_hModule, IDS_VIDEO, szType, ARRAY_SIZE(szType)); else if (streamHdr.fccType == streamtypeAUDIO) LoadStringW(AVIFILE_hModule, IDS_AUDIO, szType, ARRAY_SIZE(szType)); else wsprintfW(szType, streamTypeFmt, (char*)&streamHdr.fccType); /* get count of this streamtype up to this stream */ count = 0; for (n = nStream; 0 <= n; n--) { if (This->ppStreams[n]->sInfo.fccHandler == streamHdr.fccType) count++; } memset(pStream->sInfo.szName, 0, sizeof(pStream->sInfo.szName)); /* FIXME: avoid overflow -- better use wsnprintfW, which doesn't exists ! */ wsprintfW(pStream->sInfo.szName, streamNameFmt, AVIFILE_BasenameW(This->szFileName), szType, count); } break; case ckidSTREAMNAME: { /* streamname will be saved as ASCII string */ LPSTR str = HeapAlloc(GetProcessHeap(), 0, ck.cksize); if (str == NULL) return AVIERR_MEMORY; if (mmioRead(This->hmmio, str, ck.cksize) != ck.cksize) { HeapFree(GetProcessHeap(), 0, str); return AVIERR_FILEREAD; } MultiByteToWideChar(CP_ACP, 0, str, -1, pStream->sInfo.szName, ARRAY_SIZE(pStream->sInfo.szName)); HeapFree(GetProcessHeap(), 0, str); } break; case ckidAVIPADDING: case mmioFOURCC('p','a','d','d'): break; default: WARN(": found extra chunk 0x%08X\n", ck.ckid); hr = ReadChunkIntoExtra(&pStream->extra, This->hmmio, &ck); if (FAILED(hr)) return hr; }; if (pStream->lpFormat != NULL && pStream->sInfo.fccType == streamtypeAUDIO) { WAVEFORMATEX *wfx = pStream->lpFormat; /* wfx->nBlockAlign = wfx->nChannels * wfx->wBitsPerSample / 8; could be added */ pStream->sInfo.dwSampleSize = wfx->nBlockAlign; /* to deal with corrupt wfx->nBlockAlign but Windows doesn't do this */ TRACE("Block size reset to %u, chan=%u bpp=%u\n", wfx->nBlockAlign, wfx->nChannels, wfx->wBitsPerSample); pStream->sInfo.dwScale = 1; pStream->sInfo.dwRate = wfx->nSamplesPerSec; } if (mmioAscend(This->hmmio, &ck, 0) != S_OK) return AVIERR_FILEREAD; } } else { /* nested chunks in "LIST","hdrl" which are not of type "LIST","strl" */ hr = ReadChunkIntoExtra(&This->fileextra, This->hmmio, &ckLIST2); if (FAILED(hr)) return hr; } if (mmioAscend(This->hmmio, &ckLIST2, 0) != S_OK) return AVIERR_FILEREAD; } /* read any extra headers in "LIST","hdrl" */ FindChunkAndKeepExtras(&This->fileextra, This->hmmio, &ck, &ckLIST1, 0); if (mmioAscend(This->hmmio, &ckLIST1, 0) != S_OK) return AVIERR_FILEREAD; /* search "LIST","movi" chunk in "RIFF","AVI " */ ckLIST1.fccType = listtypeAVIMOVIE; hr = FindChunkAndKeepExtras(&This->fileextra, This->hmmio, &ckLIST1, &ckRIFF, MMIO_FINDLIST); if (FAILED(hr)) return hr; This->dwMoviChunkPos = ckLIST1.dwDataOffset; This->dwIdxChunkPos = ckLIST1.cksize + ckLIST1.dwDataOffset; if (mmioAscend(This->hmmio, &ckLIST1, 0) != S_OK) return AVIERR_FILEREAD; /* try to find an index */ ck.ckid = ckidAVINEWINDEX; hr = FindChunkAndKeepExtras(&This->fileextra, This->hmmio, &ck, &ckRIFF, MMIO_FINDCHUNK); if (SUCCEEDED(hr) && ck.cksize > 0) { if (FAILED(AVIFILE_LoadIndex(This, ck.cksize, ckLIST1.dwDataOffset))) This->fInfo.dwFlags &= ~AVIFILEINFO_HASINDEX; } /* when we haven't found an index or it's bad, then build one * by parsing 'movi' chunk */ if ((This->fInfo.dwFlags & AVIFILEINFO_HASINDEX) == 0) { for (nStream = 0; nStream < This->fInfo.dwStreams; nStream++) This->ppStreams[nStream]->lLastFrame = -1; if (mmioSeek(This->hmmio, ckLIST1.dwDataOffset + sizeof(DWORD), SEEK_SET) == -1) { ERR(": Oops, can't seek back to 'movi' chunk!\n"); return AVIERR_FILEREAD; } /* seek through the 'movi' list until end */ while (mmioDescend(This->hmmio, &ck, &ckLIST1, 0) == S_OK) { if (ck.ckid != FOURCC_LIST) { if (mmioAscend(This->hmmio, &ck, 0) == S_OK) { nStream = StreamFromFOURCC(ck.ckid); if (nStream > This->fInfo.dwStreams) return AVIERR_BADFORMAT; AVIFILE_AddFrame(This->ppStreams[nStream], ck.ckid, ck.cksize, ck.dwDataOffset - 2 * sizeof(DWORD), 0); } else { nStream = StreamFromFOURCC(ck.ckid); WARN(": file seems to be truncated!\n"); if (nStream <= This->fInfo.dwStreams && This->ppStreams[nStream]->sInfo.dwSampleSize > 0) { ck.cksize = mmioSeek(This->hmmio, 0, SEEK_END); if (ck.cksize != -1) { ck.cksize -= ck.dwDataOffset; ck.cksize &= ~(This->ppStreams[nStream]->sInfo.dwSampleSize - 1); AVIFILE_AddFrame(This->ppStreams[nStream], ck.ckid, ck.cksize, ck.dwDataOffset - 2 * sizeof(DWORD), 0); } } } } } } for (nStream = 0; nStream < This->fInfo.dwStreams; nStream++) { DWORD sugbuf = This->ppStreams[nStream]->sInfo.dwSuggestedBufferSize; if (This->fInfo.dwSuggestedBufferSize < sugbuf) This->fInfo.dwSuggestedBufferSize = sugbuf; } /* find other chunks */ FindChunkAndKeepExtras(&This->fileextra, This->hmmio, &ck, &ckRIFF, 0); return AVIERR_OK; } static HRESULT AVIFILE_LoadIndex(const IAVIFileImpl *This, DWORD size, DWORD offset) { AVIINDEXENTRY *lp; DWORD pos, n; HRESULT hr = AVIERR_OK; BOOL bAbsolute = TRUE; lp = HeapAlloc(GetProcessHeap(), 0, IDX_PER_BLOCK * sizeof(AVIINDEXENTRY)); if (lp == NULL) return AVIERR_MEMORY; /* adjust limits for index tables, so that inserting will be faster */ for (n = 0; n < This->fInfo.dwStreams; n++) { IAVIStreamImpl *pStream = This->ppStreams[n]; pStream->lLastFrame = -1; if (pStream->idxFrames != NULL) { HeapFree(GetProcessHeap(), 0, pStream->idxFrames); pStream->idxFrames = NULL; pStream->nIdxFrames = 0; } if (pStream->sInfo.dwSampleSize != 0) { if (n > 0 && This->fInfo.dwFlags & AVIFILEINFO_ISINTERLEAVED) { pStream->nIdxFrames = This->ppStreams[0]->nIdxFrames; } else if (pStream->sInfo.dwSuggestedBufferSize) { pStream->nIdxFrames = pStream->sInfo.dwLength / pStream->sInfo.dwSuggestedBufferSize; } } else pStream->nIdxFrames = pStream->sInfo.dwLength; pStream->idxFrames = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, pStream->nIdxFrames * sizeof(AVIINDEXENTRY)); if (pStream->idxFrames == NULL && pStream->nIdxFrames > 0) { pStream->nIdxFrames = 0; HeapFree(GetProcessHeap(), 0, lp); return AVIERR_MEMORY; } } pos = (DWORD)-1; while (size != 0) { LONG read = min(IDX_PER_BLOCK * sizeof(AVIINDEXENTRY), size); if (mmioRead(This->hmmio, (HPSTR)lp, read) != read) { hr = AVIERR_FILEREAD; break; } size -= read; if (pos == (DWORD)-1) pos = offset - lp->dwChunkOffset + sizeof(DWORD); AVIFILE_ParseIndex(This, lp, read / sizeof(AVIINDEXENTRY), pos, &bAbsolute); } HeapFree(GetProcessHeap(), 0, lp); /* checking ... */ for (n = 0; n < This->fInfo.dwStreams; n++) { IAVIStreamImpl *pStream = This->ppStreams[n]; if (pStream->sInfo.dwSampleSize == 0 && pStream->sInfo.dwLength != pStream->lLastFrame+1) ERR("stream %u length mismatch: dwLength=%u found=%d\n", n, pStream->sInfo.dwLength, pStream->lLastFrame); } return hr; } static HRESULT AVIFILE_ParseIndex(const IAVIFileImpl *This, AVIINDEXENTRY *lp, LONG count, DWORD pos, BOOL *bAbsolute) { if (lp == NULL) return AVIERR_BADPARAM; for (; count > 0; count--, lp++) { WORD nStream = StreamFromFOURCC(lp->ckid); if (lp->ckid == listtypeAVIRECORD || nStream == 0x7F) continue; /* skip these */ if (nStream > This->fInfo.dwStreams) return AVIERR_BADFORMAT; /* Video frames can be either indexed in a relative position to the * "movi" chunk or in a absolute position in the file. If the index * is relative the frame offset will always be so small that it will * virtually never reach the "movi" offset so we can detect if the * video is relative very fast. */ if (*bAbsolute && lp->dwChunkOffset < This->dwMoviChunkPos) *bAbsolute = FALSE; if (!*bAbsolute) lp->dwChunkOffset += pos; /* make the offset absolute */ if (FAILED(AVIFILE_AddFrame(This->ppStreams[nStream], lp->ckid, lp->dwChunkLength, lp->dwChunkOffset, lp->dwFlags))) return AVIERR_MEMORY; } return AVIERR_OK; } static HRESULT AVIFILE_ReadBlock(IAVIStreamImpl *This, DWORD pos, LPVOID buffer, DWORD size) { /* pre-conditions */ assert(This != NULL); assert(This->paf != NULL); assert(This->paf->hmmio != NULL); assert(This->sInfo.dwStart <= pos && pos < This->sInfo.dwLength); assert(pos <= This->lLastFrame); /* should we read as much as block gives us? */ if (size == 0 || size > This->idxFrames[pos].dwChunkLength) size = This->idxFrames[pos].dwChunkLength; /* read into out own buffer or given one? */ if (buffer == NULL) { /* we also read the chunk */ size += 2 * sizeof(DWORD); /* check that buffer is big enough -- don't trust dwSuggestedBufferSize */ if (This->lpBuffer == NULL || This->cbBuffer < size) { DWORD maxSize = max(size, This->sInfo.dwSuggestedBufferSize); if (This->lpBuffer == NULL) { This->lpBuffer = HeapAlloc(GetProcessHeap(), 0, maxSize); if (!This->lpBuffer) return AVIERR_MEMORY; } else { void *new_buffer = HeapReAlloc(GetProcessHeap(), 0, This->lpBuffer, maxSize); if (!new_buffer) return AVIERR_MEMORY; This->lpBuffer = new_buffer; } This->cbBuffer = maxSize; } /* now read the complete chunk into our buffer */ if (mmioSeek(This->paf->hmmio, This->idxFrames[pos].dwChunkOffset, SEEK_SET) == -1) return AVIERR_FILEREAD; if (mmioRead(This->paf->hmmio, (HPSTR)This->lpBuffer, size) != size) return AVIERR_FILEREAD; /* check if it was the correct block which we have read */ if (This->lpBuffer[0] != This->idxFrames[pos].ckid || This->lpBuffer[1] != This->idxFrames[pos].dwChunkLength) { ERR(": block %d not found at 0x%08X\n", pos, This->idxFrames[pos].dwChunkOffset); ERR(": Index says: '%4.4s'(0x%08X) size 0x%08X\n", (char*)&This->idxFrames[pos].ckid, This->idxFrames[pos].ckid, This->idxFrames[pos].dwChunkLength); ERR(": Data says: '%4.4s'(0x%08X) size 0x%08X\n", (char*)&This->lpBuffer[0], This->lpBuffer[0], This->lpBuffer[1]); return AVIERR_FILEREAD; } } else { if (mmioSeek(This->paf->hmmio, This->idxFrames[pos].dwChunkOffset + 2 * sizeof(DWORD), SEEK_SET) == -1) return AVIERR_FILEREAD; if (mmioRead(This->paf->hmmio, buffer, size) != size) return AVIERR_FILEREAD; } return AVIERR_OK; } static void AVIFILE_SamplesToBlock(const IAVIStreamImpl *This, LPLONG pos, LPLONG offset) { LONG block; /* pre-conditions */ assert(This != NULL); assert(pos != NULL); assert(offset != NULL); assert(This->sInfo.dwSampleSize != 0); assert(*pos >= This->sInfo.dwStart); /* convert start sample to start bytes */ (*offset) = (*pos) - This->sInfo.dwStart; (*offset) *= This->sInfo.dwSampleSize; /* convert bytes to block number */ for (block = 0; block <= This->lLastFrame; block++) { if (This->idxFrames[block].dwChunkLength <= *offset) (*offset) -= This->idxFrames[block].dwChunkLength; else break; } *pos = block; } static HRESULT AVIFILE_SaveFile(IAVIFileImpl *This) { MainAVIHeader MainAVIHdr; IAVIStreamImpl* pStream; MMCKINFO ckRIFF; MMCKINFO ckLIST1; MMCKINFO ckLIST2; MMCKINFO ck; DWORD nStream; DWORD dwPos; HRESULT hr; /* initialize some things */ if (This->dwMoviChunkPos == 0) AVIFILE_ComputeMoviStart(This); /* written one record too much? */ if (This->ckLastRecord.dwFlags & MMIO_DIRTY) { This->dwNextFramePos -= 3 * sizeof(DWORD); if (This->nIdxRecords > 0) This->nIdxRecords--; } AVIFILE_UpdateInfo(This); assert(This->fInfo.dwScale != 0); memset(&MainAVIHdr, 0, sizeof(MainAVIHdr)); MainAVIHdr.dwMicroSecPerFrame = MulDiv(This->fInfo.dwRate, 1000000, This->fInfo.dwScale); MainAVIHdr.dwMaxBytesPerSec = This->fInfo.dwMaxBytesPerSec; MainAVIHdr.dwPaddingGranularity = AVI_HEADERSIZE; MainAVIHdr.dwFlags = This->fInfo.dwFlags; MainAVIHdr.dwTotalFrames = This->fInfo.dwLength; MainAVIHdr.dwInitialFrames = 0; MainAVIHdr.dwStreams = This->fInfo.dwStreams; MainAVIHdr.dwSuggestedBufferSize = This->fInfo.dwSuggestedBufferSize; MainAVIHdr.dwWidth = This->fInfo.dwWidth; MainAVIHdr.dwHeight = This->fInfo.dwHeight; MainAVIHdr.dwInitialFrames = This->dwInitialFrames; /* now begin writing ... */ mmioSeek(This->hmmio, 0, SEEK_SET); /* RIFF chunk */ ckRIFF.cksize = 0; ckRIFF.fccType = formtypeAVI; if (mmioCreateChunk(This->hmmio, &ckRIFF, MMIO_CREATERIFF) != S_OK) return AVIERR_FILEWRITE; /* AVI headerlist */ ckLIST1.cksize = 0; ckLIST1.fccType = listtypeAVIHEADER; if (mmioCreateChunk(This->hmmio, &ckLIST1, MMIO_CREATELIST) != S_OK) return AVIERR_FILEWRITE; /* MainAVIHeader */ ck.ckid = ckidAVIMAINHDR; ck.cksize = sizeof(MainAVIHdr); ck.fccType = 0; if (mmioCreateChunk(This->hmmio, &ck, 0) != S_OK) return AVIERR_FILEWRITE; if (mmioWrite(This->hmmio, (HPSTR)&MainAVIHdr, ck.cksize) != ck.cksize) return AVIERR_FILEWRITE; if (mmioAscend(This->hmmio, &ck, 0) != S_OK) return AVIERR_FILEWRITE; /* write the headers of each stream into a separate streamheader list */ for (nStream = 0; nStream < This->fInfo.dwStreams; nStream++) { AVIStreamHeader strHdr; pStream = This->ppStreams[nStream]; /* begin the new streamheader list */ ckLIST2.cksize = 0; ckLIST2.fccType = listtypeSTREAMHEADER; if (mmioCreateChunk(This->hmmio, &ckLIST2, MMIO_CREATELIST) != S_OK) return AVIERR_FILEWRITE; /* create an AVIStreamHeader from the AVSTREAMINFO */ strHdr.fccType = pStream->sInfo.fccType; strHdr.fccHandler = pStream->sInfo.fccHandler; strHdr.dwFlags = pStream->sInfo.dwFlags; strHdr.wPriority = pStream->sInfo.wPriority; strHdr.wLanguage = pStream->sInfo.wLanguage; strHdr.dwInitialFrames = pStream->sInfo.dwInitialFrames; strHdr.dwScale = pStream->sInfo.dwScale; strHdr.dwRate = pStream->sInfo.dwRate; strHdr.dwStart = pStream->sInfo.dwStart; strHdr.dwLength = pStream->sInfo.dwLength; strHdr.dwSuggestedBufferSize = pStream->sInfo.dwSuggestedBufferSize; strHdr.dwQuality = pStream->sInfo.dwQuality; strHdr.dwSampleSize = pStream->sInfo.dwSampleSize; strHdr.rcFrame.left = pStream->sInfo.rcFrame.left; strHdr.rcFrame.top = pStream->sInfo.rcFrame.top; strHdr.rcFrame.right = pStream->sInfo.rcFrame.right; strHdr.rcFrame.bottom = pStream->sInfo.rcFrame.bottom; /* now write the AVIStreamHeader */ ck.ckid = ckidSTREAMHEADER; ck.cksize = sizeof(strHdr); if (mmioCreateChunk(This->hmmio, &ck, 0) != S_OK) return AVIERR_FILEWRITE; if (mmioWrite(This->hmmio, (HPSTR)&strHdr, ck.cksize) != ck.cksize) return AVIERR_FILEWRITE; if (mmioAscend(This->hmmio, &ck, 0) != S_OK) return AVIERR_FILEWRITE; /* ... the hopefully ever present streamformat ... */ ck.ckid = ckidSTREAMFORMAT; ck.cksize = pStream->cbFormat; if (mmioCreateChunk(This->hmmio, &ck, 0) != S_OK) return AVIERR_FILEWRITE; if (pStream->lpFormat != NULL && ck.cksize > 0) { if (mmioWrite(This->hmmio, pStream->lpFormat, ck.cksize) != ck.cksize) return AVIERR_FILEWRITE; } if (mmioAscend(This->hmmio, &ck, 0) != S_OK) return AVIERR_FILEWRITE; /* ... some optional existing handler data ... */ if (pStream->lpHandlerData != NULL && pStream->cbHandlerData > 0) { ck.ckid = ckidSTREAMHANDLERDATA; ck.cksize = pStream->cbHandlerData; if (mmioCreateChunk(This->hmmio, &ck, 0) != S_OK) return AVIERR_FILEWRITE; if (mmioWrite(This->hmmio, pStream->lpHandlerData, ck.cksize) != ck.cksize) return AVIERR_FILEWRITE; if (mmioAscend(This->hmmio, &ck, 0) != S_OK) return AVIERR_FILEWRITE; } /* ... some optional additional extra chunk for this stream ... */ if (pStream->extra.lp != NULL && pStream->extra.cb > 0) { /* the chunk header(s) are already in the structure */ if (mmioWrite(This->hmmio, pStream->extra.lp, pStream->extra.cb) != pStream->extra.cb) return AVIERR_FILEWRITE; } /* ... an optional name for this stream ... */ if (pStream->sInfo.szName[0]) { LPSTR str; ck.ckid = ckidSTREAMNAME; ck.cksize = lstrlenW(pStream->sInfo.szName) + 1; if (ck.cksize & 1) /* align */ ck.cksize++; if (mmioCreateChunk(This->hmmio, &ck, 0) != S_OK) return AVIERR_FILEWRITE; /* the streamname must be saved in ASCII not Unicode */ str = HeapAlloc(GetProcessHeap(), 0, ck.cksize); if (str == NULL) return AVIERR_MEMORY; WideCharToMultiByte(CP_ACP, 0, pStream->sInfo.szName, -1, str, ck.cksize, NULL, NULL); if (mmioWrite(This->hmmio, str, ck.cksize) != ck.cksize) { HeapFree(GetProcessHeap(), 0, str); return AVIERR_FILEWRITE; } HeapFree(GetProcessHeap(), 0, str); if (mmioAscend(This->hmmio, &ck, 0) != S_OK) return AVIERR_FILEWRITE; } /* close streamheader list for this stream */ if (mmioAscend(This->hmmio, &ckLIST2, 0) != S_OK) return AVIERR_FILEWRITE; } /* for (0 <= nStream < MainAVIHdr.dwStreams) */ /* close the aviheader list */ if (mmioAscend(This->hmmio, &ckLIST1, 0) != S_OK) return AVIERR_FILEWRITE; /* check for padding to pre-guessed 'movi'-chunk position */ dwPos = ckLIST1.dwDataOffset + ckLIST1.cksize; if (This->dwMoviChunkPos - 2 * sizeof(DWORD) > dwPos) { ck.ckid = ckidAVIPADDING; ck.cksize = This->dwMoviChunkPos - dwPos - 4 * sizeof(DWORD); assert((LONG)ck.cksize >= 0); if (mmioCreateChunk(This->hmmio, &ck, 0) != S_OK) return AVIERR_FILEWRITE; if (mmioSeek(This->hmmio, ck.cksize, SEEK_CUR) == -1) return AVIERR_FILEWRITE; if (mmioAscend(This->hmmio, &ck, 0) != S_OK) return AVIERR_FILEWRITE; } /* now write the 'movi' chunk */ mmioSeek(This->hmmio, This->dwMoviChunkPos - 2 * sizeof(DWORD), SEEK_SET); ckLIST1.cksize = 0; ckLIST1.fccType = listtypeAVIMOVIE; if (mmioCreateChunk(This->hmmio, &ckLIST1, MMIO_CREATELIST) != S_OK) return AVIERR_FILEWRITE; if (mmioSeek(This->hmmio, This->dwNextFramePos, SEEK_SET) == -1) return AVIERR_FILEWRITE; if (mmioAscend(This->hmmio, &ckLIST1, 0) != S_OK) return AVIERR_FILEWRITE; /* write 'idx1' chunk */ hr = AVIFILE_SaveIndex(This); if (FAILED(hr)) return hr; /* write optional extra file chunks */ if (This->fileextra.lp != NULL && This->fileextra.cb > 0) { /* as for the streams, are the chunk header(s) in the structure */ if (mmioWrite(This->hmmio, This->fileextra.lp, This->fileextra.cb) != This->fileextra.cb) return AVIERR_FILEWRITE; } /* close RIFF chunk */ if (mmioAscend(This->hmmio, &ckRIFF, 0) != S_OK) return AVIERR_FILEWRITE; /* add some JUNK at end for bad parsers */ memset(&ckRIFF, 0, sizeof(ckRIFF)); mmioWrite(This->hmmio, (HPSTR)&ckRIFF, sizeof(ckRIFF)); mmioFlush(This->hmmio, 0); return AVIERR_OK; } static HRESULT AVIFILE_SaveIndex(const IAVIFileImpl *This) { IAVIStreamImpl *pStream; AVIINDEXENTRY idx; MMCKINFO ck; DWORD nStream; LONG n; ck.ckid = ckidAVINEWINDEX; ck.cksize = 0; if (mmioCreateChunk(This->hmmio, &ck, 0) != S_OK) return AVIERR_FILEWRITE; if (This->fInfo.dwFlags & AVIFILEINFO_ISINTERLEAVED) { /* is interleaved -- write block of corresponding frames */ LONG lInitialFrames = 0; LONG stepsize; LONG i; if (This->ppStreams[0]->sInfo.dwSampleSize == 0) stepsize = 1; else stepsize = AVIStreamTimeToSample(&This->ppStreams[0]->IAVIStream_iface, 1000000); assert(stepsize > 0); for (nStream = 0; nStream < This->fInfo.dwStreams; nStream++) { if (lInitialFrames < This->ppStreams[nStream]->sInfo.dwInitialFrames) lInitialFrames = This->ppStreams[nStream]->sInfo.dwInitialFrames; } for (i = -lInitialFrames; i < (LONG)This->fInfo.dwLength - lInitialFrames; i += stepsize) { DWORD nFrame = lInitialFrames + i; assert(nFrame < This->nIdxRecords); idx.ckid = listtypeAVIRECORD; idx.dwFlags = AVIIF_LIST; idx.dwChunkLength = This->idxRecords[nFrame].dwChunkLength; idx.dwChunkOffset = This->idxRecords[nFrame].dwChunkOffset - This->dwMoviChunkPos; if (mmioWrite(This->hmmio, (HPSTR)&idx, sizeof(idx)) != sizeof(idx)) return AVIERR_FILEWRITE; for (nStream = 0; nStream < This->fInfo.dwStreams; nStream++) { pStream = This->ppStreams[nStream]; /* heave we reached start of this stream? */ if (-(LONG)pStream->sInfo.dwInitialFrames > i) continue; if (pStream->sInfo.dwInitialFrames < lInitialFrames) nFrame -= (lInitialFrames - pStream->sInfo.dwInitialFrames); /* reached end of this stream? */ if (pStream->lLastFrame <= nFrame) continue; if ((pStream->sInfo.dwFlags & AVISTREAMINFO_FORMATCHANGES) && pStream->sInfo.dwFormatChangeCount != 0 && pStream->idxFmtChanges != NULL) { DWORD pos; for (pos = 0; pos < pStream->sInfo.dwFormatChangeCount; pos++) { if (pStream->idxFmtChanges[pos].ckid == nFrame) { idx.dwFlags = AVIIF_NOTIME; idx.ckid = MAKEAVICKID(cktypePALchange, pStream->nStream); idx.dwChunkLength = pStream->idxFmtChanges[pos].dwChunkLength; idx.dwChunkOffset = pStream->idxFmtChanges[pos].dwChunkOffset - This->dwMoviChunkPos; if (mmioWrite(This->hmmio, (HPSTR)&idx, sizeof(idx)) != sizeof(idx)) return AVIERR_FILEWRITE; break; } } } /* if have formatchanges */ idx.ckid = pStream->idxFrames[nFrame].ckid; idx.dwFlags = pStream->idxFrames[nFrame].dwFlags; idx.dwChunkLength = pStream->idxFrames[nFrame].dwChunkLength; idx.dwChunkOffset = pStream->idxFrames[nFrame].dwChunkOffset - This->dwMoviChunkPos; if (mmioWrite(This->hmmio, (HPSTR)&idx, sizeof(idx)) != sizeof(idx)) return AVIERR_FILEWRITE; } } } else { /* not interleaved -- write index for each stream at once */ for (nStream = 0; nStream < This->fInfo.dwStreams; nStream++) { pStream = This->ppStreams[nStream]; for (n = 0; n <= pStream->lLastFrame; n++) { if ((pStream->sInfo.dwFlags & AVISTREAMINFO_FORMATCHANGES) && (pStream->sInfo.dwFormatChangeCount != 0)) { DWORD pos; for (pos = 0; pos < pStream->sInfo.dwFormatChangeCount; pos++) { if (pStream->idxFmtChanges[pos].ckid == n) { idx.dwFlags = AVIIF_NOTIME; idx.ckid = MAKEAVICKID(cktypePALchange, pStream->nStream); idx.dwChunkLength = pStream->idxFmtChanges[pos].dwChunkLength; idx.dwChunkOffset = pStream->idxFmtChanges[pos].dwChunkOffset - This->dwMoviChunkPos; if (mmioWrite(This->hmmio, (HPSTR)&idx, sizeof(idx)) != sizeof(idx)) return AVIERR_FILEWRITE; break; } } } /* if have formatchanges */ idx.ckid = pStream->idxFrames[n].ckid; idx.dwFlags = pStream->idxFrames[n].dwFlags; idx.dwChunkLength = pStream->idxFrames[n].dwChunkLength; idx.dwChunkOffset = pStream->idxFrames[n].dwChunkOffset - This->dwMoviChunkPos; if (mmioWrite(This->hmmio, (HPSTR)&idx, sizeof(idx)) != sizeof(idx)) return AVIERR_FILEWRITE; } } } /* if not interleaved */ if (mmioAscend(This->hmmio, &ck, 0) != S_OK) return AVIERR_FILEWRITE; return AVIERR_OK; } static ULONG AVIFILE_SearchStream(const IAVIFileImpl *This, DWORD fcc, LONG lSkip) { UINT i; UINT nStream; /* pre-condition */ assert(lSkip >= 0); if (fcc != 0) { /* search the number of the specified stream */ nStream = (ULONG)-1; for (i = 0; i < This->fInfo.dwStreams; i++) { assert(This->ppStreams[i] != NULL); if (This->ppStreams[i]->sInfo.fccType == fcc) { if (lSkip == 0) { nStream = i; break; } else lSkip--; } } } else nStream = lSkip; return nStream; } static void AVIFILE_UpdateInfo(IAVIFileImpl *This) { UINT i; /* pre-conditions */ assert(This != NULL); This->fInfo.dwMaxBytesPerSec = 0; This->fInfo.dwCaps = AVIFILECAPS_CANREAD|AVIFILECAPS_CANWRITE; This->fInfo.dwSuggestedBufferSize = 0; This->fInfo.dwWidth = 0; This->fInfo.dwHeight = 0; This->fInfo.dwScale = 0; This->fInfo.dwRate = 0; This->fInfo.dwLength = 0; This->dwInitialFrames = 0; for (i = 0; i < This->fInfo.dwStreams; i++) { AVISTREAMINFOW *psi; DWORD n; /* pre-conditions */ assert(This->ppStreams[i] != NULL); psi = &This->ppStreams[i]->sInfo; assert(psi->dwScale != 0); assert(psi->dwRate != 0); if (i == 0) { /* use first stream timings as base */ This->fInfo.dwScale = psi->dwScale; This->fInfo.dwRate = psi->dwRate; This->fInfo.dwLength = psi->dwLength; } else { n = AVIStreamSampleToSample(&This->ppStreams[0]->IAVIStream_iface, &This->ppStreams[i]->IAVIStream_iface, psi->dwLength); if (This->fInfo.dwLength < n) This->fInfo.dwLength = n; } if (This->dwInitialFrames < psi->dwInitialFrames) This->dwInitialFrames = psi->dwInitialFrames; if (This->fInfo.dwSuggestedBufferSize < psi->dwSuggestedBufferSize) This->fInfo.dwSuggestedBufferSize = psi->dwSuggestedBufferSize; if (psi->dwSampleSize != 0) { /* fixed sample size -- exact computation */ This->fInfo.dwMaxBytesPerSec += MulDiv(psi->dwSampleSize, psi->dwRate, psi->dwScale); } else { /* variable sample size -- only upper limit */ This->fInfo.dwMaxBytesPerSec += MulDiv(psi->dwSuggestedBufferSize, psi->dwRate, psi->dwScale); /* update dimensions */ n = psi->rcFrame.right - psi->rcFrame.left; if (This->fInfo.dwWidth < n) This->fInfo.dwWidth = n; n = psi->rcFrame.bottom - psi->rcFrame.top; if (This->fInfo.dwHeight < n) This->fInfo.dwHeight = n; } } } static HRESULT AVIFILE_WriteBlock(IAVIStreamImpl *This, DWORD block, FOURCC ckid, DWORD flags, LPCVOID buffer, LONG size) { MMCKINFO ck; ck.ckid = ckid; ck.cksize = size; ck.fccType = 0; /* if no frame/block is already written, we must compute start of movi chunk */ if (This->paf->dwMoviChunkPos == 0) AVIFILE_ComputeMoviStart(This->paf); if (mmioSeek(This->paf->hmmio, This->paf->dwNextFramePos, SEEK_SET) == -1) return AVIERR_FILEWRITE; if (mmioCreateChunk(This->paf->hmmio, &ck, 0) != S_OK) return AVIERR_FILEWRITE; if (buffer != NULL && size > 0) { if (mmioWrite(This->paf->hmmio, buffer, size) != size) return AVIERR_FILEWRITE; } if (mmioAscend(This->paf->hmmio, &ck, 0) != S_OK) return AVIERR_FILEWRITE; This->paf->fDirty = TRUE; This->paf->dwNextFramePos = mmioSeek(This->paf->hmmio, 0, SEEK_CUR); return AVIFILE_AddFrame(This, ckid, size, ck.dwDataOffset - 2 * sizeof(DWORD), flags); }