rpcrt4: Add a comment in NdrSendReceive explaining why we change the BufferLength.
[wine] / dlls / hhctrl.ocx / content.c
1 /*
2  * Copyright 2007 Jacek Caban 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 #define NONAMELESSUNION
20 #define NONAMELESSSTRUCT
21
22 #include "hhctrl.h"
23
24 #include "wine/debug.h"
25
26 WINE_DEFAULT_DEBUG_CHANNEL(htmlhelp);
27
28 #define BLOCK_SIZE 0x1000
29
30 typedef enum {
31     INSERT_NEXT,
32     INSERT_CHILD
33 } insert_type_t;
34
35 static void free_content_item(ContentItem *item)
36 {
37     ContentItem *next;
38
39     while(item) {
40         next = item->next;
41
42         free_content_item(item->child);
43
44         heap_free(item->name);
45         heap_free(item->local);
46         heap_free(item->merge.chm_file);
47         heap_free(item->merge.chm_index);
48
49         item = next;
50     }
51 }
52
53 typedef struct {
54     char *buf;
55     int size;
56     int len;
57 } strbuf_t;
58
59 static void strbuf_init(strbuf_t *buf)
60 {
61     buf->size = 8;
62     buf->len = 0;
63     buf->buf = heap_alloc(buf->size);
64 }
65
66 static void strbuf_zero(strbuf_t *buf)
67 {
68     buf->len = 0;
69 }
70
71 static void strbuf_free(strbuf_t *buf)
72 {
73     heap_free(buf->buf);
74 }
75
76 static void strbuf_append(strbuf_t *buf, const char *data, int len)
77 {
78     if(buf->len+len > buf->size) {
79         buf->size = buf->len+len;
80         buf->buf = heap_realloc(buf->buf, buf->size);
81     }
82
83     memcpy(buf->buf+buf->len, data, len);
84     buf->len += len;
85 }
86
87 typedef struct {
88     IStream *str;
89     char buf[BLOCK_SIZE];
90     ULONG size;
91     ULONG p;
92 } stream_t;
93
94 static void stream_init(stream_t *stream, IStream *str)
95 {
96     memset(stream, 0, sizeof(stream_t));
97     stream->str = str;
98 }
99
100 static BOOL stream_chr(stream_t *stream, strbuf_t *buf, char c)
101 {
102     BOOL b = TRUE;
103     ULONG i;
104
105     while(b) {
106         for(i=stream->p; i<stream->size; i++) {
107             if(stream->buf[i] == c) {
108                 b = FALSE;
109                 break;
110             }
111         }
112
113         if(buf && i > stream->p)
114             strbuf_append(buf, stream->buf+stream->p, i-stream->p);
115         stream->p = i;
116
117         if(stream->p == stream->size) {
118             stream->p = 0;
119             IStream_Read(stream->str, stream->buf, sizeof(stream->buf), &stream->size);
120             if(!stream->size)
121                 break;
122         }
123     }
124
125     return stream->size != 0;
126 }
127
128 static void get_node_name(strbuf_t *node, strbuf_t *name)
129 {
130     const char *ptr = node->buf+1;
131
132     strbuf_zero(name);
133
134     while(*ptr != '>' && !isspace(*ptr))
135         ptr++;
136
137     strbuf_append(name, node->buf+1, ptr-node->buf-1);
138     strbuf_append(name, "", 1);
139 }
140
141 static BOOL next_node(stream_t *stream, strbuf_t *buf)
142 {
143     if(!stream_chr(stream, NULL, '<'))
144         return FALSE;
145
146     if(!stream_chr(stream, buf, '>'))
147         return FALSE;
148
149     strbuf_append(buf, ">", 2);
150
151     return TRUE;
152 }
153
154 static const char *get_attr(const char *node, const char *name, int *len)
155 {
156     const char *ptr, *ptr2;
157     char name_buf[32];
158     int nlen;
159
160     nlen = strlen(name);
161     memcpy(name_buf, name, nlen);
162     name_buf[nlen++] = '=';
163     name_buf[nlen++] = '\"';
164     name_buf[nlen] = 0;
165
166     ptr = strstr(node, name_buf);
167     if(!ptr) {
168         WARN("name not found\n");
169         return NULL;
170     }
171
172     ptr += nlen;
173     ptr2 = strchr(ptr, '\"');
174     if(!ptr2)
175         return NULL;
176
177     *len = ptr2-ptr;
178     return ptr;
179 }
180
181 static void parse_obj_node_param(ContentItem *item, ContentItem *hhc_root, const char *text)
182 {
183     const char *ptr;
184     LPWSTR *param, merge;
185     int len, wlen;
186
187     ptr = get_attr(text, "name", &len);
188     if(!ptr) {
189         WARN("name attr not found\n");
190         return;
191     }
192
193     if(!strncasecmp("name", ptr, len)) {
194         param = &item->name;
195     }else if(!strncasecmp("merge", ptr, len)) {
196         param = &merge;
197     }else if(!strncasecmp("local", ptr, len)) {
198         param = &item->local;
199     }else {
200         WARN("unhandled param %s\n", debugstr_an(ptr, len));
201         return;
202     }
203
204     ptr = get_attr(text, "value", &len);
205     if(!ptr) {
206         WARN("value attr not found\n");
207         return;
208     }
209
210     wlen = MultiByteToWideChar(CP_ACP, 0, ptr, len, NULL, 0);
211     *param = heap_alloc((wlen+1)*sizeof(WCHAR));
212     MultiByteToWideChar(CP_ACP, 0, ptr, len, *param, wlen);
213     (*param)[wlen] = 0;
214
215     if(param == &merge) {
216         SetChmPath(&item->merge, hhc_root->merge.chm_file, merge);
217         heap_free(merge);
218     }
219 }
220
221 static ContentItem *parse_hhc(HHInfo*,IStream*,ContentItem*,insert_type_t*);
222
223 static ContentItem *insert_item(ContentItem *item, ContentItem *new_item, insert_type_t insert_type)
224 {
225     if(!item)
226         return new_item;
227
228     if(!new_item)
229         return item;
230
231     switch(insert_type) {
232     case INSERT_NEXT:
233         item->next = new_item;
234         return new_item;
235     case INSERT_CHILD:
236         if(item->child) {
237             ContentItem *iter = item->child;
238             while(iter->next)
239                 iter = iter->next;
240             iter->next = new_item;
241         }else {
242             item->child = new_item;
243         }
244         return item;
245     }
246
247     return NULL;
248 }
249
250 static ContentItem *parse_sitemap_object(HHInfo *info, stream_t *stream, ContentItem *hhc_root,
251         insert_type_t *insert_type)
252 {
253     strbuf_t node, node_name;
254     ContentItem *item;
255
256     *insert_type = INSERT_NEXT;
257
258     strbuf_init(&node);
259     strbuf_init(&node_name);
260
261     item = heap_alloc_zero(sizeof(ContentItem));
262
263     while(next_node(stream, &node)) {
264         get_node_name(&node, &node_name);
265
266         TRACE("%s\n", node.buf);
267
268         if(!strcasecmp(node_name.buf, "/object"))
269             break;
270         if(!strcasecmp(node_name.buf, "param"))
271             parse_obj_node_param(item, hhc_root, node.buf);
272
273         strbuf_zero(&node);
274     }
275
276     strbuf_free(&node);
277     strbuf_free(&node_name);
278
279     if(item->merge.chm_index) {
280         IStream *merge_stream;
281
282         merge_stream = GetChmStream(info->pCHMInfo, item->merge.chm_file, &item->merge);
283         if(merge_stream) {
284             item->child = parse_hhc(info, merge_stream, hhc_root, insert_type);
285             IStream_Release(merge_stream);
286         }else {
287             WARN("Could not get %s::%s stream\n", debugstr_w(item->merge.chm_file),
288                  debugstr_w(item->merge.chm_file));
289
290             if(!item->name) {
291                 free_content_item(item);
292                 item = NULL;
293             }
294         }
295
296     }
297
298     return item;
299 }
300
301 static ContentItem *parse_ul(HHInfo *info, stream_t *stream, ContentItem *hhc_root)
302 {
303     strbuf_t node, node_name;
304     ContentItem *ret = NULL, *prev = NULL, *new_item = NULL;
305     insert_type_t it;
306
307     strbuf_init(&node);
308     strbuf_init(&node_name);
309
310     while(next_node(stream, &node)) {
311         get_node_name(&node, &node_name);
312
313         TRACE("%s\n", node.buf);
314
315         if(!strcasecmp(node_name.buf, "object")) {
316             const char *ptr;
317             int len;
318
319             static const char sz_text_sitemap[] = "text/sitemap";
320
321             ptr = get_attr(node.buf, "type", &len);
322
323             if(ptr && len == sizeof(sz_text_sitemap)-1
324                && !memcmp(ptr, sz_text_sitemap, len)) {
325                 new_item = parse_sitemap_object(info, stream, hhc_root, &it);
326                 prev = insert_item(prev, new_item, it);
327                 if(!ret)
328                     ret = prev;
329             }
330         }else if(!strcasecmp(node_name.buf, "ul")) {
331             new_item = parse_ul(info, stream, hhc_root);
332             insert_item(prev, new_item, INSERT_CHILD);
333         }else if(!strcasecmp(node_name.buf, "/ul")) {
334             break;
335         }
336
337         strbuf_zero(&node);
338     }
339
340     strbuf_free(&node);
341     strbuf_free(&node_name);
342
343     return ret;
344 }
345
346 static ContentItem *parse_hhc(HHInfo *info, IStream *str, ContentItem *hhc_root,
347         insert_type_t *insert_type)
348 {
349     stream_t stream;
350     strbuf_t node, node_name;
351     ContentItem *ret = NULL, *prev = NULL;
352
353     *insert_type = INSERT_NEXT;
354
355     strbuf_init(&node);
356     strbuf_init(&node_name);
357
358     stream_init(&stream, str);
359
360     while(next_node(&stream, &node)) {
361         get_node_name(&node, &node_name);
362
363         TRACE("%s\n", node.buf);
364
365         if(!strcasecmp(node_name.buf, "ul")) {
366             ContentItem *item = parse_ul(info, &stream, hhc_root);
367             prev = insert_item(prev, item, INSERT_CHILD);
368             if(!ret)
369                 ret = prev;
370             *insert_type = INSERT_CHILD;
371         }
372
373         strbuf_zero(&node);
374     }
375
376     strbuf_free(&node);
377     strbuf_free(&node_name);
378
379     return ret;
380 }
381
382 static void insert_content_item(HWND hwnd, ContentItem *parent, ContentItem *item)
383 {
384     TVINSERTSTRUCTW tvis;
385
386     memset(&tvis, 0, sizeof(tvis));
387     tvis.u.item.mask = TVIF_TEXT|TVIF_PARAM;
388     tvis.u.item.cchTextMax = strlenW(item->name)+1;
389     tvis.u.item.pszText = item->name;
390     tvis.u.item.lParam = (LPARAM)item;
391     tvis.hParent = parent ? parent->id : 0;
392     tvis.hInsertAfter = TVI_LAST;
393
394     item->id = (HTREEITEM)SendMessageW(hwnd, TVM_INSERTITEMW, 0, (LPARAM)&tvis);
395 }
396
397 static void fill_content_tree(HWND hwnd, ContentItem *parent, ContentItem *item)
398 {
399     while(item) {
400         if(item->name) {
401             insert_content_item(hwnd, parent, item);
402             fill_content_tree(hwnd, item, item->child);
403         }else {
404             fill_content_tree(hwnd, parent, item->child);
405         }
406         item = item->next;
407     }
408 }
409
410 static void set_item_parents(ContentItem *parent, ContentItem *item)
411 {
412     while(item) {
413         item->parent = parent;
414         set_item_parents(item, item->child);
415         item = item->next;
416     }
417 }
418
419 void InitContent(HHInfo *info)
420 {
421     IStream *stream;
422     insert_type_t insert_type;
423
424     info->content = heap_alloc_zero(sizeof(ContentItem));
425     SetChmPath(&info->content->merge, info->pCHMInfo->szFile, info->WinType.pszToc);
426
427     stream = GetChmStream(info->pCHMInfo, info->pCHMInfo->szFile, &info->content->merge);
428     if(!stream) {
429         TRACE("Could not get content stream\n");
430         return;
431     }
432
433     info->content->child = parse_hhc(info, stream, info->content, &insert_type);
434     IStream_Release(stream);
435
436     set_item_parents(NULL, info->content);
437     fill_content_tree(info->tabs[TAB_CONTENTS].hwnd, NULL, info->content);
438 }
439
440 void ReleaseContent(HHInfo *info)
441 {
442     free_content_item(info->content);
443 }