1 /*
2  * Copyright (C) 2009 Apple 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
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24  */
25 
26 #include "config.h"
27 #include "CanvasRenderingContext.h"
28 
29 #include "CachedImage.h"
30 #include "CanvasPattern.h"
31 #include "HTMLCanvasElement.h"
32 #include "HTMLImageElement.h"
33 #include "HTMLVideoElement.h"
34 #include "KURL.h"
35 #include "SecurityOrigin.h"
36 
37 namespace WebCore {
38 
CanvasRenderingContext(HTMLCanvasElement * canvas)39 CanvasRenderingContext::CanvasRenderingContext(HTMLCanvasElement* canvas)
40     : m_canvas(canvas)
41 {
42 }
43 
checkOrigin(const CanvasPattern * pattern)44 void CanvasRenderingContext::checkOrigin(const CanvasPattern* pattern)
45 {
46     if (canvas()->originClean() && pattern && !pattern->originClean())
47         canvas()->setOriginTainted();
48 }
49 
checkOrigin(const HTMLCanvasElement * sourceCanvas)50 void CanvasRenderingContext::checkOrigin(const HTMLCanvasElement* sourceCanvas)
51 {
52     if (canvas()->originClean() && sourceCanvas && !sourceCanvas->originClean())
53         canvas()->setOriginTainted();
54 }
55 
checkOrigin(const HTMLImageElement * image)56 void CanvasRenderingContext::checkOrigin(const HTMLImageElement* image)
57 {
58     if (!image || !canvas()->originClean())
59         return;
60 
61     CachedImage* cachedImage = image->cachedImage();
62     checkOrigin(cachedImage->response().url());
63 
64     if (canvas()->originClean() && !cachedImage->image()->hasSingleSecurityOrigin())
65         canvas()->setOriginTainted();
66 }
67 
checkOrigin(const HTMLVideoElement * video)68 void CanvasRenderingContext::checkOrigin(const HTMLVideoElement* video)
69 {
70 #if ENABLE(VIDEO)
71     checkOrigin(KURL(KURL(), video->currentSrc()));
72     if (canvas()->originClean() && video && !video->hasSingleSecurityOrigin())
73         canvas()->setOriginTainted();
74 #endif
75 }
76 
checkOrigin(const KURL & url)77 void CanvasRenderingContext::checkOrigin(const KURL& url)
78 {
79     if (!canvas()->originClean() || m_cleanOrigins.contains(url.string()))
80         return;
81 
82     if (canvas()->securityOrigin().taintsCanvas(url))
83         canvas()->setOriginTainted();
84     else
85         m_cleanOrigins.add(url.string());
86 }
87 
88 } // namespace WebCore
89