Allow to block client on full buffer.

Bug: 130665360
Change-Id: I2f52dbb1a86664c81c8d21c68db79eed36a637fb
diff --git a/src/profiling/memory/client.h b/src/profiling/memory/client.h
index 411484e..72b9c6c 100644
--- a/src/profiling/memory/client.h
+++ b/src/profiling/memory/client.h
@@ -98,6 +98,11 @@
   // Flush the contents of free_batch_. Must hold free_batch_lock_.
   bool FlushFreesLocked();
   bool SendControlSocketByte();
+  bool SendWireMessageWithRetriesIfBlocking(const WireMessage&);
+
+  // This is only valid for non-blocking sockets. This is when
+  // client_config_.block_client is true.
+  bool IsConnected();
 
   ClientConfiguration client_config_;
   // sampler_ operations are not thread-safe.