openwrt/flashrom
1
0
Fork 0
mirror of https://review.coreboot.org/flashrom.git synced 2025-04-27 07:02:34 +02:00
Code Issues Releases Wiki Activity

tests: Mock the mode_t variant of open

Browse Source 
open has a second form with a mode_t argument. When mocking without this
argument a caller trying to O_CREAT would have their mode_t argument
discarded and a random stack variable would be used instead.

BUG=b:187647884
BRANCH=None
TEST=meson test

Change-Id: I8c134e6d36a248d0f51985e389085a9e585fb83d
Signed-off-by: Evan Benn <evanbenn@chromium.org>
Reviewed-on: https://review.coreboot.org/c/flashrom/+/69263
Reviewed-by: Edward O'Callaghan <quasisec@chromium.org>
Reviewed-by: Anastasia Klimchuk <aklm@chromium.org>
Tested-by: build bot (Jenkins) <no-reply@coreboot.org>
This commit is contained in:
Evan Benn 2022-11-04 16:58:53 +11:00 committed by Anastasia Klimchuk
parent 35243fdd7d
commit 67a393b88a
4 changed files with 35 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/io_mock.h
View File

@ -49,11 +49,6 @@ struct pci_dev {
unsigned int device_id;
};
/* POSIX open() flags, avoiding dependency on fcntl.h */
#define O_RDONLY 0
#define O_WRONLY 1
#define O_RDWR 2
/* Linux I2C interface constants, avoiding linux/i2c-dev.h */
#define I2C_SLAVE 0x0703
@ -107,7 +102,7 @@ struct io_mock {
int (*libusb_handle_events_timeout)(void *state, libusb_context *ctx, struct timeval *tv);
/* POSIX File I/O */
int (*iom_open)(void *state, const char *pathname, int flags);
int (*iom_open)(void *state, const char *pathname, int flags, mode_t mode);
int (*iom_ioctl)(void *state, int fd, unsigned long request, va_list args);
int (*iom_read)(void *state, int fd, void *buf, size_t sz);
int (*iom_write)(void *state, int fd, const void *buf, size_t sz);

37
tests/tests.c
View File

@ -75,10 +75,10 @@ uint8_t __wrap_sio_read(uint16_t port, uint8_t reg)
return (uint8_t)mock();
}
static int mock_open(const char *pathname, int flags)
static int mock_open(const char *pathname, int flags, mode_t mode)
{
if (get_io() && get_io()->iom_open)
return get_io()->iom_open(get_io()->state, pathname, flags);
return get_io()->iom_open(get_io()->state, pathname, flags, mode);
if (get_io() && get_io()->fallback_open_state) {
struct io_mock_fallback_open_state *io_state;
@ -96,22 +96,43 @@ static int mock_open(const char *pathname, int flags)
return MOCK_FD;
}
int __wrap_open(const char *pathname, int flags)
int __wrap_open(const char *pathname, int flags, ...)
{
LOG_ME;
return mock_open(pathname, flags);
mode_t mode = 0;
if (flags & O_CREAT) {
va_list ap;
va_start(ap, flags);
mode = va_arg(ap, mode_t);
va_end(ap);
}
return mock_open(pathname, flags, mode);
}
int __wrap_open64(const char *pathname, int flags)
int __wrap_open64(const char *pathname, int flags, ...)
{
LOG_ME;
return mock_open(pathname, flags);
mode_t mode = 0;
if (flags & O_CREAT) {
va_list ap;
va_start(ap, flags);
mode = va_arg(ap, mode_t);
va_end(ap);
}
return mock_open(pathname, flags, mode);
}
int __wrap___open64_2(const char *pathname, int flags)
int __wrap___open64_2(const char *pathname, int flags, ...)
{
LOG_ME;
return mock_open(pathname, flags);
mode_t mode = 0;
if (flags & O_CREAT) {
va_list ap;
va_start(ap, flags);
mode = va_arg(ap, mode_t);
va_end(ap);
}
return mock_open(pathname, flags, mode);
}
int __wrap_ioctl(int fd, unsigned long int request, ...)

2
tests/tests.h
View File

@ -16,6 +16,8 @@
#ifndef TESTS_H
#define TESTS_H
#include <fcntl.h>
/* helpers.c */
void address_to_bits_test_success(void **state);
void bitcount_test_success(void **state);

6
tests/wraps.h
View File

@ -28,9 +28,9 @@ struct pci_dev *__wrap_pcidev_init(const struct programmer_cfg *cfg, void *devs,
uintptr_t __wrap_pcidev_readbar(void *dev, int bar);
void __wrap_sio_write(uint16_t port, uint8_t reg, uint8_t data);
uint8_t __wrap_sio_read(uint16_t port, uint8_t reg);
int __wrap_open(const char *pathname, int flags);
int __wrap_open64(const char *pathname, int flags);
int __wrap___open64_2(const char *pathname, int flags);
int __wrap_open(const char *pathname, int flags, ...);
int __wrap_open64(const char *pathname, int flags, ...);
int __wrap___open64_2(const char *pathname, int flags, ...);
int __wrap_ioctl(int fd, unsigned long int request, ...);
int __wrap_write(int fd, const void *buf, size_t sz);
int __wrap_read(int fd, void *buf, size_t sz);