| 1 | #include <linux/export.h>
|
| 2 | #include <linux/bug.h>
|
| 3 | #include <linux/bitmap.h>
|
| 4 |
|
| 5 | |
| 6 | * memweight - count the total number of bits set in memory area
|
| 7 | * @ptr: pointer to the start of the area
|
| 8 | * @bytes: the size of the area
|
| 9 |
|
| 10 | size_t memweight(const void *ptr, size_t bytes)
|
| 11 | {
|
| 12 | size_t ret = 0;
|
| 13 | size_t longs;
|
| 14 | const unsigned char *bitmap = ptr;
|
| 15 |
|
| 16 | for (; bytes > 0 && ((unsigned long)bitmap) % sizeof(long);
|
| 17 | bytes--, bitmap++)
|
| 18 | ret += hweight8(*bitmap);
|
| 19 |
|
| 20 | longs = bytes / sizeof(long);
|
| 21 | if (longs) {
|
| 22 | BUG_ON(longs >= INT_MAX / BITS_PER_LONG);
|
| 23 | ret += bitmap_weight((unsigned long *)bitmap,
|
| 24 | longs * BITS_PER_LONG);
|
| 25 | bytes -= longs * sizeof(long);
|
| 26 | bitmap += longs * sizeof(long);
|
| 27 | }
|
| 28 | |
| 29 | * The reason that this last loop is distinct from the preceding
|
| 30 | * bitmap_weight() call is to compute 1-bits in the last region smaller
|
| 31 | * than sizeof(long) properly on big-endian systems.
|
| 32 |
|
| 33 | for (; bytes > 0; bytes--, bitmap++)
|
| 34 | ret += hweight8(*bitmap);
|
| 35 |
|
| 36 | return ret;
|
| 37 | }
|
| 38 | EXPORT_SYMBOL(memweight);
|