atl80: Added AtlComModuleRegisterServer implementation (based on AtlModuleRegisterSer...
[wine] / dlls / msi / font.c
1 /*
2  * Implementation of the Microsoft Installer (msi.dll)
3  *
4  * Copyright 2004,2005 Aric Stewart 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 received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #include <stdarg.h>
22 #include "windef.h"
23 #include "winbase.h"
24 #include "winerror.h"
25 #include "winreg.h"
26 #include "wine/debug.h"
27 #include "msipriv.h"
28 #include "wine/unicode.h"
29
30 WINE_DEFAULT_DEBUG_CHANNEL(msi);
31
32 typedef struct _tagTT_OFFSET_TABLE {
33     USHORT uMajorVersion;
34     USHORT uMinorVersion;
35     USHORT uNumOfTables;
36     USHORT uSearchRange;
37     USHORT uEntrySelector;
38     USHORT uRangeShift;
39 } TT_OFFSET_TABLE;
40
41 typedef struct _tagTT_TABLE_DIRECTORY {
42     char szTag[4]; /* table name */
43     ULONG uCheckSum; /* Check sum */
44     ULONG uOffset; /* Offset from beginning of file */
45     ULONG uLength; /* length of the table in bytes */
46 } TT_TABLE_DIRECTORY;
47
48 typedef struct _tagTT_NAME_TABLE_HEADER {
49     USHORT uFSelector; /* format selector. Always 0 */
50     USHORT uNRCount; /* Name Records count */
51     USHORT uStorageOffset; /* Offset for strings storage,
52                             * from start of the table */
53 } TT_NAME_TABLE_HEADER;
54
55 #define NAME_ID_FULL_FONT_NAME  4
56 #define NAME_ID_VERSION         5
57
58 typedef struct _tagTT_NAME_RECORD {
59     USHORT uPlatformID;
60     USHORT uEncodingID;
61     USHORT uLanguageID;
62     USHORT uNameID;
63     USHORT uStringLength;
64     USHORT uStringOffset; /* from start of storage area */
65 } TT_NAME_RECORD;
66
67 #define SWAPWORD(x) MAKEWORD(HIBYTE(x), LOBYTE(x))
68 #define SWAPLONG(x) MAKELONG(SWAPWORD(HIWORD(x)), SWAPWORD(LOWORD(x)))
69
70 static const WCHAR regfont1[] =
71     {'S','o','f','t','w','a','r','e','\\',
72      'M','i','c','r','o','s','o','f','t','\\',
73      'W','i','n','d','o','w','s',' ','N','T','\\',
74      'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\',
75      'F','o','n','t','s',0};
76 static const WCHAR regfont2[] =
77     {'S','o','f','t','w','a','r','e','\\',
78      'M','i','c','r','o','s','o','f','t','\\',
79      'W','i','n','d','o','w','s','\\',
80      'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\',
81      'F','o','n','t','s',0};
82
83 /*
84  * Code based off of code located here
85  * http://www.codeproject.com/gdi/fontnamefromfile.asp
86  */
87 static WCHAR *load_ttf_name_id( const WCHAR *filename, DWORD id )
88 {
89     TT_TABLE_DIRECTORY tblDir;
90     BOOL bFound = FALSE;
91     TT_OFFSET_TABLE ttOffsetTable;
92     TT_NAME_TABLE_HEADER ttNTHeader;
93     TT_NAME_RECORD ttRecord;
94     DWORD dwRead;
95     HANDLE handle;
96     LPWSTR ret = NULL;
97     int i;
98
99     handle = CreateFileW(filename ,GENERIC_READ, 0, NULL, OPEN_EXISTING,
100                     FILE_ATTRIBUTE_NORMAL, 0 );
101     if (handle == INVALID_HANDLE_VALUE)
102     {
103         ERR("Unable to open font file %s\n", debugstr_w(filename));
104         return NULL;
105     }
106
107     if (!ReadFile(handle,&ttOffsetTable, sizeof(TT_OFFSET_TABLE),&dwRead,NULL))
108         goto end;
109
110     ttOffsetTable.uNumOfTables = SWAPWORD(ttOffsetTable.uNumOfTables);
111     ttOffsetTable.uMajorVersion = SWAPWORD(ttOffsetTable.uMajorVersion);
112     ttOffsetTable.uMinorVersion = SWAPWORD(ttOffsetTable.uMinorVersion);
113
114     if ((ttOffsetTable.uMajorVersion != 1 || ttOffsetTable.uMinorVersion != 0) &&
115         (ttOffsetTable.uMajorVersion != 0x4f54 || ttOffsetTable.uMinorVersion != 0x544f))
116         goto end;
117
118     for (i=0; i< ttOffsetTable.uNumOfTables; i++)
119     {
120         if (!ReadFile(handle,&tblDir, sizeof(TT_TABLE_DIRECTORY),&dwRead,NULL))
121             break;
122         if (memcmp(tblDir.szTag,"name",4)==0)
123         {
124             bFound = TRUE;
125             tblDir.uLength = SWAPLONG(tblDir.uLength);
126             tblDir.uOffset = SWAPLONG(tblDir.uOffset);
127             break;
128         }
129     }
130
131     if (!bFound)
132         goto end;
133
134     SetFilePointer(handle, tblDir.uOffset, NULL, FILE_BEGIN);
135     if (!ReadFile(handle,&ttNTHeader, sizeof(TT_NAME_TABLE_HEADER), &dwRead,NULL))
136         goto end;
137
138     ttNTHeader.uNRCount = SWAPWORD(ttNTHeader.uNRCount);
139     ttNTHeader.uStorageOffset = SWAPWORD(ttNTHeader.uStorageOffset);
140     for(i=0; i<ttNTHeader.uNRCount; i++)
141     {
142         if (!ReadFile(handle,&ttRecord, sizeof(TT_NAME_RECORD),&dwRead,NULL))
143             break;
144
145         ttRecord.uNameID = SWAPWORD(ttRecord.uNameID);
146         ttRecord.uPlatformID = SWAPWORD(ttRecord.uPlatformID);
147         ttRecord.uEncodingID = SWAPWORD(ttRecord.uEncodingID);
148         if (ttRecord.uNameID == id && ttRecord.uPlatformID == 3 &&
149             (ttRecord.uEncodingID == 0 || ttRecord.uEncodingID == 1))
150         {
151             WCHAR *buf;
152             unsigned int i;
153
154             ttRecord.uStringLength = SWAPWORD(ttRecord.uStringLength);
155             ttRecord.uStringOffset = SWAPWORD(ttRecord.uStringOffset);
156             SetFilePointer(handle, tblDir.uOffset + ttRecord.uStringOffset + ttNTHeader.uStorageOffset,
157                            NULL, FILE_BEGIN);
158             if (!(buf = msi_alloc_zero( ttRecord.uStringLength + sizeof(WCHAR) ))) goto end;
159             dwRead = 0;
160             ReadFile(handle, buf, ttRecord.uStringLength, &dwRead, NULL);
161             if (dwRead % sizeof(WCHAR))
162             {
163                 msi_free(buf);
164                 goto end;
165             }
166             for (i = 0; i < dwRead / sizeof(WCHAR); i++) buf[i] = SWAPWORD(buf[i]);
167             ret = strdupW(buf);
168             msi_free(buf);
169             break;
170         }
171     }
172
173 end:
174     CloseHandle(handle);
175     TRACE("Returning %s\n", debugstr_w(ret));
176     return ret;
177 }
178
179 static WCHAR *font_name_from_file( const WCHAR *filename )
180 {
181     static const WCHAR truetypeW[] = {' ','(','T','r','u','e','T','y','p','e',')',0};
182     WCHAR *name, *ret = NULL;
183
184     if ((name = load_ttf_name_id( filename, NAME_ID_FULL_FONT_NAME )))
185     {
186         if (!name[0])
187         {
188             WARN("empty font name\n");
189             msi_free( name );
190             return NULL;
191         }
192         ret = msi_alloc( (strlenW( name ) + strlenW( truetypeW ) + 1 ) * sizeof(WCHAR) );
193         strcpyW( ret, name );
194         strcatW( ret, truetypeW );
195         msi_free( name );
196     }
197     return ret;
198 }
199
200 WCHAR *msi_font_version_from_file( const WCHAR *filename )
201 {
202     static const WCHAR fmtW[] = {'%','u','.','%','u','.','0','.','0',0};
203     WCHAR *version, *p, *q, *ret = NULL;
204
205     if ((version = load_ttf_name_id( filename, NAME_ID_VERSION )))
206     {
207         int len, major = 0, minor = 0;
208         if ((p = strchrW( version, ';' ))) *p = 0;
209         p = version;
210         while (*p && !isdigitW( *p )) p++;
211         if ((q = strchrW( p, '.' )))
212         {
213             major = atoiW( p );
214             p = ++q;
215             while (*q && isdigitW( *q )) q++;
216             if (!*q || *q == ' ') minor = atoiW( p );
217             else major = 0;
218         }
219         len = strlenW( fmtW ) + 20;
220         ret = msi_alloc( len * sizeof(WCHAR) );
221         sprintfW( ret, fmtW, major, minor );
222         msi_free( version );
223     }
224     return ret;
225 }
226
227 static UINT ITERATE_RegisterFonts(MSIRECORD *row, LPVOID param)
228 {
229     MSIPACKAGE *package = param;
230     LPWSTR name;
231     LPCWSTR filename;
232     MSIFILE *file;
233     MSICOMPONENT *comp;
234     HKEY hkey1, hkey2;
235     MSIRECORD *uirow;
236     LPWSTR uipath, p;
237
238     filename = MSI_RecordGetString( row, 1 );
239     file = msi_get_loaded_file( package, filename );
240     if (!file)
241     {
242         WARN("unable to find file %s\n", debugstr_w(filename));
243         return ERROR_SUCCESS;
244     }
245     comp = msi_get_loaded_component( package, file->Component->Component );
246     if (!comp)
247     {
248         WARN("unable to find component %s\n", debugstr_w(file->Component->Component));
249         return ERROR_SUCCESS;
250     }
251     comp->Action = msi_get_component_action( package, comp );
252     if (comp->Action != INSTALLSTATE_LOCAL)
253     {
254         TRACE("component not scheduled for installation %s\n", debugstr_w(comp->Component));
255         return ERROR_SUCCESS;
256     }
257
258     RegCreateKeyW(HKEY_LOCAL_MACHINE,regfont1,&hkey1);
259     RegCreateKeyW(HKEY_LOCAL_MACHINE,regfont2,&hkey2);
260
261     if (MSI_RecordIsNull(row,2))
262         name = font_name_from_file( file->TargetPath );
263     else
264         name = msi_dup_record_field(row,2);
265
266     if (name)
267     {
268         msi_reg_set_val_str( hkey1, name, file->TargetPath);
269         msi_reg_set_val_str( hkey2, name, file->TargetPath);
270     }
271
272     msi_free(name);
273     RegCloseKey(hkey1);
274     RegCloseKey(hkey2);
275
276     /* the UI chunk */
277     uirow = MSI_CreateRecord( 1 );
278     uipath = strdupW( file->TargetPath );
279     p = strrchrW(uipath,'\\');
280     if (p) p++;
281     else p = uipath;
282     MSI_RecordSetStringW( uirow, 1, p );
283     msi_ui_actiondata( package, szRegisterFonts, uirow );
284     msiobj_release( &uirow->hdr );
285     msi_free( uipath );
286     /* FIXME: call msi_ui_progress? */
287
288     return ERROR_SUCCESS;
289 }
290
291 UINT ACTION_RegisterFonts(MSIPACKAGE *package)
292 {
293     static const WCHAR query[] = {
294         'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ','`','F','o','n','t','`',0};
295     MSIQUERY *view;
296     UINT rc;
297
298     rc = MSI_DatabaseOpenViewW(package->db, query, &view);
299     if (rc != ERROR_SUCCESS)
300         return ERROR_SUCCESS;
301
302     rc = MSI_IterateRecords(view, NULL, ITERATE_RegisterFonts, package);
303     msiobj_release(&view->hdr);
304     return rc;
305 }
306
307 static UINT ITERATE_UnregisterFonts( MSIRECORD *row, LPVOID param )
308 {
309     MSIPACKAGE *package = param;
310     LPWSTR name;
311     LPCWSTR filename;
312     MSIFILE *file;
313     MSICOMPONENT *comp;
314     HKEY hkey1, hkey2;
315     MSIRECORD *uirow;
316     LPWSTR uipath, p;
317
318     filename = MSI_RecordGetString( row, 1 );
319     file = msi_get_loaded_file( package, filename );
320     if (!file)
321     {
322         WARN("unable to find file %s\n", debugstr_w(filename));
323         return ERROR_SUCCESS;
324     }
325     comp = msi_get_loaded_component( package, file->Component->Component );
326     if (!comp)
327     {
328         WARN("unable to find component %s\n", debugstr_w(file->Component->Component));
329         return ERROR_SUCCESS;
330     }
331     comp->Action = msi_get_component_action( package, comp );
332     if (comp->Action != INSTALLSTATE_ABSENT)
333     {
334         TRACE("component not scheduled for removal %s\n", debugstr_w(comp->Component));
335         return ERROR_SUCCESS;
336     }
337
338     RegCreateKeyW( HKEY_LOCAL_MACHINE, regfont1, &hkey1 );
339     RegCreateKeyW( HKEY_LOCAL_MACHINE, regfont2, &hkey2 );
340
341     if (MSI_RecordIsNull( row, 2 ))
342         name = font_name_from_file( file->TargetPath );
343     else
344         name = msi_dup_record_field( row, 2 );
345
346     if (name)
347     {
348         RegDeleteValueW( hkey1, name );
349         RegDeleteValueW( hkey2, name );
350     }
351
352     msi_free( name );
353     RegCloseKey( hkey1 );
354     RegCloseKey( hkey2 );
355
356     /* the UI chunk */
357     uirow = MSI_CreateRecord( 1 );
358     uipath = strdupW( file->TargetPath );
359     p = strrchrW( uipath,'\\' );
360     if (p) p++;
361     else p = uipath;
362     MSI_RecordSetStringW( uirow, 1, p );
363     msi_ui_actiondata( package, szUnregisterFonts, uirow );
364     msiobj_release( &uirow->hdr );
365     msi_free( uipath );
366     /* FIXME: call msi_ui_progress? */
367
368     return ERROR_SUCCESS;
369 }
370
371 UINT ACTION_UnregisterFonts( MSIPACKAGE *package )
372 {
373     static const WCHAR query[] = {
374         'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ','`','F','o','n','t','`',0};
375     MSIQUERY *view;
376     UINT r;
377
378     r = MSI_DatabaseOpenViewW( package->db, query, &view );
379     if (r != ERROR_SUCCESS)
380         return ERROR_SUCCESS;
381
382     r = MSI_IterateRecords( view, NULL, ITERATE_UnregisterFonts, package );
383     msiobj_release( &view->hdr );
384     return r;
385 }