]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/lib/smartall.h
Win32 fix -- remove debug O_NONBLOCK code
[bacula/bacula] / bacula / src / lib / smartall.h
index b879905411844e68d73a64e149763188dd3805c2..98305061d5d0c34b181b60d444a90ba85570ac5a 100644 (file)
@@ -1,13 +1,13 @@
 /*
 
-        Definitions for the smart memory allocator
-  
+       Definitions for the smart memory allocator
+
      Version $Id$
 
 */
 
 /*
-   Copyright (C) 2000, 2001, 2002 Kern Sibbald and John Walker
+   Copyright (C) 2000-2004 Kern Sibbald and John Walker
 
    This program is free software; you can redistribute it and/or
    modify it under the terms of the GNU General Public License as
@@ -31,18 +31,21 @@ extern uint64_t sm_bytes;
 extern uint32_t sm_max_buffers;
 extern uint32_t sm_buffers;
 
-#ifdef SMARTALLOC
+#ifdef  SMARTALLOC
+#undef  SMARTALLOC
+#define SMARTALLOC SMARTALLOC
+
 
 extern void *sm_malloc(const char *fname, int lineno, unsigned int nbytes),
-            *sm_calloc(const char *fname, int lineno,
-                unsigned int nelem, unsigned int elsize),
-            *sm_realloc(const char *fname, int lineno, void *ptr, unsigned int size),
-            *actuallymalloc(unsigned int size),
-            *actuallycalloc(unsigned int nelem, unsigned int elsize),
-            *actuallyrealloc(void *ptr, unsigned int size);
+           *sm_calloc(const char *fname, int lineno,
+               unsigned int nelem, unsigned int elsize),
+           *sm_realloc(const char *fname, int lineno, void *ptr, unsigned int size),
+           *actuallymalloc(unsigned int size),
+           *actuallycalloc(unsigned int nelem, unsigned int elsize),
+           *actuallyrealloc(void *ptr, unsigned int size);
 extern void sm_free(const char *fname, int lineno, void *fp);
 extern void actuallyfree(void *cp),
-            sm_dump(bool bufdump), sm_static(int mode);
+           sm_dump(bool bufdump), sm_static(int mode);
 extern void sm_new_owner(const char *fname, int lineno, char *buf);
 
 #ifdef SMCHECK
@@ -81,7 +84,77 @@ extern int sm_check_rtn(const char *fname, int lineno, bool bufdump);
 #define sm_check_rtn(f, l, fl) 1
 
 extern void *b_malloc();
-#define malloc(x) b_malloc(__FILE__, __LINE__, (x))                  
+#define malloc(x) b_malloc(__FILE__, __LINE__, (x))
+
+
+#endif
+
+#ifdef SMARTALLOC
+// #ifdef xxx
+
+#define New(type) new(__FILE__, __LINE__) type
+
+class SMARTALLOC
+{
+public:
+
+void *operator new(size_t s, const char *fname, int line)
+{
+  void *p = sm_malloc(fname, line, s > sizeof(int) ? s : sizeof(int));
+  return p;
+}
+void *operator new[](size_t s, const char *fname, int line)
+{
+   void *p = sm_malloc(fname, line, s > sizeof(int) ? s : sizeof(int));
+   return p;
+}
+void  operator delete(void *ptr)
+{
+   free(ptr);
+}
+void  operator delete[](void *ptr, size_t i)
+{
+   free(ptr);
+}
+
+void  operator delete(void *ptr, const char *fname, int line)
+{
+   free(ptr);
+}
+void  operator delete[](void *ptr, size_t i, const char *fname, int line)
+{
+   free(ptr);
+}
+
+
+private:
+void *operator new(size_t s) throw() { return 0; }
+void *operator new[](size_t s) throw() { return 0; }
+};
 
 
+#else
+
+#define New(type) new type
+
+class SMARTALLOC
+{
+   public:
+      void *operator new(size_t s)
+      {
+         return malloc(s);
+      }
+      void *operator new[](size_t s)
+      {
+         return malloc(s);
+      }
+      void  operator delete(void *ptr)
+      {
+         free(ptr);
+      }
+      void  operator delete[](void *ptr, size_t i)
+      {
+         free(ptr);
+      }
+};
 #endif