acpi_battery: use user_strlcpy in acpi_battery_read().

* also check for user addresses in acpi_battery_control().
This commit is contained in:
Jérôme Duval
2018-02-28 20:10:34 +01:00
parent 6470e36518
commit 959fdbd314
3 changed files with 34 additions and 15 deletions
@@ -1,6 +1,6 @@
SubDir HAIKU_TOP src add-ons kernel drivers power acpi_battery ; SubDir HAIKU_TOP src add-ons kernel drivers power acpi_battery ;
UsePrivateHeaders kernel ; UsePrivateKernelHeaders ;
KernelAddon acpi_battery : KernelAddon acpi_battery :
acpi_battery.cpp acpi_battery.cpp
@@ -17,6 +17,8 @@
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
#include <kernel.h>
#include "device/power_managment.h" #include "device/power_managment.h"
@@ -321,43 +323,46 @@ acpi_battery_read(void* _cookie, off_t position, void *buffer, size_t* numBytes)
ReadBatteryInfo(device->driver_cookie, &batteryInfo); ReadBatteryInfo(device->driver_cookie, &batteryInfo);
if (position == 0) { if (position == 0) {
size_t max_len = *numBytes; char string[512];
char *str = (char *)buffer; char *str = string;
size_t max_len = sizeof(string);
snprintf(str, max_len, "Battery Status:\n"); snprintf(str, max_len, "Battery Status:\n");
max_len-= strlen(str); max_len -= strlen(str);
str += strlen(str); str += strlen(str);
snprintf(str, max_len, " State %i, Current Rate %i, Capacity %i, " snprintf(str, max_len, " State %i, Current Rate %i, Capacity %i, "
"Voltage %i\n", batteryStatus.state, batteryStatus.current_rate, "Voltage %i\n", batteryStatus.state, batteryStatus.current_rate,
batteryStatus.capacity, batteryStatus.voltage); batteryStatus.capacity, batteryStatus.voltage);
max_len-= strlen(str); max_len -= strlen(str);
str += strlen(str); str += strlen(str);
snprintf(str, max_len, "\nBattery Info:\n"); snprintf(str, max_len, "\nBattery Info:\n");
max_len-= strlen(str); max_len -= strlen(str);
str += strlen(str); str += strlen(str);
snprintf(str, max_len, " Power Unit %i, Design Capacity %i, " snprintf(str, max_len, " Power Unit %i, Design Capacity %i, "
"Last Full Charge %i, Technology %i\n", batteryInfo.power_unit, "Last Full Charge %i, Technology %i\n", batteryInfo.power_unit,
batteryInfo.design_capacity, batteryInfo.last_full_charge, batteryInfo.design_capacity, batteryInfo.last_full_charge,
batteryInfo.technology); batteryInfo.technology);
max_len-= strlen(str); max_len -= strlen(str);
str += strlen(str); str += strlen(str);
snprintf(str, max_len, " Design Voltage %i, Design Capacity Warning %i, " snprintf(str, max_len, " Design Voltage %i, Design Capacity Warning %i, "
"Design Capacity Low %i, Capacity Granularity1 %i, " "Design Capacity Low %i, Capacity Granularity1 %i, "
"Capacity Granularity1 %i\n", batteryInfo.design_voltage, "Capacity Granularity1 %i\n", batteryInfo.design_voltage,
batteryInfo.design_capacity_warning, batteryInfo.design_capacity_low, batteryInfo.design_capacity_warning, batteryInfo.design_capacity_low,
batteryInfo.capacity_granularity_1, batteryInfo.capacity_granularity_1); batteryInfo.capacity_granularity_1, batteryInfo.capacity_granularity_1);
max_len-= strlen(str); max_len -= strlen(str);
str += strlen(str); str += strlen(str);
snprintf(str, max_len, " Model Number %s, Serial Number %s, " snprintf(str, max_len, " Model Number %s, Serial Number %s, "
"Type %s, OEM Info %s\n", batteryInfo.model_number, "Type %s, OEM Info %s\n", batteryInfo.model_number,
batteryInfo.serial_number, batteryInfo.type, batteryInfo.oem_info); batteryInfo.serial_number, batteryInfo.type, batteryInfo.oem_info);
max_len-= strlen(str); max_len -= strlen(str);
str += strlen(str); str += strlen(str);
*numBytes = strlen((char *)buffer); max_len = user_strlcpy((char*)buffer, string, *numBytes);
if (max_len < B_OK)
return B_BAD_ADDRESS;
*numBytes = max_len;
} else } else
*numBytes = 0; *numBytes = 0;
@@ -384,7 +389,11 @@ acpi_battery_control(void* _cookie, uint32 op, void* arg, size_t len)
return B_BAD_VALUE; return B_BAD_VALUE;
uint32 magicId = kMagicACPIBatteryID; uint32 magicId = kMagicACPIBatteryID;
return user_memcpy(arg, &magicId, sizeof(magicId)); if (!IS_USER_ADDRESS(arg)
|| user_memcpy(arg, &magicId, sizeof(magicId)) < B_OK) {
return B_BAD_ADDRESS;
}
return B_OK;
} }
case GET_BATTERY_INFO: { case GET_BATTERY_INFO: {
@@ -395,7 +404,12 @@ acpi_battery_control(void* _cookie, uint32 op, void* arg, size_t len)
err = ReadBatteryStatus(device->driver_cookie, &batteryInfo); err = ReadBatteryStatus(device->driver_cookie, &batteryInfo);
if (err != B_OK) if (err != B_OK)
return err; return err;
return user_memcpy(arg, &batteryInfo, sizeof(batteryInfo)); if (!IS_USER_ADDRESS(arg)
|| user_memcpy(arg, &batteryInfo, sizeof(batteryInfo))
< B_OK) {
return B_BAD_ADDRESS;
}
return B_OK;
} }
case GET_EXTENDED_BATTERY_INFO: { case GET_EXTENDED_BATTERY_INFO: {
@@ -406,7 +420,12 @@ acpi_battery_control(void* _cookie, uint32 op, void* arg, size_t len)
err = ReadBatteryInfo(device->driver_cookie, &extBatteryInfo); err = ReadBatteryInfo(device->driver_cookie, &extBatteryInfo);
if (err != B_OK) if (err != B_OK)
return err; return err;
return user_memcpy(arg, &extBatteryInfo, sizeof(extBatteryInfo)); if (!IS_USER_ADDRESS(arg)
|| user_memcpy(arg, &extBatteryInfo, sizeof(extBatteryInfo))
< B_OK) {
return B_BAD_ADDRESS;
}
return B_OK;
} }
case WATCH_BATTERY: case WATCH_BATTERY:
@@ -210,7 +210,7 @@ acpi_button_select(void *_cookie, uint8 event, selectsync *sync)
status_t error = add_select_sync_pool_entry(&device->select_pool, sync, status_t error = add_select_sync_pool_entry(&device->select_pool, sync,
event); event);
if (error != B_OK) { if (error != B_OK) {
ERROR("add_select_sync_pool_entry() failed: %" B_PRId32 "\n", error); ERROR("add_select_sync_pool_entry() failed: %" B_PRIx32 "\n", error);
return error; return error;
} }