123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110 |
- //////////////////////////////////////////////////////////////////////////////
- //
- // (C) Copyright Ion Gaztanaga 2006-2012. Distributed under the Boost
- // Software License, Version 1.0. (See accompanying file
- // LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
- //
- // See http://www.boost.org/libs/interprocess for documentation.
- //
- //////////////////////////////////////////////////////////////////////////////
- #include <boost/interprocess/detail/config_begin.hpp>
- #include <boost/interprocess/detail/workaround.hpp>
- //[doc_bufferstream
- #include <boost/interprocess/managed_shared_memory.hpp>
- #include <boost/interprocess/streams/bufferstream.hpp>
- #include <vector>
- #include <iterator>
- #include <cstddef>
- //<-
- #include "../test/get_process_id_name.hpp"
- //->
- using namespace boost::interprocess;
- int main ()
- {
- //Remove shared memory on construction and destruction
- struct shm_remove
- {
- //<-
- #if 1
- shm_remove() { shared_memory_object::remove(test::get_process_id_name()); }
- ~shm_remove(){ shared_memory_object::remove(test::get_process_id_name()); }
- #else
- //->
- shm_remove() { shared_memory_object::remove("MySharedMemory"); }
- ~shm_remove(){ shared_memory_object::remove("MySharedMemory"); }
- //<-
- #endif
- //->
- } remover;
- //<-
- (void)remover;
- //->
- //Create shared memory
- //<-
- #if 1
- managed_shared_memory segment(create_only,test::get_process_id_name(), 65536);
- #else
- //->
- managed_shared_memory segment(create_only,
- "MySharedMemory", //segment name
- 65536);
- //<-
- #endif
- //->
- //Fill data
- std::vector<int> data;
- data.reserve(100);
- for(int i = 0; i < 100; ++i){
- data.push_back(i);
- }
- const std::size_t BufferSize = 100*5;
- //Allocate a buffer in shared memory to write data
- char *my_cstring =
- segment.construct<char>("MyCString")[BufferSize](0);
- bufferstream mybufstream(my_cstring, BufferSize);
- //Now write data to the buffer
- for(int i = 0; i < 100; ++i){
- mybufstream << data[i] << std::endl;
- }
- //Check there was no overflow attempt
- assert(mybufstream.good());
- //Extract all values from the shared memory string
- //directly to a vector.
- std::vector<int> data2;
- std::istream_iterator<int> it(mybufstream), itend;
- std::copy(it, itend, std::back_inserter(data2));
- //This extraction should have ended will fail error since
- //the numbers formatted in the buffer end before the end
- //of the buffer. (Otherwise it would trigger eofbit)
- assert(mybufstream.fail());
- //Compare data
- assert(std::equal(data.begin(), data.end(), data2.begin()));
- //Clear errors and rewind
- mybufstream.clear();
- mybufstream.seekp(0, std::ios::beg);
- //Now write again the data trying to do a buffer overflow
- for(int i = 0, m = data.size()*5; i < m; ++i){
- mybufstream << data[i%5] << std::endl;
- }
- //Now make sure badbit is active
- //which means overflow attempt.
- assert(!mybufstream.good());
- assert(mybufstream.bad());
- segment.destroy_ptr(my_cstring);
- return 0;
- }
- //]
- #include <boost/interprocess/detail/config_end.hpp>
|