streams.c 16.3 KB
Newer Older
1 2 3 4
/*
 * Implementation of the Microsoft Installer (msi.dll)
 *
 * Copyright 2007 James Hawkins
5
 * Copyright 2015 Hans Leidekker for CodeWeavers
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32
 *
 * 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
 */

#include <stdarg.h>

#define COBJMACROS

#include "windef.h"
#include "winbase.h"
#include "winerror.h"
#include "msi.h"
#include "msiquery.h"
#include "objbase.h"
#include "msipriv.h"
33
#include "query.h"
34 35

#include "wine/debug.h"
36
#include "wine/unicode.h"
37 38 39 40 41 42 43 44 45

WINE_DEFAULT_DEBUG_CHANNEL(msidb);

#define NUM_STREAMS_COLS    2

typedef struct tagMSISTREAMSVIEW
{
    MSIVIEW view;
    MSIDATABASE *db;
46
    UINT num_cols;
47 48
} MSISTREAMSVIEW;

49
static BOOL streams_resize_table( MSIDATABASE *db, UINT size )
50
{
51
    if (!db->num_streams_allocated)
52
    {
53 54 55
        if (!(db->streams = msi_alloc_zero( size * sizeof(MSISTREAM) ))) return FALSE;
        db->num_streams_allocated = size;
        return TRUE;
56
    }
57
    while (size >= db->num_streams_allocated)
58
    {
59 60 61 62 63
        MSISTREAM *tmp;
        UINT new_size = db->num_streams_allocated * 2;
        if (!(tmp = msi_realloc_zero( db->streams, new_size * sizeof(MSISTREAM) ))) return FALSE;
        db->streams = tmp;
        db->num_streams_allocated = new_size;
64
    }
65
    return TRUE;
66 67 68 69 70 71 72 73 74 75 76
}

static UINT STREAMS_fetch_int(struct tagMSIVIEW *view, UINT row, UINT col, UINT *val)
{
    MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;

    TRACE("(%p, %d, %d, %p)\n", view, row, col, val);

    if (col != 1)
        return ERROR_INVALID_PARAMETER;

77
    if (row >= sv->db->num_streams)
78 79
        return ERROR_NO_MORE_ITEMS;

80
    *val = sv->db->streams[row].str_index;
81 82 83 84 85 86 87

    return ERROR_SUCCESS;
}

static UINT STREAMS_fetch_stream(struct tagMSIVIEW *view, UINT row, UINT col, IStream **stm)
{
    MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;
88 89
    LARGE_INTEGER pos;
    HRESULT hr;
90 91 92

    TRACE("(%p, %d, %d, %p)\n", view, row, col, stm);

93
    if (row >= sv->db->num_streams)
94 95
        return ERROR_FUNCTION_FAILED;

96 97 98 99 100 101 102
    pos.QuadPart = 0;
    hr = IStream_Seek( sv->db->streams[row].stream, pos, STREAM_SEEK_SET, NULL );
    if (FAILED( hr ))
        return ERROR_FUNCTION_FAILED;

    *stm = sv->db->streams[row].stream;
    IStream_AddRef( *stm );
103 104 105 106

    return ERROR_SUCCESS;
}

107 108 109 110
static UINT STREAMS_get_row( struct tagMSIVIEW *view, UINT row, MSIRECORD **rec )
{
    MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;

111
    TRACE("%p %d %p\n", sv, row, rec);
112

113
    return msi_view_get_row( sv->db, view, row, rec );
114 115
}

116 117 118 119
static UINT STREAMS_set_row(struct tagMSIVIEW *view, UINT row, MSIRECORD *rec, UINT mask)
{
    MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;

120
    TRACE("(%p, %d, %p, %08x)\n", view, row, rec, mask);
121

122 123
    if (row > sv->db->num_streams || mask >= (1 << sv->num_cols))
        return ERROR_INVALID_PARAMETER;
124

125
    if (mask & 1)
126
    {
127
        const WCHAR *name = MSI_RecordGetString( rec, 1 );
128

129 130
        if (!name) return ERROR_INVALID_PARAMETER;
        sv->db->streams[row].str_index = msi_add_string( sv->db->strings, name, -1, StringNonPersistent );
131
    }
132
    if (mask & 2)
133
    {
134 135 136
        IStream *old, *new;
        HRESULT hr;
        UINT r;
137

138 139 140
        r = MSI_RecordGetIStream( rec, 2, &new );
        if (r != ERROR_SUCCESS)
            return r;
141

142 143 144 145 146 147 148 149
        old = sv->db->streams[row].stream;
        hr = IStream_QueryInterface( new, &IID_IStream, (void **)&sv->db->streams[row].stream );
        if (FAILED( hr ))
        {
            IStream_Release( new );
            return ERROR_FUNCTION_FAILED;
        }
        if (old) IStream_Release( old );
150 151
    }

152 153
    return ERROR_SUCCESS;
}
154

