public int GetData(ref FORMATETC format, out STGMEDIUM medium)
{
try
{
//Get data into passed medium
log.DebugFormat("IDataObject.GetData called -- cfFormat {0} dwAspect {1} lindex {2} ptd {3} tymed {4}", format.cfFormat, format.dwAspect, format.lindex, format.ptd, format.tymed);
log.DebugFormat("Format name: {0}", System.Windows.Forms.DataFormats.GetFormat((ushort)format.cfFormat).Name);
if (format.cfFormat == NativeMethods.CF_HDROP)
{
medium = new STGMEDIUM();
//Validate index
if (format.lindex != -1)
{
return(NativeMethods.DV_E_LINDEX);
}
//Validate medium type
if (!format.tymed.HasFlag(TYMED.TYMED_HGLOBAL))
{
return(NativeMethods.DV_E_TYMED);
}
//Validate DV aspect
if (format.dwAspect != DVASPECT.DVASPECT_CONTENT)
{
return(NativeMethods.DV_E_DVASPECT);
}
//Extract files if not already extracted
if (tempFilenames == null)
{
ExtractFiles();
}
//Get list of dropped files
log.Debug("Setting drop files");
DataObjectHelper.SetDropFiles(ref medium, tempFilenames);
FilesDropped = true;
return(NativeMethods.S_OK);
}
else if (format.cfFormat == NativeMethods.CF_TEXT || format.cfFormat == NativeMethods.CF_UNICODETEXT || format.cfFormat == (ushort)DataObjectHelper.GetClipboardFormat("Csv"))
{
//Do not return text formats -- some applications try to get text before files
medium = new STGMEDIUM();
return(NativeMethods.DV_E_FORMATETC);
}
else
{
int result = innerData.GetData(format, out medium);
log.DebugFormat("Result: {0}", result);
return(result);
}
}
catch (Exception ex)
{
log.Error("Exception in IDataObject.GetData", ex);
medium = new STGMEDIUM();
return(NativeMethods.E_UNEXPECTED);
}
}
internal static void ReadFileContents(NativeMethods.IDataObject data, int index, Stream stream)
{
STGMEDIUM medium = new STGMEDIUM();
try
{
//Define FileContents format
FORMATETC format = new FORMATETC();
format.cfFormat = (short)GetClipboardFormat("FileContents");
format.dwAspect = DVASPECT.DVASPECT_CONTENT;
format.lindex = index;
format.ptd = IntPtr.Zero;
format.tymed = TYMED.TYMED_ISTREAM | TYMED.TYMED_ISTORAGE | TYMED.TYMED_HGLOBAL;
//Get data
int retVal = data.GetData(format, out medium);
if (retVal != NativeMethods.S_OK)
{
throw new Exception(string.Format("Could not get FileContents format. Error returned: {0}", retVal));
}
//Read medium into stream
ReadMediumIntoStream(medium, stream);
}
finally
{
//Release all unmanaged objects
if (medium.pUnkForRelease == null)
{
NativeMethods.ReleaseStgMedium(ref medium);
}
}
}
internal static void ReadFileContents(NativeMethods.IDataObject data, int index, Stream stream)
{
STGMEDIUM medium = new STGMEDIUM();
try
{
//Define FileContents format
FORMATETC format = new FORMATETC();
format.cfFormat = (short)System.Windows.Forms.DataFormats.GetFormat("FileContents").Id;
format.dwAspect = DVASPECT.DVASPECT_CONTENT;
format.lindex = index;
format.ptd = IntPtr.Zero;
format.tymed = TYMED.TYMED_ISTREAM | TYMED.TYMED_ISTORAGE | TYMED.TYMED_HGLOBAL;
//Get data
medium = new STGMEDIUM();
data.GetData(format, out medium);
//Read medium into stream
ReadMediumIntoStream(medium, stream);
}
finally
{
//Release all unmanaged objects
if (medium.pUnkForRelease == null)
{
NativeMethods.ReleaseStgMedium(ref medium);
}
}
}
internal static string[] GetFilenamesAnsi(NativeMethods.IDataObject data)
{
IntPtr fgdaPtr = IntPtr.Zero;
try
{
//Define FileGroupDescriptor format
FORMATETC format = new FORMATETC();
format.cfFormat = (short)System.Windows.Forms.DataFormats.GetFormat("FileGroupDescriptor").Id;
format.dwAspect = DVASPECT.DVASPECT_CONTENT;
format.lindex = -1;
format.ptd = IntPtr.Zero;
format.tymed = TYMED.TYMED_ISTREAM | TYMED.TYMED_ISTORAGE | TYMED.TYMED_HGLOBAL;
//Query if format exists in data
if (data.QueryGetData(format) != NativeMethods.S_OK)
{
return(null);
}
//Get data into medium
STGMEDIUM medium = new STGMEDIUM();
data.GetData(format, out medium);
//Read medium into byte array
byte[] bytes;
using (MemoryStream stream = new MemoryStream())
{
DataObjectHelper.ReadMediumIntoStream(medium, stream);
bytes = new byte[stream.Length];
stream.Seek(0, SeekOrigin.Begin);
stream.Read(bytes, 0, bytes.Length);
}
//Copy the file group descriptor into unmanaged memory
fgdaPtr = Marshal.AllocHGlobal(bytes.Length);
Marshal.Copy(bytes, 0, fgdaPtr, bytes.Length);
//Marshal the unmanaged memory to a FILEGROUPDESCRIPTORA struct
object fgdObj = Marshal.PtrToStructure(fgdaPtr, typeof(NativeMethods.FILEGROUPDESCRIPTORA));
NativeMethods.FILEGROUPDESCRIPTORA fgd = (NativeMethods.FILEGROUPDESCRIPTORA)fgdObj;
//Create an array to store file names
string[] filenames = new string[fgd.cItems];
//Get the pointer to the first file descriptor
IntPtr fdPtr = (IntPtr)((int)fgdaPtr + Marshal.SizeOf(fgdaPtr));
//Loop for the number of files acording to the file group descriptor
for (int fdIndex = 0; fdIndex < fgd.cItems; fdIndex++)
{
//Marshal the pointer to the file descriptor as a FILEDESCRIPTORA struct
object fdObj = Marshal.PtrToStructure(fdPtr, typeof(NativeMethods.FILEDESCRIPTORA));
NativeMethods.FILEDESCRIPTORA fd = (NativeMethods.FILEDESCRIPTORA)fdObj;
//Get filename of file descriptor and put in array
filenames[fdIndex] = fd.cFileName;
//Move the file descriptor pointer to the next file descriptor
fdPtr = (IntPtr)((int)fdPtr + Marshal.SizeOf(fd));
}
return(filenames);
}
finally
{
Marshal.FreeHGlobal(fgdaPtr);
}
}
internal static string[] GetFilenamesAnsi(NativeMethods.IDataObject data)
{
log.Debug("Getting filenames (ANSI)");
IntPtr ptrFgd = IntPtr.Zero;
STGMEDIUM medium = new STGMEDIUM();
try
{
//Define FileGroupDescriptor format
FORMATETC format = new FORMATETC();
format.cfFormat = (short)GetClipboardFormat("FileGroupDescriptor");
format.dwAspect = DVASPECT.DVASPECT_CONTENT;
format.lindex = -1;
format.ptd = IntPtr.Zero;
format.tymed = TYMED.TYMED_ISTREAM | TYMED.TYMED_ISTORAGE | TYMED.TYMED_HGLOBAL;
//Query if format exists in data
if (data.QueryGetData(format) != NativeMethods.S_OK)
{
log.Debug("No filenames found");
return(null);
}
//Get data into medium
int retVal = data.GetData(format, out medium);
if (retVal != NativeMethods.S_OK)
{
throw new Exception(string.Format("Could not get FileGroupDescriptor format. Error returned: {0}", retVal));
}
//Read medium into byte array
log.Debug("Reading structure into memory stream");
byte[] bytes;
using (MemoryStream stream = new MemoryStream())
{
DataObjectHelper.ReadMediumIntoStream(medium, stream);
bytes = new byte[stream.Length];
stream.Seek(0, SeekOrigin.Begin);
stream.Read(bytes, 0, bytes.Length);
}
//Copy byte array into unmanaged memory
log.Debug("Copying structure into unmanaged memory");
ptrFgd = Marshal.AllocHGlobal(bytes.Length);
Marshal.Copy(bytes, 0, ptrFgd, bytes.Length);
//Marshal unmanaged memory to a FILEGROUPDESCRIPTORA struct
log.Debug("Marshaling unmanaged memory into FILEGROUPDESCRIPTORA struct");
NativeMethods.FILEGROUPDESCRIPTORA fgd = (NativeMethods.FILEGROUPDESCRIPTORA)Marshal.PtrToStructure(ptrFgd, typeof(NativeMethods.FILEGROUPDESCRIPTORA));
log.Debug(string.Format("Files found: {0}", fgd.cItems));
//Create an array to store file names
string[] filenames = new string[fgd.cItems];
//Get the pointer to the first file descriptor
IntPtr fdPtr = IntPtr.Add(ptrFgd, sizeof(uint));
//Loop for the number of files acording to the file group descriptor
for (int fdIndex = 0; fdIndex < fgd.cItems; fdIndex++)
{
log.DebugFormat("Filenames found: {0}", string.Join(", ", filenames));
//Marshal pointer to a FILEDESCRIPTORA struct
NativeMethods.FILEDESCRIPTORA fd = (NativeMethods.FILEDESCRIPTORA)Marshal.PtrToStructure(fdPtr, typeof(NativeMethods.FILEDESCRIPTORA));
//Get filename of file descriptor and put in array
filenames[fdIndex] = fd.cFileName;
//Move the file descriptor pointer to the next file descriptor
fdPtr = IntPtr.Add(fdPtr, Marshal.SizeOf(fd));
}
log.DebugFormat("Filenames found: {0}", string.Join(", ", filenames));
return(filenames);
}
finally
{
//Release all unmanaged objects
Marshal.FreeHGlobal(ptrFgd);
if (medium.pUnkForRelease == null)
{
NativeMethods.ReleaseStgMedium(ref medium);
}
}
}
private DirectoryObject[] ProcessSelections(IntPtr dataObj)
{
if (dataObj == IntPtr.Zero)
{
throw new ArgumentNullException("dataObj");
}
DirectoryObject[] selections = null;
// The STGMEDIUM structure is a generalized global memory handle used for data transfer operations
NativeMethods.STGMEDIUM stg = new NativeMethods.STGMEDIUM();
stg.tymed = (uint)NativeMethods.TYMED.TYMED_HGLOBAL;
stg.hGlobal = IntPtr.Zero;
stg.pUnkForRelease = IntPtr.Zero;
// The FORMATETC structure is a generalized Clipboard format.
NativeMethods.FORMATETC fe = new NativeMethods.FORMATETC();
// The CFSTR_DSOP_DS_SELECTION_LIST clipboard format is provided by the IDataObject obtained by calling IDsObjectPicker::InvokeDialog
fe.cfFormat = (uint)DataFormats.GetFormat(NativeMethods.CLIPBOARD_FORMAT.CFSTR_DSOP_DS_SELECTION_LIST).Id;
fe.ptd = IntPtr.Zero;
// DVASPECT_CONTENT = 1
fe.dwAspect = 1;
// all of the data
fe.lindex = -1;
// The storage medium is a global memory handle (HGLOBAL)
fe.tymed = (uint)NativeMethods.TYMED.TYMED_HGLOBAL;
NativeMethods.DS_SELECTION[] dataArray = null;
int dsAttributesCount = 0;
NativeMethods.DS_SELECTION_ATTRIBUTES[] dsAttributes = null;
NativeMethods.IDataObject typedObjectForIUnknown = (NativeMethods.IDataObject)Marshal.GetTypedObjectForIUnknown(dataObj, typeof(NativeMethods.IDataObject));
if (typedObjectForIUnknown.GetData(ref fe, ref stg) == 0)
{
IntPtr ptr = NativeMethods.GlobalLock(stg.hGlobal);
try
{
NativeMethods.DS_SELECTION_LIST ds_selection_list = (NativeMethods.DS_SELECTION_LIST)Marshal.PtrToStructure(ptr, typeof(NativeMethods.DS_SELECTION_LIST));
int numberOfItems = (int)ds_selection_list.cItems;
dsAttributesCount = (int)ds_selection_list.cFetchedAttributes;
selections = new DirectoryObject[numberOfItems];
if (numberOfItems > 0)
{
dataArray = new NativeMethods.DS_SELECTION[numberOfItems];
dsAttributes = new NativeMethods.DS_SELECTION_ATTRIBUTES[numberOfItems];
IntPtr currentPtr = (IntPtr)(((long)ptr) + Marshal.SizeOf(ds_selection_list));
byte[] binaryForm = null;
for (int index = 0; index < numberOfItems; index++)
{
dataArray[index] = new NativeMethods.DS_SELECTION();
dataArray[index] = (NativeMethods.DS_SELECTION)Marshal.PtrToStructure(currentPtr, typeof(NativeMethods.DS_SELECTION));
if ((length > 0) && (dsAttributesCount > 0))
{
dsAttributes[index] = new NativeMethods.DS_SELECTION_ATTRIBUTES();
dsAttributes[index].attributeValues = Marshal.GetObjectsForNativeVariants(dataArray[index].pvarFetchedAttributes, dsAttributesCount);
binaryForm = dsAttributes[index].attributeValues[0] as byte[];
}
string schemaClassName = dataArray[index].pwzClass;
string name = dataArray[index].pwzName;
string upn = dataArray[index].pwzUPN;
string path = dataArray[index].pwzADsPath;
selections[index] = new DirectoryObject(schemaClassName, name, upn, path);
selections[index].BinarySid = binaryForm;
currentPtr = (IntPtr)(((long)currentPtr) + Marshal.SizeOf(dataArray[index]));
}
}
}
finally
{
NativeMethods.GlobalUnlock(stg.hGlobal);
}
}
return(selections);
}