extmod/modframebuf: Fix crash in scroll() for large inputs.
If mp_int_t is wider than int, then the tests such as `xend < 0` can fail even when the amount of scrolling requested is out of range. This resulted in a segmentation fault when attempting an out-of-bounds access to the framebuffer. Signed-off-by: Jeff Epler <jepler@gmail.com>
This commit is contained in:
parent
2bba507148
commit
e15219800e
2 changed files with 20 additions and 14 deletions
|
|
@ -787,39 +787,40 @@ static mp_obj_t framebuf_scroll(mp_obj_t self_in, mp_obj_t xstep_in, mp_obj_t ys
|
|||
mp_obj_framebuf_t *self = MP_OBJ_TO_PTR(self_in);
|
||||
mp_int_t xstep = mp_obj_get_int(xstep_in);
|
||||
mp_int_t ystep = mp_obj_get_int(ystep_in);
|
||||
int sx, y, xend, yend, dx, dy;
|
||||
unsigned int sx, y, xend, yend;
|
||||
int dx, dy;
|
||||
if (xstep < 0) {
|
||||
sx = 0;
|
||||
xend = self->width + xstep;
|
||||
if (xend <= 0) {
|
||||
if (-xstep >= self->width) {
|
||||
return mp_const_none;
|
||||
}
|
||||
sx = 0;
|
||||
xend = self->width + (int)xstep;
|
||||
dx = 1;
|
||||
} else {
|
||||
sx = self->width - 1;
|
||||
xend = xstep - 1;
|
||||
if (xend >= sx) {
|
||||
if (xstep >= self->width) {
|
||||
return mp_const_none;
|
||||
}
|
||||
sx = self->width - 1;
|
||||
xend = (int)xstep - 1;
|
||||
dx = -1;
|
||||
}
|
||||
if (ystep < 0) {
|
||||
y = 0;
|
||||
yend = self->height + ystep;
|
||||
if (yend <= 0) {
|
||||
if (-ystep >= self->height) {
|
||||
return mp_const_none;
|
||||
}
|
||||
y = 0;
|
||||
yend = self->height + (int)ystep;
|
||||
dy = 1;
|
||||
} else {
|
||||
y = self->height - 1;
|
||||
yend = ystep - 1;
|
||||
if (yend >= y) {
|
||||
if (ystep >= self->height) {
|
||||
return mp_const_none;
|
||||
}
|
||||
y = self->height - 1;
|
||||
yend = (int)ystep - 1;
|
||||
dy = -1;
|
||||
}
|
||||
for (; y != yend; y += dy) {
|
||||
for (int x = sx; x != xend; x += dx) {
|
||||
for (unsigned x = sx; x != xend; x += dx) {
|
||||
setpixel(self, x, y, getpixel(self, x - xstep, y - ystep));
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -42,4 +42,9 @@ prepare_buffer()
|
|||
fbuf.scroll(15, 7)
|
||||
fbuf.scroll(10, -1)
|
||||
fbuf.scroll(1, -10)
|
||||
try:
|
||||
fbuf.scroll(1000000000000, -1)
|
||||
except OverflowError:
|
||||
# When mp_int_t is 32 bits, this throws OverflowError.
|
||||
pass
|
||||
printbuf()
|
||||
|
|
|
|||
Loading…
Reference in a new issue