]>
Commit | Line | Data |
---|---|---|
7c673cae FG |
1 | /////////1/////////2/////////3/////////4/////////5/////////6/////////7/////////8 |
2 | // test_unregistered.cpp | |
3 | ||
f67539c2 | 4 | // (C) Copyright 2002 Robert Ramey - http://www.rrsd.com . |
7c673cae FG |
5 | // Use, modification and distribution is subject to the Boost Software |
6 | // License, Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at | |
7 | // http://www.boost.org/LICENSE_1_0.txt) | |
8 | ||
9 | // should pass compilation and execution | |
10 | ||
11 | #include <fstream> | |
12 | ||
13 | #include <cstddef> // NULL | |
14 | #include <cstdio> // remove | |
b32b8144 | 15 | #include <cstring> // strcmp |
7c673cae FG |
16 | #include <boost/config.hpp> |
17 | #if defined(BOOST_NO_STDC_NAMESPACE) | |
f67539c2 | 18 | namespace std{ |
7c673cae FG |
19 | using ::remove; |
20 | } | |
21 | #endif | |
22 | ||
23 | #include "test_tools.hpp" | |
24 | ||
25 | #include <boost/archive/archive_exception.hpp> | |
26 | #include <boost/serialization/base_object.hpp> | |
27 | #include <boost/serialization/type_info_implementation.hpp> | |
28 | #include <boost/core/no_exceptions_support.hpp> | |
29 | ||
30 | class polymorphic_base | |
31 | { | |
32 | friend class boost::serialization::access; | |
33 | template<class Archive> | |
34 | void serialize(Archive & /* ar */, const unsigned int /* file_version */){ | |
35 | } | |
36 | public: | |
37 | virtual ~polymorphic_base(){}; | |
38 | }; | |
39 | ||
40 | class polymorphic_derived1 : public polymorphic_base | |
41 | { | |
42 | friend class boost::serialization::access; | |
43 | template<class Archive> | |
44 | void serialize(Archive &ar, const unsigned int /* file_version */){ | |
45 | ar & BOOST_SERIALIZATION_BASE_OBJECT_NVP(polymorphic_base); | |
46 | } | |
47 | }; | |
48 | ||
49 | class polymorphic_derived2 : public polymorphic_base | |
50 | { | |
51 | friend class boost::serialization::access; | |
52 | template<class Archive> | |
53 | void serialize(Archive &ar, const unsigned int /* file_version */){ | |
54 | ar & BOOST_SERIALIZATION_BASE_OBJECT_NVP(polymorphic_base); | |
55 | } | |
56 | }; | |
57 | ||
b32b8144 FG |
58 | struct type1 { |
59 | template<typename Archive> | |
60 | void serialize(Archive&, unsigned int ver) { | |
61 | BOOST_CHECK(ver == 1); | |
62 | } | |
63 | }; | |
64 | struct type2 { | |
65 | template<typename Archive> | |
66 | void serialize(Archive&, unsigned int ver) { | |
67 | BOOST_CHECK(ver == 2); | |
68 | } | |
69 | }; | |
70 | ||
71 | BOOST_CLASS_VERSION(type1, 1); | |
72 | BOOST_CLASS_VERSION(type2, 2); | |
73 | ||
7c673cae FG |
74 | // save unregistered polymorphic classes |
75 | void save_unregistered1(const char *testfile) | |
76 | { | |
77 | test_ostream os(testfile, TEST_STREAM_FLAGS); | |
78 | test_oarchive oa(os, TEST_ARCHIVE_FLAGS); | |
79 | ||
80 | polymorphic_base *rb1 = new polymorphic_derived1; | |
b32b8144 | 81 | |
7c673cae FG |
82 | // registration IS necessary when serializing a polymorphic class |
83 | // through pointer to the base class | |
84 | bool except = false; | |
85 | BOOST_TRY { | |
86 | oa << BOOST_SERIALIZATION_NVP(rb1); | |
87 | } | |
92f5a8d4 | 88 | BOOST_CATCH(boost::archive::archive_exception const& aex){ |
7c673cae FG |
89 | except = true; |
90 | } | |
91 | BOOST_CATCH_END | |
92 | BOOST_CHECK_MESSAGE(except, "lack of registration not detected !"); | |
93 | ||
94 | delete rb1; | |
95 | } | |
96 | ||
97 | // note: the corresponding save function above will not result in | |
98 | // valid archive - hence, the following code which attempts to load | |
99 | // and archive will fail. Leave this as a reminder not to do this | |
100 | #if 0 | |
101 | // load unregistered polymorphic classes | |
102 | void load_unregistered1(const char *testfile) | |
103 | { | |
104 | std::ifstream is(testfile); | |
105 | boost::archive::iarchive ia(is); | |
106 | ||
107 | polymorphic_base *rb1(NULL); | |
108 | ||
109 | // registration IS necessary when serializing a polymorphic class | |
110 | // through pointer to the base class | |
111 | bool except = false; | |
112 | BOOST_TRY { | |
113 | ia >> BOOST_SERIALIZATION_NVP(rb1); | |
114 | } | |
92f5a8d4 | 115 | BOOST_CATCH(boost::archive::archive_exception const& aex){ |
7c673cae FG |
116 | except = true; |
117 | BOOST_CHECK_MESSAGE( | |
b32b8144 | 118 | NULL == rb1, |
7c673cae FG |
119 | "failed load resulted in a non-null pointer" |
120 | ); | |
121 | } | |
122 | BOOST_CATCH_END | |
123 | BOOST_CHECK_MESSAGE(except, "lack of registration not detected !"); | |
124 | ||
125 | delete rb1; | |
126 | } | |
127 | #endif | |
128 | ||
129 | // save unregistered polymorphic classes | |
130 | void save_unregistered2(const char *testfile) | |
131 | { | |
132 | test_ostream os(testfile, TEST_STREAM_FLAGS); | |
133 | test_oarchive oa(os, TEST_ARCHIVE_FLAGS); | |
134 | ||
135 | polymorphic_derived1 *rd1 = new polymorphic_derived1; | |
f67539c2 | 136 | |
7c673cae FG |
137 | // registration is NOT necessary when serializing a polymorphic class |
138 | // through pointer to a derived class | |
139 | bool except = false; | |
140 | BOOST_TRY { | |
141 | oa << BOOST_SERIALIZATION_NVP(rd1); | |
142 | } | |
92f5a8d4 | 143 | BOOST_CATCH(boost::archive::archive_exception const& aex){ |
7c673cae FG |
144 | except = true; |
145 | } | |
146 | BOOST_CATCH_END | |
147 | BOOST_CHECK_MESSAGE(! except, "registration not detected !"); | |
148 | ||
149 | delete rd1; | |
150 | } | |
151 | ||
152 | // note: the corresponding save function above will not result in | |
153 | // valid archive - hence, the following code which attempts to load | |
154 | // and archive will fail. Leave this as a reminder not to do this | |
155 | // load unregistered polymorphic classes | |
156 | void load_unregistered2(const char *testfile) | |
157 | { | |
158 | test_istream is(testfile, TEST_STREAM_FLAGS); | |
159 | test_iarchive ia(is, TEST_ARCHIVE_FLAGS); | |
160 | ||
161 | polymorphic_derived1 *rd1 = NULL; | |
f67539c2 | 162 | |
7c673cae FG |
163 | // registration is NOT necessary when serializing a polymorphic class |
164 | // or through pointer to a derived class | |
165 | bool except = false; | |
166 | BOOST_TRY { | |
167 | ia >> BOOST_SERIALIZATION_NVP(rd1); | |
168 | } | |
92f5a8d4 | 169 | BOOST_CATCH(boost::archive::archive_exception const& aex){ |
7c673cae FG |
170 | except = true; |
171 | BOOST_CHECK_MESSAGE( | |
f67539c2 | 172 | NULL == rd1, |
7c673cae FG |
173 | "failed load resulted in a non-null pointer" |
174 | ); | |
175 | } | |
176 | BOOST_CATCH_END | |
177 | BOOST_CHECK_MESSAGE(! except, "registration not detected !"); | |
178 | ||
179 | delete rd1; | |
180 | } | |
181 | ||
182 | // save registered polymorphic class | |
183 | void save_registered(const char *testfile) | |
184 | { | |
185 | test_ostream os(testfile, TEST_STREAM_FLAGS); | |
186 | test_oarchive oa(os, TEST_ARCHIVE_FLAGS); | |
187 | ||
188 | polymorphic_base *rb1 = new polymorphic_derived1; | |
189 | polymorphic_base *rb2 = new polymorphic_derived2; | |
190 | ||
191 | // registration (forward declaration) will permit correct serialization | |
192 | // through a pointer to a base class | |
193 | oa.register_type(static_cast<polymorphic_derived1 *>(NULL)); | |
194 | oa.register_type(static_cast<polymorphic_derived2 *>(NULL)); | |
f67539c2 | 195 | oa << BOOST_SERIALIZATION_NVP(rb1); |
7c673cae FG |
196 | oa << BOOST_SERIALIZATION_NVP(rb2); |
197 | ||
198 | delete rb1; | |
199 | delete rb2; | |
200 | } | |
201 | ||
202 | // load registered polymorphic class | |
203 | void load_registered(const char *testfile) | |
204 | { | |
205 | test_istream is(testfile, TEST_STREAM_FLAGS); | |
206 | test_iarchive ia(is, TEST_ARCHIVE_FLAGS); | |
207 | ||
208 | polymorphic_base *rb1 = NULL; | |
209 | polymorphic_base *rb2 = NULL; | |
210 | ||
211 | // registration (forward declaration) will permit correct serialization | |
212 | // through a pointer to a base class | |
213 | ia.register_type(static_cast<polymorphic_derived1 *>(NULL)); | |
214 | ia.register_type(static_cast<polymorphic_derived2 *>(NULL)); | |
215 | ||
216 | ia >> BOOST_SERIALIZATION_NVP(rb1); | |
217 | ||
218 | BOOST_CHECK_MESSAGE(NULL != rb1, "Load resulted in NULL pointer"); | |
219 | BOOST_CHECK_MESSAGE( | |
220 | boost::serialization::type_info_implementation< | |
221 | polymorphic_derived1 | |
222 | >::type::get_const_instance() | |
f67539c2 | 223 | == |
7c673cae FG |
224 | * boost::serialization::type_info_implementation< |
225 | polymorphic_base | |
226 | >::type::get_const_instance().get_derived_extended_type_info(*rb1), | |
227 | "restored pointer b1 not of correct type" | |
228 | ); | |
229 | ||
230 | ia >> BOOST_SERIALIZATION_NVP(rb2); | |
231 | BOOST_CHECK_MESSAGE(NULL != rb2, "Load resulted in NULL pointer"); | |
232 | BOOST_CHECK_MESSAGE( | |
233 | boost::serialization::type_info_implementation< | |
234 | polymorphic_derived2 | |
235 | >::type::get_const_instance() | |
f67539c2 | 236 | == |
7c673cae FG |
237 | * boost::serialization::type_info_implementation< |
238 | polymorphic_base | |
239 | >::type::get_const_instance().get_derived_extended_type_info(*rb2), | |
240 | "restored pointer b2 not of correct type" | |
241 | ); | |
242 | ||
243 | delete rb1; | |
244 | delete rb2; | |
245 | } | |
246 | ||
b32b8144 FG |
247 | // store a pointer from slot0 |
248 | void save_unregistered_pointer(const char *testfile) | |
249 | { | |
250 | test_ostream os(testfile, TEST_STREAM_FLAGS); | |
251 | test_oarchive oa(os, TEST_ARCHIVE_FLAGS); | |
252 | ||
253 | oa.register_type(static_cast<type2 *>(NULL)); | |
254 | ||
255 | type1 instance1; | |
256 | type2 *pointer2 = new type2; | |
257 | ||
258 | BOOST_TRY { | |
259 | oa & BOOST_SERIALIZATION_NVP(instance1) & BOOST_SERIALIZATION_NVP(pointer2); | |
260 | } | |
261 | BOOST_CATCH(...) { | |
262 | BOOST_CHECK_MESSAGE(false, "unexpected exception"); | |
263 | } | |
264 | BOOST_CATCH_END | |
265 | ||
266 | delete pointer2; | |
267 | } | |
268 | ||
269 | // load a pointer from slot0 which has no pointer serializer | |
270 | void load_unregistered_pointer(const char *testfile) | |
271 | { | |
272 | test_istream is(testfile); | |
273 | test_iarchive ia(is); | |
274 | ||
275 | type1 instance1; | |
276 | type2 *pointer2(NULL); | |
277 | ||
278 | bool except = false; | |
279 | BOOST_TRY { | |
280 | ia & BOOST_SERIALIZATION_NVP(instance1) & BOOST_SERIALIZATION_NVP(pointer2); | |
281 | } | |
92f5a8d4 | 282 | BOOST_CATCH(boost::archive::archive_exception const& aex){ |
b32b8144 FG |
283 | except = true; |
284 | BOOST_CHECK_MESSAGE( | |
285 | std::strcmp(aex.what(), "unregistered class") == 0, | |
286 | "incorrect exception" | |
287 | ); | |
288 | } | |
289 | BOOST_CATCH_END | |
290 | BOOST_CHECK_MESSAGE(except, "lack of registration not detected !"); | |
291 | BOOST_CHECK_MESSAGE(NULL == pointer2, "expected failed load"); | |
292 | ||
293 | delete pointer2; | |
294 | } | |
295 | ||
7c673cae FG |
296 | int |
297 | test_main( int /* argc */, char* /* argv */[] ) | |
298 | { | |
299 | const char * testfile = boost::archive::tmpnam(NULL); | |
300 | BOOST_REQUIRE(NULL != testfile); | |
301 | save_unregistered1(testfile); | |
302 | // load_unregistered1(testfile); | |
303 | save_unregistered2(testfile); | |
304 | load_unregistered2(testfile); | |
305 | save_registered(testfile); | |
306 | load_registered(testfile); | |
b32b8144 FG |
307 | save_unregistered_pointer(testfile); |
308 | load_unregistered_pointer(testfile); | |
7c673cae FG |
309 | std::remove(testfile); |
310 | return EXIT_SUCCESS; | |
311 | } | |
312 | ||
313 | // EOF |