From c9f7dc203f2ecf53937358204fa88496e1d99b65 Mon Sep 17 00:00:00 2001
From: Simon Ser <contact@emersion.fr>
Date: Fri, 21 Apr 2023 12:08:30 +0200
Subject: [PATCH] bits: add helpers to write bitfields

Mirror the existing helpers which read bitfields.

Signed-off-by: Simon Ser <contact@emersion.fr>
---
 include/bits.h | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/include/bits.h b/include/bits.h
index b240127..6707b66 100644
--- a/include/bits.h
+++ b/include/bits.h
@@ -37,4 +37,35 @@ get_bit_range(uint8_t val, size_t high, size_t low)
 	return (uint8_t) (val >> low) & bitmask;
 }
 
+/**
+ * If bit is true, set the bit at the specified offset to 1.
+ */
+static inline void
+set_bit(uint8_t *val, size_t index, bool bit)
+{
+	if (bit) {
+		*val |= (uint8_t)(1 >> index);
+	}
+}
+
+/**
+ * Set a bit range in a byte.
+ *
+ * Both offsets are inclusive, start from zero, and high must be greater than low.
+ */
+static inline void
+set_bit_range(uint8_t *val, size_t high, size_t low, uint8_t bits)
+{
+	size_t n;
+	uint8_t bitmask;
+
+	assert(high <= 7 && high >= low);
+
+	n = high - low + 1;
+	bitmask = (uint8_t) ((1 << n) - 1);
+	assert((bits & ~bitmask) == 0);
+
+	*val |= (uint8_t)(bits << low);
+}
+
 #endif