ffi: Add sq_writer_stack_write_all.

2 files changed, 33 insertions, 1 deletions

diff --git a/ffi/include/sequoia/openpgp.h b/ffi/include/sequoia/openpgp.h
index f09ff1ea..23959b4b 100644
--- a/ffi/include/sequoia/openpgp.h
+++ b/ffi/include/sequoia/openpgp.h
@@ -955,11 +955,22 @@ sq_writer_stack_t sq_writer_stack_message (sq_writer_t writer);
 
 /*/
 /// Writes up to `len` bytes of `buf` into `writer`.
-o/*/
+/*/
 ssize_t sq_writer_stack_write (sq_context_t ctx, sq_writer_stack_t writer,
                                const uint8_t *buf, size_t len);
 
 /*/
+/// Writes up to `len` bytes of `buf` into `writer`.
+///
+/// Unlike sq_writer_stack_write, unless an error occurs, the whole
+/// buffer will be written.  Also, this version automatically catches
+/// EINTR.
+/*/
+sq_status_t sq_writer_stack_write_all (sq_context_t ctx,
+                                       sq_writer_stack_t writer,
+                                       const uint8_t *buf, size_t len);
+
+/*/
 /// Finalizes this writer, returning the underlying writer.
 /*/
 sq_writer_stack_t sq_writer_stack_finalize_one (sq_context_t ctx,
diff --git a/ffi/src/openpgp.rs b/ffi/src/openpgp.rs
index 07bf359d..28cbc726 100644
--- a/ffi/src/openpgp.rs
+++ b/ffi/src/openpgp.rs
@@ -1593,6 +1593,27 @@ pub extern "system" fn sq_writer_stack_write
     fry_or!(ctx, writer.write(buf).map_err(|e| e.into()), -1) as ssize_t
 }
 
+/// Writes up to `len` bytes of `buf` into `writer`.
+///
+/// Unlike sq_writer_stack_write, unless an error occurs, the whole
+/// buffer will be written.  Also, this version automatically catches
+/// EINTR.
+#[no_mangle]
+pub extern "system" fn sq_writer_stack_write_all
+    (ctx: Option<&mut Context>,
+     writer: Option<&mut writer::Stack<'static, Cookie>>,
+     buf: *const uint8_t, len: size_t)
+     -> Status
+{
+    let ctx = ctx.expect("Context is NULL");
+    let writer = writer.expect("Writer is NULL");
+    assert!(!buf.is_null());
+    let buf = unsafe {
+        slice::from_raw_parts(buf, len as usize)
+    };
+    fry_status!(ctx, writer.write_all(buf).map_err(|e| e.into()))
+}
+
 /// Finalizes this writer, returning the underlying writer.
 #[no_mangle]
 pub extern "system" fn sq_writer_stack_finalize_one