1 /* Copyright (C) 2012 Red Hat
2
3 This file is part of IcedTea.
4
5 IcedTea is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation; either version 2, or (at your option)
8 any later version.
9
10 IcedTea is distributed in the hope that it will be useful, but
11 WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 General Public License for more details.
14
15 You should have received a copy of the GNU General Public License
16 along with IcedTea; see the file COPYING. If not, write to the
17 Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
18 02110-1301 USA.
19
20 Linking this library statically or dynamically with other modules is
21 making a combined work based on this library. Thus, the terms and
22 conditions of the GNU General Public License cover the whole
23 combination.
24
25 As a special exception, the copyright holders of this library give you
26 permission to link this library with independent modules to produce an
27 executable, regardless of the license terms of these independent
28 modules, and to copy and distribute the resulting executable under
29 terms of your choice, provided that you also meet, for each linked
30 independent module, the terms and conditions of the license of that
31 module. An independent module is a module which is not derived from
32 or based on this library. If you modify this library, you may extend
33 this exception to your version of the library, but you are not
34 obligated to do so. If you do not wish to do so, delete this
35 exception statement from your version. */
36
37 // Browser mock functions. Add more as needed.
38
39 #include <new>
40 #include <map>
41 #include <cstring>
42
43 #include <npapi.h>
44
45 #include "checked_allocations.h"
46
47 #include "UnitTest++.h"
48
49 #include "browser_mock.h"
50 #include "browser_mock_npidentifier.h"
51
52 // 'Browser' global data
53
54 // Stores NPAPI allocations
55 static AllocationSet __allocations;
56
57 // Mocked functions
58
59 // It is expected that these will only run during a unit test
mock_memalloc(uint32_t size)60 static void* mock_memalloc(uint32_t size) {
61 void* mem = malloc(size);
62 __allocations.insert(mem);
63 return mem;
64 }
65
mock_memfree(void * ptr)66 static void mock_memfree(void* ptr) {
67 if (__allocations.erase(ptr)) {
68 free(ptr);
69 } else {
70 printf("Attempt to free memory with browserfunctions.memfree that was not allocated by the browser!\n");
71 CHECK(false);
72 }
73 }
74
mock_retainobject(NPObject * obj)75 static NPObject* mock_retainobject(NPObject* obj) {
76 obj->referenceCount++;
77 return obj;
78 }
79
mock_releaseobject(NPObject * obj)80 static void mock_releaseobject(NPObject* obj) {
81 if (--(obj->referenceCount) == 0) {
82 if (obj->_class->deallocate) {
83 obj->_class->deallocate(obj);
84 } else {
85 mock_memfree(obj);
86 }
87 }
88 }
89
mock_releasevariantvalue(NPVariant * variant)90 static void mock_releasevariantvalue(NPVariant* variant) {
91 if (variant->type == NPVariantType_String) {
92 /* Only string and object values require freeing */
93 mock_memfree((void*)variant->value.stringValue.UTF8Characters);
94 } else if (variant->type == NPVariantType_Object) {
95 mock_releaseobject(variant->value.objectValue);
96 }
97 }
98
mock_createobject(NPP instance,NPClass * np_class)99 static NPObject* mock_createobject(NPP instance, NPClass* np_class) {
100 NPObject* obj;
101 if (np_class->allocate) {
102 obj = np_class->allocate(instance, np_class);
103 } else {
104 obj = (NPObject*) mock_memalloc(sizeof(NPObject));
105 }
106 obj->referenceCount = 1;
107 obj->_class = np_class;
108 return obj;
109 }
110
mock_getproperty(NPP npp,NPObject * obj,NPIdentifier property_name,NPVariant * result)111 static bool mock_getproperty(NPP npp, NPObject* obj, NPIdentifier property_name, NPVariant* result) {
112 if (obj->_class->getProperty) {
113 return obj->_class->getProperty(obj, property_name, result);
114 } else {
115 return false;
116 }
117 }
118
mock_setproperty(NPP npp,NPObject * obj,NPIdentifier property_name,const NPVariant * value)119 static bool mock_setproperty(NPP npp, NPObject* obj, NPIdentifier property_name, const NPVariant* value) {
120 if (obj->_class->setProperty) {
121 return obj->_class->setProperty(obj, property_name, value);
122 } else {
123 return false;
124 }
125 }
126
browsermock_create_table()127 NPNetscapeFuncs browsermock_create_table() {
128 NPNetscapeFuncs browser_functions;
129 memset(&browser_functions, 0, sizeof(NPNetscapeFuncs));
130 browser_functions.size = sizeof(NPNetscapeFuncs);
131
132 browser_functions.memalloc = &mock_memalloc;
133 browser_functions.memfree = &mock_memfree;
134
135 browser_functions.createobject = &mock_createobject;
136 browser_functions.retainobject = &mock_retainobject;
137 browser_functions.releaseobject = &mock_releaseobject;
138 browser_functions.releasevariantvalue = &mock_releasevariantvalue;
139
140 browser_functions.setproperty = &mock_setproperty;
141 browser_functions.getproperty = &mock_getproperty;
142
143 browser_functions.utf8fromidentifier = &browsermock_utf8fromidentifier;
144 browser_functions.getstringidentifier = &browsermock_getstringidentifier;
145 browser_functions.identifierisstring = &browsermock_identifierisstring;
146
147 return browser_functions;
148 }
149
browsermock_clear_state()150 void browsermock_clear_state() {
151 __allocations.clear();
152 }
153
browsermock_unfreed_allocations()154 int browsermock_unfreed_allocations() {
155 return __allocations.size();
156 }
157