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