Protecting the IO class a bit more, and adding the raw filename paradigm for at least the Input class.

2 files changed, 6 insertions, 3 deletions

diff --git a/includes/Handle.h b/includes/Handle.h
index 96f1736..321d8db 100644
--- a/includes/Handle.h
+++ b/includes/Handle.h
@@ -63,7 +63,7 @@ class IOBase {
       IOBase() : m_h(NULL) { }
       ~IOBase() { if (m_h) m_h->delRef(); }
   protected:
-    void setHandle(Handle * h) { m_h = h; m_h->addRef(); }
+    void setHandle(Handle * h) { m_h = h; if (m_h) m_h->addRef(); }
     Handle * m_h;
     template<class T>
     friend class IO;
@@ -74,11 +74,12 @@ class IO : public IOBase {
   public:
       IO() { }
       IO(T * h) { setHandle(h); }
-      IO(const IO<T> & io) { setHandle(io.m_h); }
+      IO(const IO<T> & io) { if (io.m_h) setHandle(io.m_h); }
       template<class U>
-      IO(const IO<U> & io) { setHandle(io.m_h); }
+      IO(const IO<U> & io) { if (io.m_h) setHandle(io.m_h); }
     IO<T> & operator=(const IO<T> & io) { if (m_h) m_h->delRef(); setHandle(io.m_h); return *this; }
     T * operator->() { Assert(m_h); return dynamic_cast<T *>(m_h); }
+    bool isNull() { return dynamic_cast<T *>(m_h); }
 };
 
 class SeekableHandle : public Handle {
diff --git a/includes/Input.h b/includes/Input.h
index 518db39..9d45baa 100644
--- a/includes/Input.h
+++ b/includes/Input.h
@@ -14,9 +14,11 @@ class Input : public SeekableHandle {
     virtual const char * getName();
     virtual off_t getSize();
     virtual time_t getMTime();
+    const char * getFName() { return m_fname.to_charp(); }
   private:
     int m_fd;
     String m_name;
+    String m_fname;
     off_t m_size;
     time_t m_mtime;
 };