155 156 157 158
static UINT streams_find_row( MSISTREAMSVIEW *sv, MSIRECORD *rec, UINT *row )
{
    const WCHAR *str;
    UINT r, i, id, val;
159

160 161 162 163
    str = MSI_RecordGetString( rec, 1 );
    r = msi_string2id( sv->db->strings, str, -1, &id );
    if (r != ERROR_SUCCESS)
        return r;
164

165 166 167
    for (i = 0; i < sv->db->num_streams; i++)
    {
        STREAMS_fetch_int( &sv->view, i, 1, &val );
168

169 170 171 172 173 174
        if (val == id)
        {
            if (row) *row = i;
            return ERROR_SUCCESS;
        }
    }
175

176
    return ERROR_FUNCTION_FAILED;
177 178
}

179
static UINT STREAMS_insert_row(struct tagMSIVIEW *view, MSIRECORD *rec, UINT row, BOOL temporary)
180 181
{
    MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;
182
    UINT i, r, num_rows = sv->db->num_streams + 1;
183

184 185
    TRACE("(%p, %p, %d, %d)\n", view, rec, row, temporary);

186 187 188 189 190
    r = streams_find_row( sv, rec, NULL );
    if (r == ERROR_SUCCESS)
        return ERROR_FUNCTION_FAILED;

    if (!streams_resize_table( sv->db, num_rows ))
191 192
        return ERROR_FUNCTION_FAILED;

193
    if (row == -1)
194
        row = num_rows - 1;
195

196
    /* shift the rows to make room for the new row */
197
    for (i = num_rows - 1; i > row; i--)
198
    {
199
        sv->db->streams[i] = sv->db->streams[i - 1];
200
    }
201

202 203 204 205 206
    r = STREAMS_set_row( view, row, rec, (1 << sv->num_cols) - 1 );
    if (r == ERROR_SUCCESS)
        sv->db->num_streams = num_rows;

    return r;
207 208
}

209 210 211 212 213 214
static UINT STREAMS_delete_row(struct tagMSIVIEW *view, UINT row)
{
    FIXME("(%p %d): stub!\n", view, row);
    return ERROR_SUCCESS;
}

215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232
static UINT STREAMS_execute(struct tagMSIVIEW *view, MSIRECORD *record)
{
    TRACE("(%p, %p)\n", view, record);
    return ERROR_SUCCESS;
}

static UINT STREAMS_close(struct tagMSIVIEW *view)
{
    TRACE("(%p)\n", view);
    return ERROR_SUCCESS;
}

static UINT STREAMS_get_dimensions(struct tagMSIVIEW *view, UINT *rows, UINT *cols)
{
    MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;

    TRACE("(%p, %p, %p)\n", view, rows, cols);

233 234
    if (cols) *cols = sv->num_cols;
    if (rows) *rows = sv->db->num_streams;
235 236 237 238

    return ERROR_SUCCESS;
}

239 240
static UINT STREAMS_get_column_info( struct tagMSIVIEW *view, UINT n, LPCWSTR *name,
                                     UINT *type, BOOL *temporary, LPCWSTR *table_name )
241
{
242 243 244
    MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;

    TRACE("(%p, %d, %p, %p, %p, %p)\n", view, n, name, type, temporary, table_name);
245

246
    if (!n || n > sv->num_cols)
247 248 249 250 251
        return ERROR_INVALID_PARAMETER;

    switch (n)
    {
    case 1:
252
        if (name) *name = szName;
253
        if (type) *type = MSITYPE_STRING | MSITYPE_VALID | MAX_STREAM_NAME_LEN;
254 255 256
        break;

    case 2:
257
        if (name) *name = szData;
258 259 260
        if (type) *type = MSITYPE_STRING | MSITYPE_VALID | MSITYPE_NULLABLE;
        break;
    }
261 262
    if (table_name) *table_name = szStreams;
    if (temporary) *temporary = FALSE;
263 264 265
    return ERROR_SUCCESS;
}

