Merge pull request #249 from redboltz/replace_assert_with_ow

Replaced BOOST_ASSERT with overwrite.
redboltz · May 8, 2019 · 58d6681 · 58d6681
2 parents a29da19 + e54f2e4
commit 58d6681
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/include/mqtt/endpoint.hpp b/include/mqtt/endpoint.hpp
@@ -10172,7 +10172,24 @@ class endpoint : public std::enable_shared_from_this<endpoint<Socket, Mutex, Loc
                         control_packet_type::pubcomp,
                         msg
                     );
-                    BOOST_ASSERT(ret.second);
+                    // publish store is erased when pubrec is received.
+                    // pubrel store is erased when pubcomp is received.
+                    // If invalid client send pubrec twice with the same packet id,
+                    // then send corresponding pubrel twice is a possible client/server
+                    // implementation.
+                    // In this case, overwrite store_.
+                    if (!ret.second) {
+                        store_.modify(
+                            ret.first,
+                            [&] (auto& e) {
+                                e = store(
+                                    packet_id,
+                                    control_packet_type::pubcomp,
+                                    msg
+                                );
+                            }
+                        );
+                    }
                 }
 
                 if (serialize) {