1 files changed, 35 insertions, 20 deletions

diff --git a/src/entropy/beos_stats/es_beos.cpp b/src/entropy/beos_stats/es_beos.cpp
index 80d5605d0..0fd9199bd 100644
--- a/src/entropy/beos_stats/es_beos.cpp
+++ b/src/entropy/beos_stats/es_beos.cpp
@@ -1,65 +1,80 @@
-/*************************************************
-* BeOS EntropySource Source File                 *
-* (C) 1999-2007 Jack Lloyd                       *
-*************************************************/
+/**
+* BeOS EntropySource Source File
+* (C) 1999-2008 Jack Lloyd
+*/
 
 #include <botan/es_beos.h>
+#include <botan/xor_buf.h>
+
 #include <kernel/OS.h>
 #include <kernel/image.h>
 #include <interface/InterfaceDefs.h>
 
 namespace Botan {
 
-/*************************************************
-* BeOS Fast Poll                                 *
-*************************************************/
-void BeOS_EntropySource::do_fast_poll()
+/**
+* BeOS Fast Poll
+*/
+u32bit BeOS_EntropySource::fast_poll(byte buf[], u32bit length)
    {
+   if(length == 0)
+      return 0;
+
+   u32bit buf_i = 0;
+
    system_info info_sys;
    get_system_info(&info_sys);
-   add_bytes(&info_sys, sizeof(system_info));
+   buf_i = xor_into_buf(buf, buf_i, length, info_sys);
 
    key_info info_key;
    get_key_info(&info_key);
-   add_bytes(&info_key, sizeof(key_info));
+   buf_i = xor_into_buf(buf, buf_i, length, key_info);
 
-   add_bytes(idle_time());
+   buf_i = xor_into_buf(buf, buf_i, length, idle_time());
+
+   return length;
    }
 
-/*************************************************
-* BeOS Slow Poll                                 *
-*************************************************/
-void BeOS_EntropySource::do_slow_poll()
+/**
+* BeOS slow poll
+*/
+u32bit BeOS_EntropySource::slow_poll(byte buf[], u32bit length)
    {
+   if(length == 0)
+      return 0;
+
+   u32bit buf_i = 0;
    team_info info_team;
    int32 cookie_team = 0;
 
    while(get_next_team_info(&cookie_team, &info_team) == B_OK)
       {
-      add_bytes(&info_team, sizeof(team_info));
+      buf_i = xor_into_buf(buf, buf_i, length, info_team);
 
       team_id id = info_team.team;
       int32 cookie = 0;
 
       thread_info info_thr;
       while(get_next_thread_info(id, &cookie, &info_thr) == B_OK)
-         add_bytes(&info_thr, sizeof(thread_info));
+         buf_i = xor_into_buf(buf, buf_i, length, info_thr);
 
       cookie = 0;
       image_info info_img;
       while(get_next_image_info(id, &cookie, &info_img) == B_OK)
-         add_bytes(&info_img, sizeof(image_info));
+         buf_i = xor_into_buf(buf, buf_i, length, info_img);
 
       cookie = 0;
       sem_info info_sem;
       while(get_next_sem_info(id, &cookie, &info_sem) == B_OK)
-         add_bytes(&info_sem, sizeof(sem_info));
+         buf_i = xor_into_buf(buf, buf_i, length, info_sem);
 
       cookie = 0;
       area_info info_area;
       while(get_next_area_info(id, &cookie, &info_area) == B_OK)
-         add_bytes(&info_area, sizeof(area_info));
+         buf_i = xor_into_buf(buf, buf_i, length, info_area);
       }
+
+   return length;
    }
 
 }