1 files changed, 19 insertions, 12 deletions
diff --git a/gr-blocks/lib/ctrlport_probe2_i_impl.cc b/gr-blocks/lib/ctrlport_probe2_i_impl.cc
index 086ebe7cf0..7e17d8e8fe 100644
--- a/gr-blocks/lib/ctrlport_probe2_i_impl.cc
+++ b/gr-blocks/lib/ctrlport_probe2_i_impl.cc
@@ -65,21 +65,26 @@ namespace gr {
     }
 
     std::vector<int>
-    ctrlport_probe2_i_impl::get() {
+    ctrlport_probe2_i_impl::get()
+    {
       return buffered_get.get();
     }
 
     void
     ctrlport_probe2_i_impl::set_length(int len)
     {
+      gr::thread::scoped_lock guard(d_setlock);
+
       if(len > 8191) {
-        std::cerr << "probe2_i: length " << len
-                  << " exceeds maximum buffer size of 8191" << std::endl;
+        GR_LOG_WARN(d_logger,
+                    boost::format("probe2_i: length %1% exceeds maximum"
+                                  " buffer size of 8191") % len);
         len = 8191;
       }
 
       d_len = len;
-      d_buffer.reserve(d_len);
+      d_buffer.resize(d_len);
+      d_index = 0;
     }
 
     int
@@ -94,20 +99,22 @@ namespace gr {
                                  gr_vector_void_star &output_items)
     {
       const int *in = (const int*)input_items[0];
+
+      gr::thread::scoped_lock guard(d_setlock);
+
       // copy samples to get buffer if we need samples
-      if(d_buffer.size() < d_len) {
+      if(d_index < d_len) {
         // copy smaller of remaining buffer space and num inputs to work()
-        int num_copy = std::min( (int)(d_len - d_buffer.size()), noutput_items );
+        int num_copy = std::min( (int)(d_len - d_index), noutput_items );
 
-        // TODO: convert this to a copy operator for speed...
-        for(int i = 0; i < num_copy; i++) {
-          d_buffer.push_back(in[i]);
-        }
+        memcpy(&d_buffer[d_index], in, num_copy*sizeof(int));
+        d_index += num_copy;
       }
 
       // notify the waiting get() if we fill up the buffer
-      if(d_buffer.size() == d_len) {
-          buffered_get.offer_data(d_buffer);
+      if(d_index == d_len) {
+        buffered_get.offer_data(d_buffer);
+        d_index = 0;
       }
 
       return noutput_items;