]>
git.saurik.com Git - wxWidgets.git/blob - src/html/chm.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: src/html/chm.cpp
3 // Purpose: CHM (Help) support for wxHTML
4 // Author: Markus Sinner
5 // Copyright: (c) 2003 Herd Software Development
7 // Licence: wxWindows licence
8 /////////////////////////////////////////////////////////////////////////////
10 #include "wx/wxprec.h"
23 #include "wx/module.h"
26 #include "wx/filesys.h"
27 #include "wx/mstream.h"
28 #include "wx/wfstream.h"
30 #include "wx/html/forcelnk.h"
31 FORCE_LINK_ME(wxhtml_chm_support
)
33 // ----------------------------------------------------------------------------
36 /// this class is used to abstract access to CHM-Archives
37 /// with library mspack written by Stuart Caie
38 /// http://www.kyz.uklinux.net/libmspack/
39 // ----------------------------------------------------------------------------
44 wxChmTools(const wxFileName
&archive
);
48 /// Generate error-string for error-code
49 static const wxString
ChmErrorMsg(int error
);
51 /// get an array of archive-member-filenames
52 const wxArrayString
*GetFileNames()
57 /// get the name of the archive representated by this class
58 const wxString
GetArchiveName()
63 /// Find a file in the archive
64 const wxString
Find(const wxString
& pattern
,
65 const wxString
& startfrom
= wxEmptyString
);
67 /// Extract a file in the archive into a file
68 size_t Extract(const wxString
& pattern
, const wxString
& filename
);
70 /// check archive for a file
71 bool Contains(const wxString
& pattern
);
73 /// get a string for the last error which occurred
74 const wxString
GetLastErrorMessage();
80 // these vars are used by FindFirst/Next:
81 wxString m_chmFileName
;
82 char *m_chmFileNameANSI
;
84 /// mspack-pointer to mschmd_header
85 struct mschmd_header
*m_archive
;
86 /// mspack-pointer to mschm_decompressor
87 struct mschm_decompressor
*m_decompressor
;
89 /// Array of filenames in archive
90 wxArrayString
* m_fileNames
;
92 /// Internal function to get filepointer
93 struct mschmd_file
*GetMschmdFile(const wxString
& pattern
);
100 * @param archive The filename of the archive to open
102 wxChmTools::wxChmTools(const wxFileName
&archive
)
104 m_chmFileName
= archive
.GetFullPath();
106 wxASSERT_MSG( !m_chmFileName
.empty(), wxT("empty archive name") );
109 m_decompressor
= NULL
;
113 struct mschmd_header
*chmh
;
114 struct mschm_decompressor
*chmd
;
115 struct mschmd_file
*file
;
117 // Create decompressor
118 chmd
= mspack_create_chm_decompressor(NULL
);
119 m_decompressor
= (struct mschm_decompressor
*) chmd
;
121 // NB: we must make a copy of the string because chmd->open won't call
122 // strdup() [libmspack-20030726], which would cause crashes in
123 // Unicode build when mb_str() returns temporary buffer
124 m_chmFileNameANSI
= strdup((const char*)m_chmFileName
.mb_str(wxConvFile
));
126 // Open the archive and store it in class:
127 if ( (chmh
= chmd
->open(chmd
, (char*)m_chmFileNameANSI
)) )
131 // Create Filenamearray
132 m_fileNames
= new wxArrayString
;
134 // Store Filenames in array
135 for (file
= chmh
->files
; file
; file
= file
->next
)
137 m_fileNames
->Add(wxString::FromAscii(file
->filename
));
142 wxLogError(_("Failed to open CHM archive '%s'."),
143 archive
.GetFullPath().c_str());
144 m_lasterror
= (chmd
->last_error(chmd
));
153 wxChmTools::~wxChmTools()
155 struct mschm_decompressor
*chmd
= m_decompressor
;
156 struct mschmd_header
*chmh
= m_archive
;
162 chmd
->close(chmd
, chmh
);
164 free(m_chmFileNameANSI
);
166 // Destroy Decompressor
168 mspack_destroy_chm_decompressor(chmd
);
174 * Checks if the given pattern matches to any
175 * filename stored in archive
177 * @param pattern The filename pattern, may include '*' and/or '?'
178 * @return true, if any file matching pattern has been found,
181 bool wxChmTools::Contains(const wxString
& pattern
)
184 wxString pattern_tmp
= wxString(pattern
).MakeLower();
186 // loop through filearay
187 if ( m_fileNames
&& (count
= m_fileNames
->GetCount()) > 0 )
189 for (int i
= 0; i
< count
; i
++)
191 wxString tmp
= m_fileNames
->Item(i
).MakeLower();
192 if ( tmp
.Matches(pattern_tmp
) || tmp
.Mid(1).Matches(pattern_tmp
))
205 * Finds the next file descibed by a pattern in the archive, starting
206 * the file given by second parameter
208 * @param pattern The file-pattern to search for. May contain '*' and/or '?'
209 * @param startfrom The filename which the search should start after
210 * @returns The full pathname of the found file
212 const wxString
wxChmTools::Find(const wxString
& pattern
,
213 const wxString
& startfrom
)
217 wxString
pattern_tmp(pattern
);
218 wxString
startfrom_tmp(startfrom
);
219 pattern_tmp
.MakeLower();
220 startfrom_tmp
.MakeLower();
222 if ( m_fileNames
&& (count
= m_fileNames
->GetCount()) > 0 )
224 for (int i
= 0; i
< count
; i
++)
226 tmp
= m_fileNames
->Item(i
).MakeLower();
227 // if we find the string where the search should began
228 if ( tmp
.Matches(startfrom_tmp
) ||
229 tmp
.Mid(1).Matches(startfrom_tmp
) )
231 if ( tmp
.Matches(pattern_tmp
) ||
232 tmp
.Mid(1).Matches(pattern_tmp
) )
239 return wxEmptyString
;
246 * extracts the first hit of pattern to the given position
248 * @param pattern A filename pattern (may contain * and ? chars)
249 * @param filename The FileName where to temporary extract the file to
250 * @return 0 at no file extracted<br>
251 * number of bytes extracted else
253 size_t wxChmTools::Extract(const wxString
& pattern
, const wxString
& filename
)
255 struct mschm_decompressor
*d
= m_decompressor
;
256 struct mschmd_header
*h
= m_archive
;
257 struct mschmd_file
*f
;
260 wxString pattern_tmp
= (wxString(pattern
)).MakeLower();
262 for (f
= h
->files
; f
; f
= f
->next
)
264 tmp
= wxString::FromAscii(f
->filename
).MakeLower();
265 if ( tmp
.Matches(pattern_tmp
) ||
266 tmp
.Mid(1).Matches(pattern_tmp
) )
268 // ignore leading '/'
270 (char*)(const char*)filename
.mb_str(wxConvFile
)))
273 m_lasterror
= d
->last_error(d
);
274 wxLogError(_("Could not extract %s into %s: %s"),
275 wxString::FromAscii(f
->filename
).c_str(),
277 ChmErrorMsg(m_lasterror
).c_str());
282 return (size_t) f
->length
;
293 * Find a file by pattern
295 * @param pattern A filename pattern (may contain * and ? chars)
296 * @return A pointer to the file (mschmd_file*)
298 struct mschmd_file
*wxChmTools::GetMschmdFile(const wxString
& pattern_orig
)
300 struct mschmd_file
*f
;
301 struct mschmd_header
*h
= (struct mschmd_header
*) m_archive
;
303 wxString pattern
= wxString(pattern_orig
).MakeLower();
305 for (f
= h
->files
; f
; f
= f
->next
)
307 tmp
= wxString::FromAscii(f
->filename
).MakeLower();
308 if ( tmp
.Matches(pattern
) || tmp
.Mid(1).Matches(pattern
) )
310 // ignore leading '/'
318 const wxString
wxChmTools::GetLastErrorMessage()
320 return ChmErrorMsg(m_lasterror
);
323 const wxString
wxChmTools::ChmErrorMsg(int error
)
328 return _("no error");
329 case MSPACK_ERR_ARGS
:
330 return _("bad arguments to library function");
331 case MSPACK_ERR_OPEN
:
332 return _("error opening file");
333 case MSPACK_ERR_READ
:
334 return _("read error");
335 case MSPACK_ERR_WRITE
:
336 return _("write error");
337 case MSPACK_ERR_SEEK
:
338 return _("seek error");
339 case MSPACK_ERR_NOMEMORY
:
340 return _("out of memory");
341 case MSPACK_ERR_SIGNATURE
:
342 return _("bad signature");
343 case MSPACK_ERR_DATAFORMAT
:
344 return _("error in data format");
345 case MSPACK_ERR_CHECKSUM
:
346 return _("checksum error");
347 case MSPACK_ERR_CRUNCH
:
348 return _("compression error");
349 case MSPACK_ERR_DECRUNCH
:
350 return _("decompression error");
352 return _("unknown error");
356 // ---------------------------------------------------------------------------
358 // ---------------------------------------------------------------------------
360 class wxChmInputStream
: public wxInputStream
364 wxChmInputStream(const wxString
& archive
,
365 const wxString
& file
, bool simulate
= false);
367 virtual ~wxChmInputStream();
369 /// Return the size of the accessed file in archive
370 virtual size_t GetSize() const { return m_size
; }
372 virtual bool Eof() const;
373 /// Set simulation-mode of HHP-File (if non is found)
374 void SimulateHHP(bool sim
) { m_simulateHHP
= sim
; }
377 /// See wxInputStream
378 virtual size_t OnSysRead(void *buffer
, size_t bufsize
);
379 /// See wxInputStream
380 virtual wxFileOffset
OnSysSeek(wxFileOffset seek
, wxSeekMode mode
);
381 /// See wxInputStream
382 virtual wxFileOffset
OnSysTell() const { return m_pos
; }
390 wxInputStream
* m_contentStream
;
392 void CreateHHPStream();
393 bool CreateFileStream(const wxString
& pattern
);
394 // this void* is handle of archive . I'm sorry it is void and not proper
395 // type but I don't want to make unzip.h header public.
398 // locates the file and returns a mspack_file *
399 mspack_file
*LocateFile(wxString filename
);
401 // should store pointer to current file
404 // The Chm-Class for extracting the data
413 * @param archive The name of the .chm archive. Remember that archive must
414 * be local file accesible via fopen, fread functions!
415 * @param filename The Name of the file to be extracted from archive
416 * @param simulate if true than class should simulate .HHP-File based on #SYSTEM
417 * if false than class does nothing if it doesn't find .hhp
419 wxChmInputStream::wxChmInputStream(const wxString
& archive
,
420 const wxString
& filename
, bool simulate
)
426 m_contentStream
= NULL
;
427 m_lasterror
= wxSTREAM_NO_ERROR
;
428 m_chm
= new wxChmTools (wxFileName(archive
));
430 m_fileName
= wxString(filename
).MakeLower();
431 m_simulateHHP
= simulate
;
433 if ( !m_chm
->Contains(m_fileName
) )
435 // if the file could not be located, but was *.hhp, than we create
436 // the content of the hhp-file on the fly and store it for reading
437 // by the application
438 if ( m_fileName
.Find(wxT(".hhp")) != wxNOT_FOUND
&& m_simulateHHP
)
440 // now we open an hhp-file
445 wxLogError(_("Could not locate file '%s'."), filename
.c_str());
446 m_lasterror
= wxSTREAM_READ_ERROR
;
452 CreateFileStream(m_fileName
);
457 wxChmInputStream::~wxChmInputStream()
461 delete m_contentStream
;
470 bool wxChmInputStream::Eof() const
472 return (m_content
==NULL
||
473 m_contentStream
==NULL
||
474 m_contentStream
->Eof() ||
480 size_t wxChmInputStream::OnSysRead(void *buffer
, size_t bufsize
)
482 if ( m_pos
>= m_size
)
484 m_lasterror
= wxSTREAM_EOF
;
487 m_lasterror
= wxSTREAM_NO_ERROR
;
489 // If the rest to read from the stream is less
490 // than the buffer size, then only read the rest
491 if ( m_pos
+ bufsize
> m_size
)
492 bufsize
= m_size
- m_pos
;
494 if (m_contentStream
->SeekI(m_pos
) == wxInvalidOffset
)
496 m_lasterror
= wxSTREAM_EOF
;
500 size_t read
= m_contentStream
->Read(buffer
, bufsize
).LastRead();
503 if (m_contentStream
->SeekI(m_pos
) == wxInvalidOffset
)
505 m_lasterror
= wxSTREAM_READ_ERROR
;
510 m_lasterror
= m_contentStream
->GetLastError();
518 wxFileOffset
wxChmInputStream::OnSysSeek(wxFileOffset seek
, wxSeekMode mode
)
520 wxString mode_str
= wxEmptyString
;
522 if ( !m_contentStream
|| m_contentStream
->Eof() )
524 m_lasterror
= wxSTREAM_EOF
;
527 m_lasterror
= wxSTREAM_NO_ERROR
;
529 wxFileOffset nextpos
;
534 nextpos
= seek
+ m_pos
;
540 nextpos
= m_size
- 1 + seek
;
544 break; /* just to fool compiler, never happens */
548 // Set current position on stream
549 m_contentStream
->SeekI(m_pos
);
556 * Help Browser tries to read the contents of the
557 * file by interpreting a .hhp file in the Archiv.
558 * For .chm doesn't include such a file, we need
559 * to rebuild the information based on stored
563 wxChmInputStream::CreateHHPStream()
569 wxMemoryOutputStream
*out
;
572 // Try to open the #SYSTEM-File and create the HHP File out of it
573 // see http://bonedaddy.net/pabs3/chmspec/0.1.2/Internal.html#SYSTEM
574 if ( ! m_chm
->Contains(wxT("/#SYSTEM")) )
577 wxLogDebug("Archive doesn't contain #SYSTEM file");
583 file
= wxFileName(wxT("/#SYSTEM"));
586 if ( CreateFileStream(wxT("/#SYSTEM")) )
588 // New stream for writing a memory area to simulate the
590 out
= new wxMemoryOutputStream();
592 tmp
= "[OPTIONS]\r\n";
593 out
->Write((const void *) tmp
, strlen(tmp
));
599 // use the actual stream for reading
602 /* Now read the contents, and try to get the needed information */
604 // First 4 Bytes are Version information, skip
609 // Read #SYSTEM-Code and length
611 code
= wxUINT16_SWAP_ON_BE( code
) ;
613 len
= wxUINT16_SWAP_ON_BE( len
) ;
620 case 0: // CONTENTS_FILE
623 tmp
= "Contents file=";
627 case 1: // INDEX_FILE
631 case 2: // DEFAULT_TOPIC
632 tmp
= "Default Topic=";
637 // case 6: // COMPILED_FILE
638 // tmp = "Compiled File=";
640 case 7: // COMPILED_FILE
641 tmp
= "Binary Index=YES\r\n";
642 out
->Write( (const void *) tmp
, strlen(tmp
));
645 case 4: // STRUCT SYSTEM INFO
649 char *structptr
= (char*) buf
;
650 // LCID at position 0
651 wxUint32 dummy
= *((wxUint32
*)(structptr
+0)) ;
652 wxUint32 lcid
= wxUINT32_SWAP_ON_BE( dummy
) ;
654 int len
= sprintf(msg
, "Language=0x%X\r\n", lcid
) ;
656 out
->Write(msg
, len
) ;
665 out
->Write((const void *) tmp
, strlen(tmp
));
666 out
->Write(buf
, strlen((char*)buf
));
667 out
->Write("\r\n", 2);
675 // Free the old data which wont be used any more
676 delete m_contentStream
;
680 // Now add entries which are missing
681 if ( !hhc
&& m_chm
->Contains(wxT("*.hhc")) )
683 tmp
= "Contents File=*.hhc\r\n";
684 out
->Write((const void *) tmp
, strlen(tmp
));
687 if ( !hhk
&& m_chm
->Contains(wxT("*.hhk")) )
689 tmp
= "Index File=*.hhk\r\n";
690 out
->Write((const void *) tmp
, strlen(tmp
));
693 // Now copy the Data from the memory
694 out
->SeekO(0, wxFromEnd
);
695 m_size
= out
->TellO();
696 out
->SeekO(0, wxFromStart
);
697 m_content
= (char *) malloc (m_size
+1);
698 out
->CopyTo(m_content
, m_size
);
699 m_content
[m_size
]='\0';
701 m_contentStream
= new wxMemoryInputStream(m_content
, m_size
);
709 * Creates a Stream pointing to a virtual file in
710 * the current archive
712 bool wxChmInputStream::CreateFileStream(const wxString
& pattern
)
714 wxFileInputStream
* fin
;
715 wxString tmpfile
= wxFileName::CreateTempFileName(wxT("chmstrm"));
717 if ( tmpfile
.empty() )
719 wxLogError(_("Could not create temporary file '%s'"), tmpfile
.c_str());
723 // try to extract the file
724 if ( m_chm
->Extract(pattern
, tmpfile
) <= 0 )
726 wxLogError(_("Extraction of '%s' into '%s' failed."),
727 pattern
.c_str(), tmpfile
.c_str());
728 if ( wxFileExists(tmpfile
) )
729 wxRemoveFile(tmpfile
);
734 // Open a filestream to extracted file
735 fin
= new wxFileInputStream(tmpfile
);
739 m_size
= fin
->GetSize();
740 m_content
= (char *) malloc(m_size
+1);
741 fin
->Read(m_content
, m_size
);
742 m_content
[m_size
]='\0';
744 wxRemoveFile(tmpfile
);
748 m_contentStream
= new wxMemoryInputStream (m_content
, m_size
);
750 return m_contentStream
->IsOk();
756 // ----------------------------------------------------------------------------
758 // ----------------------------------------------------------------------------
760 class wxChmFSHandler
: public wxFileSystemHandler
763 /// Constructor and Destructor
765 virtual ~wxChmFSHandler();
767 /// Is able to open location?
768 virtual bool CanOpen(const wxString
& location
);
770 virtual wxFSFile
* OpenFile(wxFileSystem
& fs
, const wxString
& location
);
771 /// Find first occurrence of spec
772 virtual wxString
FindFirst(const wxString
& spec
, int flags
= 0);
773 /// Find next occurrence of spec
774 virtual wxString
FindNext();
783 wxChmFSHandler::wxChmFSHandler() : wxFileSystemHandler()
786 m_pattern
=wxEmptyString
;
787 m_found
=wxEmptyString
;
791 wxChmFSHandler::~wxChmFSHandler()
797 bool wxChmFSHandler::CanOpen(const wxString
& location
)
799 wxString p
= GetProtocol(location
);
800 return (p
== wxT("chm")) &&
801 (GetProtocol(GetLeftLocation(location
)) == wxT("file"));
804 wxFSFile
* wxChmFSHandler::OpenFile(wxFileSystem
& WXUNUSED(fs
),
805 const wxString
& location
)
807 wxString right
= GetRightLocation(location
);
808 wxString left
= GetLeftLocation(location
);
814 if ( GetProtocol(left
) != wxT("file") )
816 wxLogError(_("CHM handler currently supports only local files!"));
820 // Work around javascript
821 wxString tmp
= wxString(right
);
822 if ( tmp
.MakeLower().Contains(wxT("javascipt")) && tmp
.Contains(wxT("\'")) )
824 right
= right
.AfterFirst(wxT('\'')).BeforeLast(wxT('\''));
827 // now work on the right location
828 if (right
.Contains(wxT("..")))
830 wxFileName
abs(right
);
831 abs
.MakeAbsolute(wxT("/"));
832 right
= abs
.GetFullPath();
835 // a workaround for absolute links to root
836 if ( (index
=right
.Index(wxT("//"))) != wxNOT_FOUND
)
838 right
=wxString(right
.Mid(index
+1));
839 wxLogWarning(_("Link contained '//', converted to absolute link."));
842 wxFileName leftFilename
= wxFileSystem::URLToFileName(left
);
843 if (!leftFilename
.FileExists())
846 // Open a stream to read the content of the chm-file
847 s
= new wxChmInputStream(leftFilename
.GetFullPath(), right
, true);
851 return new wxFSFile(s
,
852 left
+ wxT("#chm:") + right
,
855 wxDateTime(leftFilename
.GetModificationTime()));
865 * Doku see wxFileSystemHandler
867 wxString
wxChmFSHandler::FindFirst(const wxString
& spec
, int flags
)
869 wxString right
= GetRightLocation(spec
);
870 wxString left
= GetLeftLocation(spec
);
871 wxString nativename
= wxFileSystem::URLToFileName(left
).GetFullPath();
873 if ( GetProtocol(left
) != wxT("file") )
875 wxLogError(_("CHM handler currently supports only local files!"));
876 return wxEmptyString
;
879 m_chm
= new wxChmTools(wxFileName(nativename
));
880 m_pattern
= right
.AfterLast(wxT('/'));
882 wxString m_found
= m_chm
->Find(m_pattern
);
884 // now fake around hhp-files which are not existing in projects...
885 if (m_found
.empty() &&
886 m_pattern
.Contains(wxT(".hhp")) &&
887 !m_pattern
.Contains(wxT(".hhp.cached")))
889 m_found
.Printf(wxT("%s#chm:%s.hhp"),
890 left
.c_str(), m_pattern
.BeforeLast(wxT('.')).c_str());
899 wxString
wxChmFSHandler::FindNext()
901 if (m_pattern
.empty())
902 return wxEmptyString
;
904 return m_chm
->Find(m_pattern
, m_found
);
907 // ---------------------------------------------------------------------------
908 // wxModule to register CHM handler
909 // ---------------------------------------------------------------------------
911 class wxChmSupportModule
: public wxModule
913 DECLARE_DYNAMIC_CLASS(wxChmSupportModule
)
916 virtual bool OnInit()
918 wxFileSystem::AddHandler(new wxChmFSHandler
);
921 virtual void OnExit() {}
925 IMPLEMENT_DYNAMIC_CLASS(wxChmSupportModule
, wxModule
)
927 #endif // wxUSE_LIBMSPACK