switch to _exact_recv on publish(), assert self._pid instead of unbound pid

Author: brentru

adafruit_minimqtt/adafruit_minimqtt.py

@@ -642,14 +642,13 @@ def publish(self, topic, msg, retain=False, qos=0):
             while True:
                 op = self._wait_for_msg()
                 if op == 0x40:
-                    self._recv_into(buf, 1)
-                    assert buf == b"\x02"
-                    # rcv_pid = self._sock.recv(2)
-                    self._recv_into(buf, 2)
-                    buf = buf[0] << 0x08 | buf[1]
-                    if pid == buf:
+                    sz = self._sock_exact_recv(1)
+                    assert sz == b"\x02"
+                    rcv_pid = self._sock_exact_recv(2)
+                    rcv_pid = rcv_pid[0] << 0x08 | rcv_pid[1]
+                    if self._pid == rcv_pid:
                         if self.on_publish is not None:
-                            self.on_publish(self, self._user_data, topic, buf)
+                            self.on_publish(self, self._user_data, topic, rcv_pid)
                         return
 
     def subscribe(self, topic, qos=0):