266 267 268 269 270 271 272 273 274
static UINT streams_modify_update(struct tagMSIVIEW *view, MSIRECORD *rec)
{
    MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;
    UINT r, row;

    r = streams_find_row(sv, rec, &row);
    if (r != ERROR_SUCCESS)
        return ERROR_FUNCTION_FAILED;

275
    return STREAMS_set_row( view, row, rec, (1 << sv->num_cols) - 1 );
276 277 278 279 280
}

static UINT streams_modify_assign(struct tagMSIVIEW *view, MSIRECORD *rec)
{
    MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;
281
    UINT r;
282

283
    r = streams_find_row( sv, rec, NULL );
284 285 286
    if (r == ERROR_SUCCESS)
        return streams_modify_update(view, rec);

287
    return STREAMS_insert_row(view, rec, -1, FALSE);
288 289
}

290
static UINT STREAMS_modify(struct tagMSIVIEW *view, MSIMODIFY eModifyMode, MSIRECORD *rec, UINT row)
291
{
292 293 294 295 296 297
    UINT r;

    TRACE("%p %d %p\n", view, eModifyMode, rec);

    switch (eModifyMode)
    {
298 299 300 301
    case MSIMODIFY_ASSIGN:
        r = streams_modify_assign(view, rec);
        break;

302
    case MSIMODIFY_INSERT:
303
        r = STREAMS_insert_row(view, rec, -1, FALSE);
304 305
        break;

306 307 308 309
    case MSIMODIFY_UPDATE:
        r = streams_modify_update(view, rec);
        break;

310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327
    case MSIMODIFY_VALIDATE_NEW:
    case MSIMODIFY_INSERT_TEMPORARY:
    case MSIMODIFY_REFRESH:
    case MSIMODIFY_REPLACE:
    case MSIMODIFY_MERGE:
    case MSIMODIFY_DELETE:
    case MSIMODIFY_VALIDATE:
    case MSIMODIFY_VALIDATE_FIELD:
    case MSIMODIFY_VALIDATE_DELETE:
        FIXME("%p %d %p - mode not implemented\n", view, eModifyMode, rec );
        r = ERROR_CALL_NOT_IMPLEMENTED;
        break;

    default:
        r = ERROR_INVALID_DATA;
    }

    return r;
328 329 330 331 332 333 334 335
}

static UINT STREAMS_delete(struct tagMSIVIEW *view)
{
    MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;

    TRACE("(%p)\n", view);

336
    msi_free(sv);
337 338 339 340 341 342 343
    return ERROR_SUCCESS;
}

static UINT STREAMS_find_matching_rows(struct tagMSIVIEW *view, UINT col,
                                       UINT val, UINT *row, MSIITERHANDLE *handle)
{
    MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;
344
    UINT index = PtrToUlong(*handle);
345

346
    TRACE("(%p, %d, %d, %p, %p)\n", view, col, val, row, handle);
347

348
    if (!col || col > sv->num_cols)
349 350
        return ERROR_INVALID_PARAMETER;

351
    while (index < sv->db->num_streams)
352
    {
353
        if (sv->db->streams[index].str_index == val)
354 355 356 357 358 359 360
        {
            *row = index;
            break;
        }
        index++;
    }

361
    *handle = UlongToPtr(++index);
362

363
    if (index > sv->db->num_streams)
364 365 366 367 368 369 370 371 372
        return ERROR_NO_MORE_ITEMS;

    return ERROR_SUCCESS;
}

static const MSIVIEWOPS streams_ops =
{
    STREAMS_fetch_int,
    STREAMS_fetch_stream,
373
    STREAMS_get_row,
374 375
    STREAMS_set_row,
    STREAMS_insert_row,
376
    STREAMS_delete_row,
377 378 379 380 381 382
    STREAMS_execute,
    STREAMS_close,
    STREAMS_get_dimensions,
    STREAMS_get_column_info,
    STREAMS_modify,
    STREAMS_delete,
383 384 385
    STREAMS_find_matching_rows,
    NULL,
    NULL,
386
    NULL,
387
    NULL,
388
    NULL,
389
    NULL,
390 391
};

