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, MSIHANDLE hrec)
 155 {
 156     MSIUPDATEVIEW *uv = (MSIUPDATEVIEW*)view;
 157
 158     TRACE("%p %d %ld\n", uv, eModifyMode, hrec );
 159
 160     return ERROR_FUNCTION_FAILED;
 161 }
 162
 163 static UINT UPDATE_delete( struct tagMSIVIEW *view )
 164 {
 165     MSIUPDATEVIEW *uv = (MSIUPDATEVIEW*)view;
 166     MSIVIEW *wv;
 167
 168     TRACE("%p\n", uv );
 169
 170     wv = uv->wv;
 171     if( wv )
 172         wv->ops->delete( wv );
 173     delete_value_list( uv->vals );
 174     HeapFree( GetProcessHeap(), 0, uv );
 175     msiobj_release( &uv->db->hdr );
 176
 177     return ERROR_SUCCESS;
 178 }
 179
 180
 181 static MSIVIEWOPS update_ops =
 182 {
 183     UPDATE_fetch_int,
 184     NULL,
 185     NULL,
 186     NULL,
 187     UPDATE_execute,
 188     UPDATE_close,
 189     UPDATE_get_dimensions,
 190     UPDATE_get_column_info,
 191     UPDATE_modify,
 192     UPDATE_delete
 193 };
 194
 195 UINT UPDATE_CreateView( MSIDATABASE *db, MSIVIEW **view, LPWSTR table,
 196                         column_assignment *list, struct expr *expr )
 197 {
 198     MSIUPDATEVIEW *uv = NULL;
 199     UINT r;
 200     MSIVIEW *tv = NULL, *sv = NULL, *wv = NULL;
 201
 202     TRACE("%p\n", uv );
 203
 204     r = TABLE_CreateView( db, table, &tv );
 205     if( r != ERROR_SUCCESS )
 206         return r;
 207
 208     /* add conditions first */
 209     r = WHERE_CreateView( db, &wv, tv, expr );
 210     if( r != ERROR_SUCCESS )
 211     {
 212         if( sv )
 213             sv->ops->delete( tv );
 214         return r;
 215     }
 216
 217     /* then select the columns we want */
 218     r = SELECT_CreateView( db, &sv, wv, list->col_list );
 219     if( r != ERROR_SUCCESS )
 220     {
 221         if( tv )
 222             tv->ops->delete( sv );
 223         return r;
 224     }
 225
 226     uv = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof *uv );
 227     if( !uv )
 228         return ERROR_FUNCTION_FAILED;
 229
 230     /* fill the structure */
 231     uv->view.ops = &update_ops;
 232     msiobj_addref( &db->hdr );
 233     uv->db = db;
 234     uv->vals = list->val_list;
 235     uv->wv = sv;
 236     *view = (MSIVIEW*) uv;
 237
 238     return ERROR_SUCCESS;
 239 }