dlls/msado15/stream.c

   1 /*
   2  * Copyright 2019 Hans Leidekker for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include <stdarg.h>
  20 #include "windef.h"
  21 #include "winbase.h"
  22 #define COBJMACROS
  23 #include "objbase.h"
  24 #include "msado15_backcompat.h"
  25
  26 #include "wine/debug.h"
  27 #include "wine/heap.h"
  28
  29 #include "msado15_private.h"
  30
  31 WINE_DEFAULT_DEBUG_CHANNEL(msado15);
  32
  33 struct stream
  34 {
  35     _Stream            Stream_iface;
  36     LONG               refs;
  37     ObjectStateEnum    state;
  38     ConnectModeEnum    mode;
  39     StreamTypeEnum     type;
  40     LineSeparatorEnum  sep;
  41     WCHAR             *charset;
  42     LONG               size;
  43     LONG               allocated;
  44     LONG               pos;
  45     BYTE              *buf;
  46 };
  47
  48 static inline struct stream *impl_from_Stream( _Stream *iface )
  49 {
  50     return CONTAINING_RECORD( iface, struct stream, Stream_iface );
  51 }
  52
  53 static ULONG WINAPI stream_AddRef( _Stream *iface )
  54 {
  55     struct stream *stream = impl_from_Stream( iface );
  56     return InterlockedIncrement( &stream->refs );
  57 }
  58
  59 static ULONG WINAPI stream_Release( _Stream *iface )
  60 {
  61     struct stream *stream = impl_from_Stream( iface );
  62     LONG refs = InterlockedDecrement( &stream->refs );
  63     if (!refs)
  64     {
  65         TRACE( "destroying %p\n", stream );
  66         heap_free( stream->charset );
  67         heap_free( stream->buf );
  68         heap_free( stream );
  69     }
  70     return refs;
  71 }
  72
  73 static HRESULT WINAPI stream_QueryInterface( _Stream *iface, REFIID riid, void **obj )
  74 {
  75     TRACE( "%p, %s, %p\n", iface, debugstr_guid(riid), obj );
  76
  77     if (IsEqualGUID( riid, &IID__Stream ) || IsEqualGUID( riid, &IID_IDispatch ) ||
  78         IsEqualGUID( riid, &IID_IUnknown ))
  79     {
  80         *obj = iface;
  81     }
  82     else
  83     {
  84         FIXME( "interface %s not implemented\n", debugstr_guid(riid) );
  85         return E_NOINTERFACE;
  86     }
  87     stream_AddRef( iface );
  88     return S_OK;
  89 }
  90
  91 static HRESULT WINAPI stream_GetTypeInfoCount( _Stream *iface, UINT *count )
  92 {
  93     FIXME( "%p, %p\n", iface, count );
  94     return E_NOTIMPL;
  95 }
  96
  97 static HRESULT WINAPI stream_GetTypeInfo( _Stream *iface, UINT index, LCID lcid, ITypeInfo **info )
  98 {
  99     FIXME( "%p, %u, %u, %p\n", iface, index, lcid, info );
 100     return E_NOTIMPL;
 101 }
 102
 103 static HRESULT WINAPI stream_GetIDsOfNames( _Stream *iface, REFIID riid, LPOLESTR *names, UINT count,
 104                                             LCID lcid, DISPID *dispid )
 105 {
 106     FIXME( "%p, %s, %p, %u, %u, %p\n", iface, debugstr_guid(riid), names, count, lcid, dispid );
 107     return E_NOTIMPL;
 108 }
 109
 110 static HRESULT WINAPI stream_Invoke( _Stream *iface, DISPID member, REFIID riid, LCID lcid, WORD flags,
 111                                      DISPPARAMS *params, VARIANT *result, EXCEPINFO *excep_info, UINT *arg_err )
 112 {
 113     FIXME( "%p, %d, %s, %d, %d, %p, %p, %p, %p\n", iface, member, debugstr_guid(riid), lcid, flags, params,
 114            result, excep_info, arg_err );
 115     return E_NOTIMPL;
 116 }
 117
 118 static HRESULT WINAPI stream_get_Size( _Stream *iface, LONG *size )
 119 {
 120     struct stream *stream = impl_from_Stream( iface );
 121     TRACE( "%p, %p\n", stream, size );
 122
 123     if (stream->state == adStateClosed) return MAKE_ADO_HRESULT( adErrObjectClosed );
 124
 125     *size = stream->size;
 126     return S_OK;
 127 }
 128
 129 static HRESULT WINAPI stream_get_EOS( _Stream *iface, VARIANT_BOOL *eos )
 130 {
 131     struct stream *stream = impl_from_Stream( iface );
 132     TRACE( "%p, %p\n", stream, eos );
 133
 134     if (stream->state == adStateClosed) return MAKE_ADO_HRESULT( adErrObjectClosed );
 135
 136     *eos = (stream->pos >= stream->size) ? VARIANT_TRUE : VARIANT_FALSE;
 137     return S_OK;
 138 }
 139
 140 static HRESULT WINAPI stream_get_Position( _Stream *iface, LONG *pos )
 141 {
 142     struct stream *stream = impl_from_Stream( iface );
 143     TRACE( "%p, %p\n", stream, pos );
 144
 145     if (stream->state == adStateClosed) return MAKE_ADO_HRESULT( adErrObjectClosed );
 146
 147     *pos = stream->pos;
 148     return S_OK;
 149 }
 150
 151 static HRESULT resize_buffer( struct stream *stream, LONG size )
 152 {
 153     if (stream->allocated < size)
 154     {
 155         BYTE *tmp;
 156         LONG new_size = max( size, stream->allocated * 2 );
 157         if (!(tmp = heap_realloc_zero( stream->buf, new_size ))) return E_OUTOFMEMORY;
 158         stream->buf = tmp;
 159         stream->allocated = new_size;
 160     }
 161     stream->size = size;
 162     return S_OK;
 163 }
 164
 165 static HRESULT WINAPI stream_put_Position( _Stream *iface, LONG pos )
 166 {
 167     struct stream *stream = impl_from_Stream( iface );
 168     HRESULT hr;
 169
 170     TRACE( "%p, %d\n", stream, pos );
 171
 172     if (stream->state == adStateClosed) return MAKE_ADO_HRESULT( adErrObjectClosed );
 173     if (pos < 0) return MAKE_ADO_HRESULT( adErrInvalidArgument );
 174
 175     if ((hr = resize_buffer( stream, stream->pos )) != S_OK) return hr;
 176     stream->pos = pos;
 177     return S_OK;
 178 }
 179
 180 static HRESULT WINAPI stream_get_Type( _Stream *iface, StreamTypeEnum *type )
 181 {
 182     struct stream *stream = impl_from_Stream( iface );
 183     TRACE( "%p, %p\n", stream, type );
 184
 185     *type = stream->type;
 186     return S_OK;
 187 }
 188
 189 static HRESULT WINAPI stream_put_Type( _Stream *iface, StreamTypeEnum type )
 190 {
 191     struct stream *stream = impl_from_Stream( iface );
 192     TRACE( "%p, %u\n", stream, type );
 193
 194     stream->type = type;
 195     return S_OK;
 196 }
 197
 198 static HRESULT WINAPI stream_get_LineSeparator( _Stream *iface, LineSeparatorEnum *sep )
 199 {
 200     struct stream *stream = impl_from_Stream( iface );
 201     TRACE( "%p, %p\n", stream, sep );
 202
 203     *sep = stream->sep;
 204     return S_OK;
 205 }
 206
 207 static HRESULT WINAPI stream_put_LineSeparator( _Stream *iface, LineSeparatorEnum sep )
 208 {
 209     struct stream *stream = impl_from_Stream( iface );
 210     TRACE( "%p, %d\n", stream, sep );
 211
 212     stream->sep = sep;
 213     return S_OK;
 214 }
 215
 216 static HRESULT WINAPI stream_get_State( _Stream *iface, ObjectStateEnum *state )
 217 {
 218     struct stream *stream = impl_from_Stream( iface );
 219     TRACE( "%p, %p\n", stream, state );
 220
 221     *state = stream->state;
 222     return S_OK;
 223 }
 224
 225 static HRESULT WINAPI stream_get_Mode( _Stream *iface, ConnectModeEnum *mode )
 226 {
 227     struct stream *stream = impl_from_Stream( iface );
 228     TRACE( "%p, %p\n", stream, mode );
 229
 230     *mode = stream->mode;
 231     return S_OK;
 232 }
 233
 234 static HRESULT WINAPI stream_put_Mode( _Stream *iface, ConnectModeEnum mode )
 235 {
 236     struct stream *stream = impl_from_Stream( iface );
 237     TRACE( "%p, %u\n", stream, mode );
 238
 239     stream->mode = mode;
 240     return S_OK;
 241 }
 242
 243 static HRESULT WINAPI stream_get_Charset( _Stream *iface, BSTR *charset )
 244 {
 245     struct stream *stream = impl_from_Stream( iface );
 246     const WCHAR *src = stream->charset ? stream->charset : L"Unicode";
 247     BSTR dst;
 248
 249     TRACE( "%p, %p\n", stream, charset );
 250
 251     if (!(dst = SysAllocString( src ))) return E_OUTOFMEMORY;
 252     *charset = dst;
 253     return S_OK;
 254 }
 255
 256 static HRESULT WINAPI stream_put_Charset( _Stream *iface, BSTR charset )
 257 {
 258     struct stream *stream = impl_from_Stream( iface );
 259     WCHAR *str;
 260
 261     TRACE( "%p, %s\n", stream, debugstr_w(charset) );
 262
 263     if (!(str = strdupW( charset ))) return E_OUTOFMEMORY;
 264     heap_free( stream->charset );
 265     stream->charset = str;
 266     return S_OK;
 267 }
 268
 269 static HRESULT create_byte_array( BYTE *data, LONG len, VARIANT *ret )
 270 {
 271     SAFEARRAY *vector;
 272     LONG i;
 273     HRESULT hr;
 274
 275     if (!len)
 276     {
 277         V_VT( ret ) = VT_NULL;
 278         return S_OK;
 279     }
 280     if (!(vector = SafeArrayCreateVector( VT_UI1, 0, len ))) return E_OUTOFMEMORY;
 281     for (i = 0; i < len; i++)
 282     {
 283         if ((hr = SafeArrayPutElement( vector, &i, &data[i] )) != S_OK)
 284         {
 285             SafeArrayDestroy( vector );
 286             return hr;
 287         }
 288     }
 289
 290     V_VT( ret ) = VT_ARRAY | VT_UI1;
 291     V_ARRAY( ret ) = vector;
 292     return S_OK;
 293 }
 294
 295 static HRESULT WINAPI stream_Read( _Stream *iface, LONG size, VARIANT *val )
 296 {
 297     struct stream *stream = impl_from_Stream( iface );
 298     HRESULT hr;
 299
 300     TRACE( "%p, %d, %p\n", stream, size, val );
 301
 302     if (stream->state == adStateClosed) return MAKE_ADO_HRESULT( adErrObjectClosed );
 303     if (stream->type != adTypeBinary) return MAKE_ADO_HRESULT( adErrIllegalOperation );
 304     if (size < adReadAll) return MAKE_ADO_HRESULT( adErrInvalidArgument );
 305
 306     if (size == adReadAll) size = stream->size - stream->pos;
 307     else size = min( size, stream->size - stream->pos );
 308
 309     if ((hr = create_byte_array( stream->buf + stream->pos, size, val )) != S_OK) return hr;
 310     stream->pos += size;
 311     return S_OK;
 312 }
 313
 314 static HRESULT WINAPI stream_Open( _Stream *iface, VARIANT src, ConnectModeEnum mode, StreamOpenOptionsEnum options,
 315                                     BSTR username, BSTR password )
 316 {
 317     struct stream *stream = impl_from_Stream( iface );
 318     FIXME( "%p, %s, %u, %d, %s, %p\n", stream, debugstr_variant(&src), mode, options, debugstr_w(username), password );
 319
 320     if (stream->state == adStateOpen) return MAKE_ADO_HRESULT( adErrObjectOpen );
 321
 322     stream->state = adStateOpen;
 323     return S_OK;
 324 }
 325
 326 static HRESULT WINAPI stream_Close( _Stream *iface )
 327 {
 328     struct stream *stream = impl_from_Stream( iface );
 329
 330     TRACE( "%p\n", stream );
 331
 332     if (stream->state == adStateClosed) return MAKE_ADO_HRESULT( adErrObjectClosed );
 333
 334     heap_free( stream->buf );
 335     stream->buf  = NULL;
 336     stream->size = stream->allocated = stream->pos = 0;
 337
 338     stream->state = adStateClosed;
 339     return S_OK;
 340 }
 341
 342 static HRESULT WINAPI stream_SkipLine( _Stream *iface )
 343 {
 344     FIXME( "%p\n", iface );
 345     return E_NOTIMPL;
 346 }
 347
 348 static HRESULT WINAPI stream_Write( _Stream *iface, VARIANT buf )
 349 {
 350     struct stream *stream = impl_from_Stream( iface );
 351     LONG bound, i;
 352     HRESULT hr;
 353
 354     TRACE( "%p, %s\n", stream, debugstr_variant(&buf) );
 355
 356     if (stream->state == adStateClosed) return MAKE_ADO_HRESULT( adErrObjectClosed );
 357     if (stream->type != adTypeBinary) return MAKE_ADO_HRESULT( adErrIllegalOperation );
 358     if (V_VT( &buf ) != (VT_ARRAY | VT_UI1)) return MAKE_ADO_HRESULT( adErrInvalidArgument );
 359
 360     if ((hr = SafeArrayGetUBound( V_ARRAY( &buf ), 1, &bound )) != S_OK) return hr;
 361     if ((hr = resize_buffer( stream, stream->size + bound + 1 )) != S_OK) return hr;
 362
 363     for (i = 0; i <= bound; i++)
 364     {
 365         if ((hr = SafeArrayGetElement( V_ARRAY( &buf ), &i, &stream->buf[stream->pos++] )) != S_OK) return hr;
 366     }
 367
 368     return S_OK;
 369 }
 370
 371 static HRESULT WINAPI stream_SetEOS( _Stream *iface )
 372 {
 373     struct stream *stream = impl_from_Stream( iface );
 374
 375     TRACE( "%p\n", stream );
 376
 377     return resize_buffer( stream, stream->pos );
 378 }
 379
 380 static HRESULT WINAPI stream_CopyTo( _Stream *iface, _Stream *dst, LONG size )
 381 {
 382     FIXME( "%p, %p, %d\n", iface, dst, size );
 383     return E_NOTIMPL;
 384 }
 385
 386 static HRESULT WINAPI stream_Flush( _Stream *iface )
 387 {
 388     FIXME( "%p\n", iface );
 389     return E_NOTIMPL;
 390 }
 391
 392 static HRESULT WINAPI stream_SaveToFile( _Stream *iface, BSTR filename, SaveOptionsEnum options )
 393 {
 394     FIXME( "%p, %s, %u\n", iface, debugstr_w(filename), options );
 395     return E_NOTIMPL;
 396 }
 397
 398 static HRESULT WINAPI stream_LoadFromFile( _Stream *iface, BSTR filename )
 399 {
 400     FIXME( "%p, %s\n", iface, debugstr_w(filename) );
 401     return E_NOTIMPL;
 402 }
 403
 404 static HRESULT WINAPI stream_ReadText( _Stream *iface, LONG len, BSTR *ret )
 405 {
 406     struct stream *stream = impl_from_Stream( iface );
 407     BSTR str;
 408
 409     TRACE( "%p, %d, %p\n", stream, len, ret );
 410     if (len == adReadLine)
 411     {
 412         FIXME( "adReadLine not supported\n" );
 413         return E_NOTIMPL;
 414     }
 415     if (stream->charset && wcscmp( stream->charset, L"Unicode" ))
 416     {
 417         FIXME( "charset %s not supported\n", debugstr_w(stream->charset) );
 418         return E_NOTIMPL;
 419     }
 420
 421     if (stream->type != adTypeText) return MAKE_ADO_HRESULT( adErrIllegalOperation );
 422     if (len < adReadLine) return MAKE_ADO_HRESULT( adErrInvalidArgument );
 423
 424     if (len == adReadAll) len = (stream->size - stream->pos) / sizeof(WCHAR);
 425     else len = min( len, stream->size - stream->pos / sizeof(WCHAR) );
 426
 427     if (!(str = SysAllocStringLen( NULL, len ))) return E_OUTOFMEMORY;
 428     memcpy( str, stream->buf + stream->pos, len * sizeof(WCHAR) );
 429     str[len] = 0;
 430
 431     stream->pos += len * sizeof(WCHAR);
 432
 433     *ret = str;
 434     return S_OK;
 435 }
 436
 437 static HRESULT WINAPI stream_WriteText( _Stream *iface, BSTR data, StreamWriteEnum options )
 438 {
 439     struct stream *stream = impl_from_Stream( iface );
 440     HRESULT hr;
 441     LONG size;
 442
 443     TRACE( "%p, %s, %u\n", stream, debugstr_w(data), options );
 444     if (options != adWriteChar)
 445     {
 446         FIXME( "options %u not supported\n", options );
 447         return E_NOTIMPL;
 448     }
 449     if (stream->charset && wcscmp( stream->charset, L"Unicode" ))
 450     {
 451         FIXME( "charset %s not supported\n", debugstr_w(stream->charset) );
 452         return E_NOTIMPL;
 453     }
 454
 455     if (stream->type != adTypeText) return MAKE_ADO_HRESULT( adErrIllegalOperation );
 456
 457     size = (lstrlenW( data ) + 1) * sizeof(WCHAR);
 458     if ((hr = resize_buffer( stream, stream->size + size )) != S_OK) return hr;
 459
 460     memcpy( stream->buf + stream->pos, data, size );
 461     stream->pos += size;
 462
 463     return S_OK;
 464 }
 465
 466 static HRESULT WINAPI stream_Cancel( _Stream *iface )
 467 {
 468     FIXME( "%p\n", iface );
 469     return E_NOTIMPL;
 470 }
 471
 472 static const struct _StreamVtbl stream_vtbl =
 473 {
 474     stream_QueryInterface,
 475     stream_AddRef,
 476     stream_Release,
 477     stream_GetTypeInfoCount,
 478     stream_GetTypeInfo,
 479     stream_GetIDsOfNames,
 480     stream_Invoke,
 481     stream_get_Size,
 482     stream_get_EOS,
 483     stream_get_Position,
 484     stream_put_Position,
 485     stream_get_Type,
 486     stream_put_Type,
 487     stream_get_LineSeparator,
 488     stream_put_LineSeparator,
 489     stream_get_State,
 490     stream_get_Mode,
 491     stream_put_Mode,
 492     stream_get_Charset,
 493     stream_put_Charset,
 494     stream_Read,
 495     stream_Open,
 496     stream_Close,
 497     stream_SkipLine,
 498     stream_Write,
 499     stream_SetEOS,
 500     stream_CopyTo,
 501     stream_Flush,
 502     stream_SaveToFile,
 503     stream_LoadFromFile,
 504     stream_ReadText,
 505     stream_WriteText,
 506     stream_Cancel
 507 };
 508
 509 HRESULT Stream_create( void **obj )
 510 {
 511     struct stream *stream;
 512
 513     if (!(stream = heap_alloc_zero( sizeof(*stream) ))) return E_OUTOFMEMORY;
 514     stream->Stream_iface.lpVtbl = &stream_vtbl;
 515     stream->refs = 1;
 516     stream->type = adTypeText;
 517     stream->sep  = adCRLF;
 518
 519     *obj = &stream->Stream_iface;
 520     TRACE( "returning iface %p\n", *obj );
 521     return S_OK;
 522 }