392
static HRESULT open_stream( MSIDATABASE *db, const WCHAR *name, IStream **stream )
393 394 395
{
    HRESULT hr;

396 397
    hr = IStorage_OpenStream( db->storage, name, NULL, STGM_READ|STGM_SHARE_EXCLUSIVE, 0, stream );
    if (FAILED( hr ))
398
    {
399
        MSITRANSFORM *transform;
400

401
        LIST_FOR_EACH_ENTRY( transform, &db->transforms, MSITRANSFORM, entry )
Hans Leidekker's avatar
Hans Leidekker committed
402
        {
403 404 405
            hr = IStorage_OpenStream( transform->stg, name, NULL, STGM_READ|STGM_SHARE_EXCLUSIVE, 0, stream );
            if (SUCCEEDED( hr ))
                break;
Hans Leidekker's avatar
Hans Leidekker committed
406
        }
407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432
    }
    return hr;
}

static MSISTREAM *find_stream( MSIDATABASE *db, const WCHAR *name )
{
    UINT r, id, i;

    r = msi_string2id( db->strings, name, -1, &id );
    if (r != ERROR_SUCCESS)
        return NULL;

    for (i = 0; i < db->num_streams; i++)
    {
        if (db->streams[i].str_index == id) return &db->streams[i];
    }
    return NULL;
}

static UINT append_stream( MSIDATABASE *db, const WCHAR *name, IStream *stream )
{
    UINT i = db->num_streams;

    if (!streams_resize_table( db, db->num_streams + 1 ))
        return ERROR_OUTOFMEMORY;

433
    db->streams[i].str_index = msi_add_string( db->strings, name, -1, StringNonPersistent );
434 435 436
    db->streams[i].stream = stream;
    db->num_streams++;

437
    TRACE("added %s\n", debugstr_w( name ));
438 439 440 441 442
    return ERROR_SUCCESS;
}

static UINT load_streams( MSIDATABASE *db )
{
443
    WCHAR decoded[MAX_STREAM_NAME_LEN + 1];
444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459
    IEnumSTATSTG *stgenum;
    STATSTG stat;
    HRESULT hr;
    UINT count, r = ERROR_SUCCESS;
    IStream *stream;

    hr = IStorage_EnumElements( db->storage, 0, NULL, 0, &stgenum );
    if (FAILED( hr ))
        return ERROR_FUNCTION_FAILED;

    for (;;)
    {
        count = 0;
        hr = IEnumSTATSTG_Next( stgenum, 1, &stat, &count );
        if (FAILED( hr ) || !count)
            break;
460

461
        /* table streams are not in the _Streams table */
462
        if (stat.type != STGTY_STREAM || *stat.pwcsName == 0x4840)
463
        {
464
            CoTaskMemFree( stat.pwcsName );
465
            continue;
466
        }
467 468 469 470 471 472 473
        decode_streamname( stat.pwcsName, decoded );
        if (find_stream( db, decoded ))
        {
            CoTaskMemFree( stat.pwcsName );
            continue;
        }
        TRACE("found new stream %s\n", debugstr_w( decoded ));
474

475
        hr = open_stream( db, stat.pwcsName, &stream );
476
        CoTaskMemFree( stat.pwcsName );
477
        if (FAILED( hr ))
478
        {
479 480
            ERR("unable to open stream %08x\n", hr);
            r = ERROR_FUNCTION_FAILED;
481 482 483
            break;
        }

484
        r = append_stream( db, decoded, stream );
485
        if (r != ERROR_SUCCESS)
486
            break;
487
    }
488

489 490 491 492 493 494 495 496 497 498 499
    TRACE("loaded %u streams\n", db->num_streams);
    IEnumSTATSTG_Release( stgenum );
    return r;
}

UINT msi_get_stream( MSIDATABASE *db, const WCHAR *name, IStream **ret )
{
    MSISTREAM *stream;
    WCHAR *encname;
    HRESULT hr;
    UINT r;
500

501 502 503 504 505 506 507 508 509 510 511 512
    if ((stream = find_stream( db, name )))
    {
        LARGE_INTEGER pos;

        pos.QuadPart = 0;
        hr = IStream_Seek( stream->stream, pos, STREAM_SEEK_SET, NULL );
        if (FAILED( hr ))
            return ERROR_FUNCTION_FAILED;

        *ret = stream->stream;
        IStream_AddRef( *ret );
        return ERROR_SUCCESS;
513 514
    }

515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531
    if (!(encname = encode_streamname( FALSE, name )))
        return ERROR_OUTOFMEMORY;

    hr = open_stream( db, encname, ret );
    msi_free( encname );
    if (FAILED( hr ))
        return ERROR_FUNCTION_FAILED;

    r = append_stream( db, name, *ret );
    if (r != ERROR_SUCCESS)
    {
        IStream_Release( *ret );
        return r;
    }

    IStream_AddRef( *ret );
    return ERROR_SUCCESS;
532 533 534 535 536
}

