Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(327)

Unified Diff: src/platform/vboot_reference/include/utility.h

Issue 572024: Add a StatefulMemcpy which can be used to safely and iteratively copy blocks of memory. (Closed)
Patch Set: Must be consistent.. Created 10 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « src/platform/vboot_reference/common/utility_stub.c ('k') | src/platform/vboot_reference/utils/Makefile » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: src/platform/vboot_reference/include/utility.h
diff --git a/src/platform/vboot_reference/include/utility.h b/src/platform/vboot_reference/include/utility.h
index ddb1dc055fcce194ac66e7cc7b8ca1e2ece4fdd6..87d17eaa42733fd1f9fd0a49c772cb086ee38764 100644
--- a/src/platform/vboot_reference/include/utility.h
+++ b/src/platform/vboot_reference/include/utility.h
@@ -10,6 +10,7 @@
#ifndef VBOOT_REFERENCE_UTILITY_H_
#define VBOOT_REFERENCE_UTILITY_H_
+#include <inttypes.h>
#include <string.h>
/* Allocate [size] bytes and return a pointer to the allocated memory. Abort
@@ -23,10 +24,30 @@ void Free(void* ptr);
/* Copy [n] bytes from [src] to [dest]. */
void* Memcpy(void* dest, const void* src, size_t n);
+/* Set [n] bytes starting at [s] to [c]. */
+void* Memset(void *dest, const uint8_t c, size_t n);
+
/* Compare [n] bytes starting at [s1] with [s2] and return 1 if they match,
* 0 if they don't. Time taken to perform the comparison is only dependent on
* [n] and not on the relationship of the match between [s1] and [s2].
*/
int SafeMemcmp(const void* s1, const void* s2, size_t n);
+/* Track remaining data to be read in a buffer. */
+typedef struct MemcpyState {
+ void* remaining_buf;
+ int remaining_len;
+} MemcpyState;
+
+/* Copy [len] bytes into [dst] only if there's enough data to read according
+ * to [state].
+ * On success, return [dst] and update [state]..
+ * On failure, return NULL, set remaining len in state to -1.
+ *
+ * Useful for iterating through a binary blob to populate a struct. After the
+ * first failure (buffer overrun), successive calls will always fail.
+ */
+void* StatefulMemcpy(MemcpyState* state, void* dst, int len);
+
+
#endif /* VBOOT_REFERENCE_UTILITY_H_ */
« no previous file with comments | « src/platform/vboot_reference/common/utility_stub.c ('k') | src/platform/vboot_reference/utils/Makefile » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698