wininet: Only accept proxy and proxy bypass if type is INTERNET_OPEN_TYPE_PROXY.
[wine.git] / dlls / hhctrl.ocx / search.c
blob571cdd628dcc2fe392e5be955bfadf898221c0de
1 /*
2 * Copyright 2010 Erich Hoover
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.
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.
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
19 #define NONAMELESSUNION
20 #define NONAMELESSSTRUCT
22 #include "hhctrl.h"
23 #include "stream.h"
25 #include "wine/debug.h"
27 WINE_DEFAULT_DEBUG_CHANNEL(htmlhelp);
29 static SearchItem *SearchCHM_Folder(SearchItem *item, IStorage *pStorage,
30 const WCHAR *folder, const char *needle);
32 /* Allocate a ListView entry for a search result. */
33 static SearchItem *alloc_search_item(WCHAR *title, const WCHAR *filename)
35 int filename_len = filename ? (strlenW(filename)+1)*sizeof(WCHAR) : 0;
36 SearchItem *item;
38 item = heap_alloc_zero(sizeof(SearchItem));
39 if(filename)
41 item->filename = heap_alloc(filename_len);
42 memcpy(item->filename, filename, filename_len);
44 item->title = title; /* Already allocated */
46 return item;
49 /* Fill the ListView object corresponding to the found Search tab items */
50 static void fill_search_tree(HWND hwndList, SearchItem *item)
52 int index = 0;
53 LVITEMW lvi;
55 SendMessageW(hwndList, LVM_DELETEALLITEMS, 0, 0);
56 while(item) {
57 TRACE("list debug: %s\n", debugstr_w(item->filename));
59 memset(&lvi, 0, sizeof(lvi));
60 lvi.iItem = index++;
61 lvi.mask = LVIF_TEXT|LVIF_PARAM;
62 lvi.cchTextMax = strlenW(item->title)+1;
63 lvi.pszText = item->title;
64 lvi.lParam = (LPARAM)item;
65 item->id = (HTREEITEM)SendMessageW(hwndList, LVM_INSERTITEMW, 0, (LPARAM)&lvi);
66 item = item->next;
70 /* Search the CHM storage stream (an HTML file) for the requested text.
72 * Before searching the HTML file all HTML tags are removed so that only
73 * the content of the document is scanned. If the search string is found
74 * then the title of the document is returned.
76 static WCHAR *SearchCHM_File(IStorage *pStorage, const WCHAR *file, const char *needle)
78 char *buffer = heap_alloc(BLOCK_SIZE);
79 strbuf_t content, node, node_name;
80 IStream *temp_stream = NULL;
81 DWORD i, buffer_size = 0;
82 WCHAR *title = NULL;
83 BOOL found = FALSE;
84 stream_t stream;
85 HRESULT hres;
87 hres = IStorage_OpenStream(pStorage, file, NULL, STGM_READ, 0, &temp_stream);
88 if(FAILED(hres)) {
89 FIXME("Could not open '%s' stream: %08x\n", debugstr_w(file), hres);
90 goto cleanup;
93 strbuf_init(&node);
94 strbuf_init(&content);
95 strbuf_init(&node_name);
97 stream_init(&stream, temp_stream);
99 /* Remove all HTML formatting and record the title */
100 while(next_node(&stream, &node)) {
101 get_node_name(&node, &node_name);
103 if(next_content(&stream, &content) && content.len > 1)
105 char *text = &content.buf[1];
106 int textlen = content.len-1;
108 if(!strcasecmp(node_name.buf, "title"))
110 int wlen = MultiByteToWideChar(CP_ACP, 0, text, textlen, NULL, 0);
111 title = heap_alloc((wlen+1)*sizeof(WCHAR));
112 MultiByteToWideChar(CP_ACP, 0, text, textlen, title, wlen);
113 title[wlen] = 0;
116 buffer = heap_realloc(buffer, buffer_size + textlen + 1);
117 memcpy(&buffer[buffer_size], text, textlen);
118 buffer[buffer_size + textlen] = '\0';
119 buffer_size += textlen;
122 strbuf_zero(&node);
123 strbuf_zero(&content);
126 /* Convert the buffer to lower case for comparison against the
127 * requested text (already in lower case).
129 for(i=0;i<buffer_size;i++)
130 buffer[i] = tolower(buffer[i]);
132 /* Search the decoded buffer for the requested text */
133 if(strstr(buffer, needle))
134 found = TRUE;
136 strbuf_free(&node);
137 strbuf_free(&content);
138 strbuf_free(&node_name);
140 cleanup:
141 heap_free(buffer);
142 if(temp_stream)
143 IStream_Release(temp_stream);
144 if(!found)
146 heap_free(title);
147 return NULL;
149 return title;
152 /* Search all children of a CHM storage object for the requested text and
153 * return the last found search item.
155 static SearchItem *SearchCHM_Storage(SearchItem *item, IStorage *pStorage,
156 const char *needle)
158 const WCHAR szHTMext[] = {'.','h','t','m',0};
159 IEnumSTATSTG *elem = NULL;
160 WCHAR *filename = NULL;
161 STATSTG entries;
162 HRESULT hres;
163 ULONG retr;
165 hres = IStorage_EnumElements(pStorage, 0, NULL, 0, &elem);
166 if(hres != S_OK)
168 FIXME("Could not enumerate '/' storage elements: %08x\n", hres);
169 return NULL;
171 while (IEnumSTATSTG_Next(elem, 1, &entries, &retr) == NOERROR)
173 switch(entries.type) {
174 case STGTY_STORAGE:
175 item = SearchCHM_Folder(item, pStorage, entries.pwcsName, needle);
176 break;
177 case STGTY_STREAM:
178 filename = entries.pwcsName;
179 while(strchrW(filename, '/'))
180 filename = strchrW(filename, '/')+1;
181 if(strstrW(filename, szHTMext))
183 WCHAR *title = SearchCHM_File(pStorage, filename, needle);
185 if(title)
187 item->next = alloc_search_item(title, entries.pwcsName);
188 item = item->next;
191 break;
192 default:
193 FIXME("Unhandled IStorage stream element.\n");
196 IEnumSTATSTG_Release(elem);
197 return item;
200 /* Open a CHM storage object (folder) by name and find all items with
201 * the requested text. The last found item is returned.
203 static SearchItem *SearchCHM_Folder(SearchItem *item, IStorage *pStorage,
204 const WCHAR *folder, const char *needle)
206 IStorage *temp_storage = NULL;
207 HRESULT hres;
209 hres = IStorage_OpenStorage(pStorage, folder, NULL, STGM_READ, NULL, 0, &temp_storage);
210 if(FAILED(hres))
212 FIXME("Could not open '%s' storage object: %08x\n", debugstr_w(folder), hres);
213 return NULL;
215 item = SearchCHM_Storage(item, temp_storage, needle);
217 IStorage_Release(temp_storage);
218 return item;
221 /* Search the entire CHM file for the requested text and add all of
222 * the found items to a ListView for the user to choose the item
223 * they want.
225 void InitSearch(HHInfo *info, const char *needle)
227 CHMInfo *chm = info->pCHMInfo;
228 SearchItem *root_item = alloc_search_item(NULL, NULL);
230 SearchCHM_Storage(root_item, chm->pStorage, needle);
231 fill_search_tree(info->search.hwndList, root_item->next);
232 if(info->search.root)
233 ReleaseSearch(info);
234 info->search.root = root_item;
237 /* Free all of the found Search items. */
238 void ReleaseSearch(HHInfo *info)
240 SearchItem *item = info->search.root;
242 info->search.root = NULL;
243 while(item) {
244 heap_free(item->filename);
245 item = item->next;