UINT STREAMS_CreateView(MSIDATABASE *db, MSIVIEW **view)
{
    MSISTREAMSVIEW *sv;
537
    UINT r;
538 539 540

    TRACE("(%p, %p)\n", db, view);

541 542 543 544 545 546
    r = load_streams( db );
    if (r != ERROR_SUCCESS)
        return r;

    if (!(sv = msi_alloc_zero( sizeof(MSISTREAMSVIEW) )))
        return ERROR_OUTOFMEMORY;
547 548

    sv->view.ops = &streams_ops;
549
    sv->num_cols = NUM_STREAMS_COLS;
550
    sv->db = db;
551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575

    *view = (MSIVIEW *)sv;

    return ERROR_SUCCESS;
}

static HRESULT write_stream( IStream *dst, IStream *src )
{
    HRESULT hr;
    char buf[4096];
    STATSTG stat;
    LARGE_INTEGER pos;
    UINT count, size;

    hr = IStream_Stat( src, &stat, STATFLAG_NONAME );
    if (FAILED( hr )) return hr;

    hr = IStream_SetSize( dst, stat.cbSize );
    if (FAILED( hr )) return hr;

    pos.QuadPart = 0;
    hr = IStream_Seek( dst, pos, STREAM_SEEK_SET, NULL );
    if (FAILED( hr )) return hr;

    for (;;)
576
    {
577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595
        size = min( sizeof(buf), stat.cbSize.QuadPart );
        hr = IStream_Read( src, buf, size, &count );
        if (FAILED( hr ) || count != size)
        {
            WARN("failed to read stream: %08x\n", hr);
            return E_INVALIDARG;
        }
        stat.cbSize.QuadPart -= count;
        if (count)
        {
            size = count;
            hr = IStream_Write( dst, buf, size, &count );
            if (FAILED( hr ) || count != size)
            {
                WARN("failed to write stream: %08x\n", hr);
                return E_INVALIDARG;
            }
        }
        if (!stat.cbSize.QuadPart) break;
596
    }
597

598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615
    return S_OK;
}

UINT msi_commit_streams( MSIDATABASE *db )
{
    UINT i;
    const WCHAR *name;
    WCHAR *encname;
    IStream *stream;
    HRESULT hr;

    TRACE("got %u streams\n", db->num_streams);

    for (i = 0; i < db->num_streams; i++)
    {
        name = msi_string_lookup( db->strings, db->streams[i].str_index, NULL );
        if (!(encname = encode_streamname( FALSE, name ))) return ERROR_OUTOFMEMORY;

616 617
        hr = IStorage_CreateStream( db->storage, encname, STGM_WRITE|STGM_SHARE_EXCLUSIVE, 0, 0, &stream );
        if (SUCCEEDED( hr ))
618
        {
619
            hr = write_stream( stream, db->streams[i].stream );
620 621
            if (FAILED( hr ))
            {
622
                ERR("failed to write stream %s (hr = %08x)\n", debugstr_w(encname), hr);
623
                msi_free( encname );
624
                IStream_Release( stream );
625 626
                return ERROR_FUNCTION_FAILED;
            }
627 628
            hr = IStream_Commit( stream, 0 );
            IStream_Release( stream );
629 630
            if (FAILED( hr ))
            {
631
                ERR("failed to commit stream %s (hr = %08x)\n", debugstr_w(encname), hr);
632 633 634 635
                msi_free( encname );
                return ERROR_FUNCTION_FAILED;
            }
        }
636
        else if (hr != STG_E_FILEALREADYEXISTS)
637
        {
638
            ERR("failed to create stream %s (hr = %08x)\n", debugstr_w(encname), hr);
639 640 641 642 643
            msi_free( encname );
            return ERROR_FUNCTION_FAILED;
        }
        msi_free( encname );
    }
644 645 646

    return ERROR_SUCCESS;
}