2011-05-24 Jay Civelli <jcivelli@chromium.org>
[WebKit-https.git] / Source / WebCore / loader / archive / mhtml / MHTMLArchive.cpp
1 /*
2  * Copyright (C) 2011 Google Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are
6  * met:
7  *
8  *     * Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *     * Redistributions in binary form must reproduce the above
11  * copyright notice, this list of conditions and the following disclaimer
12  * in the documentation and/or other materials provided with the
13  * distribution.
14  *     * Neither the name of Google Inc. nor the names of its
15  * contributors may be used to endorse or promote products derived from
16  * this software without specific prior written permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  */
30
31 #include "config.h"
32
33 #if ENABLE(MHTML)
34 #include "MHTMLArchive.h"
35
36 #include "MHTMLParser.h"
37 #include "MIMEHeader.h"
38
39 namespace WebCore {
40
41 MHTMLArchive::MHTMLArchive()
42 {
43 }
44
45 PassRefPtr<MHTMLArchive> MHTMLArchive::create()
46 {
47     return adoptRef(new MHTMLArchive);
48 }
49
50 PassRefPtr<MHTMLArchive> MHTMLArchive::create(const KURL& url, SharedBuffer* data)
51 {
52     // For security reasons we only load MHTML pages from the local file system.
53     if (!url.isLocalFile())
54         return false;
55
56     MHTMLParser parser(data);
57     RefPtr<MHTMLArchive> mainArchive = parser.parseArchive();
58     if (!mainArchive)
59         return 0; // Invalid MHTML file.
60
61     // Since MHTML is a flat format, we need to make all frames aware of all resources.
62     for (size_t i = 0; i < parser.frameCount(); ++i) {
63         RefPtr<MHTMLArchive> archive = parser.frameAt(i);
64         for (size_t j = 0; j < parser.frameCount(); ++j) {
65             if (i != j)
66                 archive->addSubframeArchive(parser.frameAt(j));
67         }
68         for (size_t j = 0; j < parser.subResourceCount(); ++j)
69             archive->addSubresource(parser.subResourceAt(j));
70     }
71     return mainArchive.release();
72 }
73
74 }
75 #endif