dlls/msi/update.c

   1 /*
   2  * Implementation of the Microsoft Installer (msi.dll)
   3  *
   4  * Copyright 2004 Mike McCormack for CodeWeavers
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have receuved a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  */
  20
  21 #include <stdarg.h>
  22
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "winerror.h"
  26 #include "wine/debug.h"
  27 #include "msi.h"
  28 #include "msiquery.h"
  29 #include "objbase.h"
  30 #include "objidl.h"
  31 #include "msipriv.h"
  32 #include "winnls.h"
  33
  34 #include "query.h"
  35
  36 WINE_DEFAULT_DEBUG_CHANNEL(msi);
  37
  38
  39 /* below is the query interface to a table */
  40
  41 typedef struct tagMSIUPDATEVIEW
  42 {
  43     MSIVIEW          view;
  44     MSIDATABASE     *db;
  45     MSIVIEW         *wv;
  46     value_list      *vals;
  47 } MSIUPDATEVIEW;
  48
  49 static UINT UPDATE_fetch_int( struct tagMSIVIEW *view, UINT row, UINT col, UINT *val )
  50 {
  51     MSIUPDATEVIEW *uv = (MSIUPDATEVIEW*)view;
  52
  53     TRACE("%p %d %d %p\n", uv, row, col, val );
  54
  55     return ERROR_FUNCTION_FAILED;
  56 }
  57
  58 static UINT UPDATE_execute( struct tagMSIVIEW *view, MSIRECORD *record )
  59 {
  60     MSIUPDATEVIEW *uv = (MSIUPDATEVIEW*)view;
  61     UINT n, type, val, r, row, col_count = 0, row_count = 0;
  62     MSIVIEW *wv;
  63
  64     TRACE("%p %p\n", uv, record );
  65
  66     if( !record )
  67         return ERROR_FUNCTION_FAILED;
  68
  69     wv = uv->wv;
  70     if( !wv )
  71         return ERROR_FUNCTION_FAILED;
  72
  73     r = wv->ops->execute( wv, 0 );
  74     TRACE("tv execute returned %x\n", r);
  75     if( r )
  76         return r;
  77
  78     r = wv->ops->get_dimensions( wv, &row_count, &col_count );
  79     if( r )
  80         goto err;
  81
  82     for( row = 0; row < row_count; row++ )
  83     {
  84         for( n = 1; n <= col_count; n++ )
  85         {
  86             r = wv->ops->get_column_info( wv, n, NULL, &type );
  87             if( r )
  88                 break;
  89
  90             if( type & MSITYPE_STRING )
  91             {
  92                 const WCHAR *str = MSI_RecordGetString( record, n );
  93                 val = msi_addstringW( uv->db->strings, 0, str, -1, 1 );
  94             }
  95             else
  96             {
  97                 val = MSI_RecordGetInteger( record, n );
  98                 val |= 0x8000;
  99             }
 100             r = wv->ops->set_int( wv, row, n, val );
 101             if( r )
 102                 break;
 103         }
 104     }
 105
 106 err:
 107     return ERROR_SUCCESS;
 108 }
 109
 110
 111 static UINT UPDATE_close( struct tagMSIVIEW *view )
 112 {
 113     MSIUPDATEVIEW *uv = (MSIUPDATEVIEW*)view;
 114     MSIVIEW *wv;
 115
 116     TRACE("%p\n", uv);
 117
 118     wv = uv->wv;
 119     if( !wv )
 120         return ERROR_FUNCTION_FAILED;
 121
 122     return wv->ops->close( wv );
 123 }
 124
 125 static UINT UPDATE_get_dimensions( struct tagMSIVIEW *view, UINT *rows, UINT *cols )
 126 {
 127     MSIUPDATEVIEW *uv = (MSIUPDATEVIEW*)view;
 128     MSIVIEW *wv;
 129
 130     TRACE("%p %p %p\n", uv, rows, cols );
 131
 132     wv = uv->wv;
 133     if( !wv )
 134         return ERROR_FUNCTION_FAILED;
 135
 136     return wv->ops->get_dimensions( wv, rows, cols );
 137 }
 138
 139 static UINT UPDATE_get_column_info( struct tagMSIVIEW *view,
 140                 UINT n, LPWSTR *name, UINT *type )
 141 {
 142     MSIUPDATEVIEW *uv = (MSIUPDATEVIEW*)view;
 143     MSIVIEW *wv;
 144
 145     TRACE("%p %d %p %p\n", uv, n, name, type );
 146
 147     wv = uv->wv;
 148     if( !wv )
 149         return ERROR_FUNCTION_FAILED;
 150
 151     return wv->ops->get_column_info( wv, n, name, type );
 152 }
 153
 154 static UINT UPDATE_modify( struct tagMSIVIEW *view, MSIMODIFY eModifyMode,
 155                 MSIRECORD *rec )
 156 {
 157     MSIUPDATEVIEW *uv = (MSIUPDATEVIEW*)view;
 158
 159     TRACE("%p %d %p\n", uv, eModifyMode, rec );
 160
 161     return ERROR_FUNCTION_FAILED;
 162 }
 163
 164 static UINT UPDATE_delete( struct tagMSIVIEW *view )
 165 {
 166     MSIUPDATEVIEW *uv = (MSIUPDATEVIEW*)view;
 167     MSIVIEW *wv;
 168
 169     TRACE("%p\n", uv );
 170
 171     wv = uv->wv;
 172     if( wv )
 173         wv->ops->delete( wv );
 174     delete_value_list( uv->vals );
 175     msiobj_release( &uv->db->hdr );
 176     HeapFree( GetProcessHeap(), 0, uv );
 177
 178     return ERROR_SUCCESS;
 179 }
 180
 181
 182 static MSIVIEWOPS update_ops =
 183 {
 184     UPDATE_fetch_int,
 185     NULL,
 186     NULL,
 187     NULL,
 188     UPDATE_execute,
 189     UPDATE_close,
 190     UPDATE_get_dimensions,
 191     UPDATE_get_column_info,
 192     UPDATE_modify,
 193     UPDATE_delete
 194 };
 195
 196 UINT UPDATE_CreateView( MSIDATABASE *db, MSIVIEW **view, LPWSTR table,
 197                         column_assignment *list, struct expr *expr )
 198 {
 199     MSIUPDATEVIEW *uv = NULL;
 200     UINT r;
 201     MSIVIEW *tv = NULL, *sv = NULL, *wv = NULL;
 202
 203     TRACE("%p\n", uv );
 204
 205     r = TABLE_CreateView( db, table, &tv );
 206     if( r != ERROR_SUCCESS )
 207         return r;
 208
 209     /* add conditions first */
 210     r = WHERE_CreateView( db, &wv, tv, expr );
 211     if( r != ERROR_SUCCESS )
 212     {
 213         if( sv )
 214             sv->ops->delete( tv );
 215         return r;
 216     }
 217
 218     /* then select the columns we want */
 219     r = SELECT_CreateView( db, &sv, wv, list->col_list );
 220     if( r != ERROR_SUCCESS )
 221     {
 222         if( tv )
 223             tv->ops->delete( sv );
 224         return r;
 225     }
 226
 227     uv = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof *uv );
 228     if( !uv )
 229         return ERROR_FUNCTION_FAILED;
 230
 231     /* fill the structure */
 232     uv->view.ops = &update_ops;
 233     msiobj_addref( &db->hdr );
 234     uv->db = db;
 235     uv->vals = list->val_list;
 236     uv->wv = sv;
 237     *view = (MSIVIEW*) uv;
 238
 239     return ERROR_SUCCESS;
 240 }