Mirror/ardupilot
5
0
Fork 0
mirror of https://github.com/ArduPilot/ardupilot synced 2025-02-22 15:53:56 -04:00
Code Issues Packages Projects Releases Wiki Activity

AP_Common: fix Bitmask out-of-range values

Browse Source
This commit is contained in:
squilter 2017-08-31 12:21:16 -04:00 committed by Randy Mackay
parent 963c9c93fe
commit 1a9bccaaa7
1 changed files with 17 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
libraries/AP_Common/Bitmask.h
View File

@ -20,9 +20,11 @@
class Bitmask {
public:
Bitmask(uint16_t numbits) :
numwords((numbits+31)/32) {
Bitmask(uint16_t num_bits) :
numbits(num_bits),
numwords((num_bits+31)/32) {
bits = new uint32_t[numwords];
clearall();
}
~Bitmask(void) {
delete[] bits;
@ -30,6 +32,10 @@ public:
// set given bitnumber
void set(uint16_t bit) {
// ignore an invalid bit number
if (bit >= numbits) {
return;
}
uint16_t word = bit/32;
uint8_t ofs = bit & 0x1f;
bits[word] |= (1U << ofs);
@ -37,9 +43,13 @@ public:
// set all bits
void setall(void) {
for (uint16_t i=0; i<numwords; i++) {
// set all words to 111... except the last one.
for (uint16_t i=0; i<numwords-1; i++) {
bits[i] = 0xffffffff;
}
// set most of the last word to 111.., leaving out-of-range bits to be 0
uint16_t num_valid_bits = numbits % 32;
bits[numwords-1] = (1 << num_valid_bits) - 1;
}
// clear given bitnumber
@ -51,9 +61,7 @@ public:
// clear all bits
void clearall(void) {
for (uint16_t i=0; i<numwords; i++) {
bits[i] = 0;
}
memset(bits, 0, numwords*sizeof(bits[0]));
}
// return true if given bitnumber is set
@ -88,12 +96,13 @@ public:
return sum;
}
// return number of bits available (may be slightly larger than number requested in constructor)
// return number of bits available
uint16_t size() const {
return numwords * 32;
return numbits;
}
private:
uint16_t numbits;
uint16_t numwords;
uint32_t *bits;
};