Add lock feature for ostream

Ali-Mirghasemi · Jul 20, 2022 · f89f1ca · f89f1ca
1 parent 4b29031
commit f89f1ca
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 1 deletion.
diff --git a/Src/OutputStream.c b/Src/OutputStream.c
@@ -201,4 +201,28 @@ Stream_LenType OStream_space(OStream* stream) {
 Stream_LenType OStream_outgoingBytes(OStream* stream) {
     return stream->OutgoingBytes;
 }
-
+#if OSTREAM_LOCK
+/**
+ * @brief lock output stream for fixed write
+ * 
+ * @param stream 
+ * @param lock 
+ * @return Stream_Result 
+ */
+Stream_Result OStream_lock(OStream* stream, OStream* lock, Stream_LenType len) {
+    Stream_Result res;
+    if ((res = Stream_lockWrite(&stream->Buffer, &lock->Buffer, len) == Stream_Ok)) {
+        memcpy(&lock->transmit, &stream->transmit, sizeof(OStream) - sizeof(Stream));
+    }
+    return res;
+}
+/**
+ * @brief unlock output stream
+ * 
+ * @param stream 
+ * @param lock 
+ */
+void OStream_unlock(OStream* stream, OStream* lock) {
+    Stream_unlockWrite(&stream->Buffer, &lock->Buffer);
+}
+#endif // OSTREAM_LOCK
diff --git a/Src/OutputStream.h b/Src/OutputStream.h
@@ -33,6 +33,10 @@ extern "C" {
  * @brief enable checkTransmit function
  */
 #define OSTREAM_CHECK_TRANSMIT      1
+/**
+ * @brief enable lock write feature
+ */
+#define OSTREAM_LOCK                STREAM_WRITE_LOCK
 
 /************************************************************************/
 
@@ -89,6 +93,11 @@ Stream_LenType OStream_outgoingBytes(OStream* stream);
     void OStream_setCheckTransmit(OStream* stream, OStream_CheckTransmitFn fn);
 #endif // OSTREAM_CHECK_TRANSMIT
 
+#if OSTREAM_LOCK
+    Stream_Result OStream_lock(OStream* stream, OStream* lock, Stream_LenType len);
+    void          OStream_unlock(OStream* stream, OStream* lock);
+#endif // OSTREAM_LOCK
+
 /**
  * @brief return number of bytes waiting for transmit 
  */