diff options
author | Chris Dyer <cdyer@cs.cmu.edu> | 2011-04-27 14:00:47 -0400 |
---|---|---|
committer | Chris Dyer <cdyer@cs.cmu.edu> | 2011-04-27 14:00:47 -0400 |
commit | 9cfc3be07edc49dbb5150c10958f50f544bbf943 (patch) | |
tree | aaabcbdbee5007a0de809f9742b19fcc6d046328 | |
parent | 213e851c6d26862bbbd1c21415559fd8697f9694 (diff) |
fix potential crash in FastSparseVector operator=
-rw-r--r-- | utils/fast_sparse_vector.h | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/utils/fast_sparse_vector.h b/utils/fast_sparse_vector.h index 00d3b74f..b9315235 100644 --- a/utils/fast_sparse_vector.h +++ b/utils/fast_sparse_vector.h @@ -25,8 +25,8 @@ // I have to avoid this since I want to use unions and c++-98 // does not let unions have types with constructors in them // this type bypasses default constructors. use with caution! -// this should work as long as T is in a acceptable state to -// have its destructor called when initialized with all zeros +// this should work as long as T does have a destructor that +// does anything template <typename T> struct PairIntT { const PairIntT& operator=(const std::pair<const int, T>& v) { @@ -104,9 +104,6 @@ class FastSparseVector { }; public: FastSparseVector() : local_size_(0), is_remote_(false) { std::memset(&data_, 0, sizeof(data_)); } - explicit FastSparseVector(const std::vector<T>& init) : local_size_(0), is_remote_(false) { - for (int i = 0; i < init.size(); ++i) set_value(i, init[i]); - } ~FastSparseVector() { clear(); } @@ -128,8 +125,8 @@ class FastSparseVector { } } } - const FastSparseVector& operator=(const FastSparseVector& other) { - if (is_remote_) delete data_.rbmap; + const FastSparseVector<T>& operator=(const FastSparseVector<T>& other) { + if (&other == this) return *this; std::memcpy(this, &other, sizeof(FastSparseVector)); if (is_remote_) data_.rbmap = new std::map<int, T>(*data_.rbmap); @@ -186,6 +183,7 @@ class FastSparseVector { } inline void clear() { if (is_remote_) delete data_.rbmap; + is_remote_ = false; local_size_ = 0; } inline bool empty() const { |