kernel: disk_device_manager: clean user interface

* avoid assignment in if statements
* use boolean conditions
* use copy_from_user_value/copy_to_user_value instead of user_memcpy
* no functional changes except user buffer addresses check in:
_user_get_next_disk_device_id(), _user_get_disk_system_info(),
_user_get_next_disk_system_info(), _user_find_disk_system()
* remove TODO "Add user address checks and check return values of user_memcpy()!".
This commit is contained in:
Jérôme Duval
2018-01-06 21:38:16 +01:00
parent 74fed5b52c
commit 4644c7cc92
@@ -38,9 +38,6 @@ using namespace BPrivate::DiskDevice;
#define DUMMY_JOB_ID 0 #define DUMMY_JOB_ID 0
// TODO: Add user address checks and check return values of user_memcpy()!
/*! \brief Wrapper around user_strlcpy() that returns a status_t /*! \brief Wrapper around user_strlcpy() that returns a status_t
indicating appropriate success or failure. indicating appropriate success or failure.
@@ -65,7 +62,7 @@ template<typename Type>
static inline status_t static inline status_t
copy_from_user_value(Type& value, const Type* userValue) copy_from_user_value(Type& value, const Type* userValue)
{ {
if (!userValue) if (userValue == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
if (!IS_USER_ADDRESS(userValue)) if (!IS_USER_ADDRESS(userValue))
@@ -79,7 +76,7 @@ template<typename Type>
static inline status_t static inline status_t
copy_to_user_value(Type* userValue, const Type& value) copy_to_user_value(Type* userValue, const Type& value)
{ {
if (!userValue) if (userValue == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
if (!IS_USER_ADDRESS(userValue)) if (!IS_USER_ADDRESS(userValue))
@@ -181,10 +178,10 @@ move_descendants_contents(KPartition *partition)
partition_id partition_id
_user_get_next_disk_device_id(int32 *_cookie, size_t *neededSize) _user_get_next_disk_device_id(int32 *_cookie, size_t *neededSize)
{ {
if (!_cookie)
return B_BAD_VALUE;
int32 cookie; int32 cookie;
user_memcpy(&cookie, _cookie, sizeof(cookie)); status_t error = copy_from_user_value(cookie, _cookie);
if (error != B_OK)
return error;
partition_id id = B_ENTRY_NOT_FOUND; partition_id id = B_ENTRY_NOT_FOUND;
KDiskDeviceManager *manager = KDiskDeviceManager::Default(); KDiskDeviceManager *manager = KDiskDeviceManager::Default();
@@ -192,7 +189,7 @@ _user_get_next_disk_device_id(int32 *_cookie, size_t *neededSize)
if (KDiskDevice *device = manager->RegisterNextDevice(&cookie)) { if (KDiskDevice *device = manager->RegisterNextDevice(&cookie)) {
PartitionRegistrar _(device, true); PartitionRegistrar _(device, true);
id = device->ID(); id = device->ID();
if (neededSize) { if (neededSize != NULL) {
if (DeviceReadLocker locker = device) { if (DeviceReadLocker locker = device) {
// get the needed size // get the needed size
UserDataWriter writer; UserDataWriter writer;
@@ -201,12 +198,14 @@ _user_get_next_disk_device_id(int32 *_cookie, size_t *neededSize)
writer.AllocatedSize()); writer.AllocatedSize());
if (status != B_OK) if (status != B_OK)
return status; return status;
} else { } else
id = B_ERROR; id = B_ERROR;
} }
} }
}
user_memcpy(_cookie, &cookie, sizeof(cookie)); error = copy_to_user_value(_cookie, cookie);
if (error != B_OK)
return error;
return id; return id;
} }
@@ -225,7 +224,7 @@ _user_find_disk_device(const char *_filename, size_t *neededSize)
if (KDiskDevice *device = manager->RegisterDevice(filename)) { if (KDiskDevice *device = manager->RegisterDevice(filename)) {
PartitionRegistrar _(device, true); PartitionRegistrar _(device, true);
id = device->ID(); id = device->ID();
if (neededSize) { if (neededSize != NULL) {
if (DeviceReadLocker locker = device) { if (DeviceReadLocker locker = device) {
// get the needed size // get the needed size
UserDataWriter writer; UserDataWriter writer;
@@ -249,17 +248,17 @@ _user_find_partition(const char *_filename, size_t *neededSize)
if (error != B_OK) if (error != B_OK)
return error; return error;
partition_id id = B_ENTRY_NOT_FOUND;
KDiskDeviceManager *manager = KDiskDeviceManager::Default(); KDiskDeviceManager *manager = KDiskDeviceManager::Default();
// find the partition // find the partition
if (KPartition *partition = manager->RegisterPartition(filename)) { KPartition *partition = manager->RegisterPartition(filename);
if (partition == NULL)
return B_ENTRY_NOT_FOUND;
PartitionRegistrar _(partition, true); PartitionRegistrar _(partition, true);
id = partition->ID(); partition_id id = partition->ID();
if (neededSize) { if (neededSize != NULL) {
// get and lock the partition's device // get and lock the partition's device
KDiskDevice *device = manager->RegisterDevice(partition->ID(), KDiskDevice *device = manager->RegisterDevice(partition->ID(), false);
false); if (device == NULL)
if (!device)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar _2(device, true); PartitionRegistrar _2(device, true);
if (DeviceReadLocker locker = device) { if (DeviceReadLocker locker = device) {
@@ -272,7 +271,6 @@ _user_find_partition(const char *_filename, size_t *neededSize)
} else } else
return B_ERROR; return B_ERROR;
} }
}
return id; return id;
} }
@@ -287,13 +285,14 @@ _user_find_file_disk_device(const char *_filename, size_t *neededSize)
KPath path(filename, KPath::NORMALIZE); KPath path(filename, KPath::NORMALIZE);
partition_id id = B_ENTRY_NOT_FOUND;
KDiskDeviceManager *manager = KDiskDeviceManager::Default(); KDiskDeviceManager *manager = KDiskDeviceManager::Default();
// find the device // find the device
if (KFileDiskDevice* device = manager->RegisterFileDevice(path.Path())) { KFileDiskDevice* device = manager->RegisterFileDevice(path.Path());
if (device == NULL)
return B_ENTRY_NOT_FOUND;
PartitionRegistrar _(device, true); PartitionRegistrar _(device, true);
id = device->ID(); partition_id id = device->ID();
if (neededSize) { if (neededSize != NULL) {
if (DeviceReadLocker locker = device) { if (DeviceReadLocker locker = device) {
// get the needed size // get the needed size
UserDataWriter writer; UserDataWriter writer;
@@ -304,7 +303,6 @@ _user_find_file_disk_device(const char *_filename, size_t *neededSize)
} else } else
return B_ERROR; return B_ERROR;
} }
}
return id; return id;
} }
@@ -351,30 +349,35 @@ status_t
_user_get_disk_device_data(partition_id id, bool deviceOnly, _user_get_disk_device_data(partition_id id, bool deviceOnly,
user_disk_device_data *buffer, size_t bufferSize, size_t *_neededSize) user_disk_device_data *buffer, size_t bufferSize, size_t *_neededSize)
{ {
if (!buffer && bufferSize > 0) if (buffer == NULL && bufferSize > 0)
return B_BAD_VALUE; return B_BAD_VALUE;
KDiskDeviceManager *manager = KDiskDeviceManager::Default(); KDiskDeviceManager *manager = KDiskDeviceManager::Default();
// get the device // get the device
if (KDiskDevice *device = manager->RegisterDevice(id, deviceOnly)) { KDiskDevice *device = manager->RegisterDevice(id, deviceOnly);
if (device == NULL)
return B_ENTRY_NOT_FOUND;
PartitionRegistrar _(device, true); PartitionRegistrar _(device, true);
if (DeviceReadLocker locker = device) { if (DeviceReadLocker locker = device) {
// do a dry run first to get the needed size // do a dry run first to get the needed size
UserDataWriter writer; UserDataWriter writer;
device->WriteUserData(writer); device->WriteUserData(writer);
size_t neededSize = writer.AllocatedSize(); size_t neededSize = writer.AllocatedSize();
if (_neededSize) { if (_neededSize != NULL) {
status_t error = copy_ref_var_to_user(neededSize, _neededSize); status_t error = copy_ref_var_to_user(neededSize, _neededSize);
if (error != B_OK) if (error != B_OK)
return error; return error;
} }
// if no buffer has been supplied or the buffer is too small, // if no buffer has been supplied or the buffer is too small,
// then we're done // then we're done
if (!buffer || bufferSize < neededSize) if (buffer == NULL || bufferSize < neededSize)
return B_BUFFER_OVERFLOW; return B_BUFFER_OVERFLOW;
if (!IS_USER_ADDRESS(buffer))
return B_BAD_ADDRESS;
// otherwise allocate a kernel buffer // otherwise allocate a kernel buffer
user_disk_device_data *kernelBuffer user_disk_device_data *kernelBuffer
= static_cast<user_disk_device_data*>(malloc(neededSize)); = static_cast<user_disk_device_data*>(malloc(neededSize));
if (!kernelBuffer) if (kernelBuffer == NULL)
return B_NO_MEMORY; return B_NO_MEMORY;
MemoryDeleter deleter(kernelBuffer); MemoryDeleter deleter(kernelBuffer);
// write the device data into the buffer // write the device data into the buffer
@@ -391,13 +394,10 @@ _user_get_disk_device_data(partition_id id, bool deviceOnly,
if (error != B_OK) if (error != B_OK)
return error; return error;
// copy out // copy out
if (buffer)
return user_memcpy(buffer, kernelBuffer, neededSize); return user_memcpy(buffer, kernelBuffer, neededSize);
} else } else
return B_ERROR; return B_ERROR;
} }
return B_ENTRY_NOT_FOUND;
}
partition_id partition_id
@@ -412,7 +412,8 @@ _user_register_file_device(const char *_filename)
KDiskDeviceManager *manager = KDiskDeviceManager::Default(); KDiskDeviceManager *manager = KDiskDeviceManager::Default();
if (ManagerLocker locker = manager) { if (ManagerLocker locker = manager) {
if (KFileDiskDevice *device = manager->FindFileDevice(path.Path())) KFileDiskDevice *device = manager->FindFileDevice(path.Path());
if (device != NULL)
return device->ID(); return device->ID();
return manager->CreateFileDevice(path.Path()); return manager->CreateFileDevice(path.Path());
} }
@@ -423,12 +424,12 @@ _user_register_file_device(const char *_filename)
status_t status_t
_user_unregister_file_device(partition_id deviceID, const char *_filename) _user_unregister_file_device(partition_id deviceID, const char *_filename)
{ {
if (deviceID < 0 && !_filename) if (deviceID < 0 && _filename == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
KDiskDeviceManager *manager = KDiskDeviceManager::Default(); KDiskDeviceManager *manager = KDiskDeviceManager::Default();
if (deviceID >= 0) { if (deviceID >= 0)
return manager->DeleteFileDevice(deviceID); return manager->DeleteFileDevice(deviceID);
} else {
UserStringParameter<false> filename; UserStringParameter<false> filename;
status_t error = filename.Init(_filename, B_PATH_NAME_LENGTH); status_t error = filename.Init(_filename, B_PATH_NAME_LENGTH);
if (error != B_OK) if (error != B_OK)
@@ -436,7 +437,6 @@ _user_unregister_file_device(partition_id deviceID, const char *_filename)
return manager->DeleteFileDevice(filename); return manager->DeleteFileDevice(filename);
} }
}
status_t status_t
@@ -445,10 +445,12 @@ _user_get_file_disk_device_path(partition_id id, char* buffer,
{ {
if (id < 0 || buffer == NULL || bufferSize == 0) if (id < 0 || buffer == NULL || bufferSize == 0)
return B_BAD_VALUE; return B_BAD_VALUE;
if (!IS_USER_ADDRESS(buffer))
return B_BAD_ADDRESS;
KDiskDeviceManager *manager = KDiskDeviceManager::Default(); KDiskDeviceManager *manager = KDiskDeviceManager::Default();
KDiskDevice *device = manager->RegisterDevice(id, true);
if (KDiskDevice *device = manager->RegisterDevice(id, true)) { if (device != NULL) {
PartitionRegistrar _(device, true); PartitionRegistrar _(device, true);
if (DeviceReadLocker locker = device) { if (DeviceReadLocker locker = device) {
KFileDiskDevice* fileDevice KFileDiskDevice* fileDevice
@@ -471,15 +473,15 @@ _user_get_file_disk_device_path(partition_id id, char* buffer,
status_t status_t
_user_get_disk_system_info(disk_system_id id, user_disk_system_info *_info) _user_get_disk_system_info(disk_system_id id, user_disk_system_info *_info)
{ {
if (!_info) if (_info == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
KDiskDeviceManager *manager = KDiskDeviceManager::Default(); KDiskDeviceManager *manager = KDiskDeviceManager::Default();
if (ManagerLocker locker = manager) { if (ManagerLocker locker = manager) {
if (KDiskSystem *diskSystem = manager->FindDiskSystem(id)) { KDiskSystem *diskSystem = manager->FindDiskSystem(id);
if (diskSystem != NULL) {
user_disk_system_info info; user_disk_system_info info;
diskSystem->GetInfo(&info); diskSystem->GetInfo(&info);
user_memcpy(_info, &info, sizeof(info)); return copy_to_user_value(_info, info);
return B_OK;
} }
} }
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
@@ -489,21 +491,27 @@ _user_get_disk_system_info(disk_system_id id, user_disk_system_info *_info)
status_t status_t
_user_get_next_disk_system_info(int32 *_cookie, user_disk_system_info *_info) _user_get_next_disk_system_info(int32 *_cookie, user_disk_system_info *_info)
{ {
if (!_cookie || !_info) if (_info == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
if (!IS_USER_ADDRESS(_info))
return B_BAD_ADDRESS;
int32 cookie; int32 cookie;
user_memcpy(&cookie, _cookie, sizeof(cookie)); status_t result = copy_from_user_value(cookie, _cookie);
status_t result = B_ENTRY_NOT_FOUND; if (result != B_OK)
return result;
result = B_ENTRY_NOT_FOUND;
KDiskDeviceManager *manager = KDiskDeviceManager::Default(); KDiskDeviceManager *manager = KDiskDeviceManager::Default();
if (ManagerLocker locker = manager) { if (ManagerLocker locker = manager) {
if (KDiskSystem *diskSystem = manager->NextDiskSystem(&cookie)) { KDiskSystem *diskSystem = manager->NextDiskSystem(&cookie);
if (diskSystem != NULL) {
user_disk_system_info info; user_disk_system_info info;
diskSystem->GetInfo(&info); diskSystem->GetInfo(&info);
user_memcpy(_info, &info, sizeof(info)); result = copy_to_user_value(_info, info);
result = B_OK;
} }
} }
user_memcpy(_cookie, &cookie, sizeof(cookie)); status_t error = copy_to_user_value(_cookie, cookie);
if (error != B_OK)
result = error;
return result; return result;
} }
@@ -511,19 +519,21 @@ _user_get_next_disk_system_info(int32 *_cookie, user_disk_system_info *_info)
status_t status_t
_user_find_disk_system(const char *_name, user_disk_system_info *_info) _user_find_disk_system(const char *_name, user_disk_system_info *_info)
{ {
if (!_name || !_info) if (_name == NULL || _info == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
if (!IS_USER_ADDRESS(_name) || !IS_USER_ADDRESS(_info))
return B_BAD_ADDRESS;
char name[B_DISK_SYSTEM_NAME_LENGTH]; char name[B_DISK_SYSTEM_NAME_LENGTH];
status_t error = ddm_strlcpy(name, _name, B_DISK_SYSTEM_NAME_LENGTH); status_t error = ddm_strlcpy(name, _name, B_DISK_SYSTEM_NAME_LENGTH);
if (error) if (error != B_OK)
return error; return error;
KDiskDeviceManager *manager = KDiskDeviceManager::Default(); KDiskDeviceManager *manager = KDiskDeviceManager::Default();
if (ManagerLocker locker = manager) { if (ManagerLocker locker = manager) {
if (KDiskSystem *diskSystem = manager->FindDiskSystem(name)) { KDiskSystem *diskSystem = manager->FindDiskSystem(name);
if (diskSystem != NULL) {
user_disk_system_info info; user_disk_system_info info;
diskSystem->GetInfo(&info); diskSystem->GetInfo(&info);
user_memcpy(_info, &info, sizeof(info)); return copy_to_user_value(_info, info);
return B_OK;
} }
} }
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
@@ -536,14 +546,14 @@ _user_defragment_partition(partition_id partitionID, int32* _changeCounter)
// copy parameters in // copy parameters in
int32 changeCounter; int32 changeCounter;
status_t error; status_t error = copy_from_user_value(changeCounter, _changeCounter);
if ((error = copy_from_user_value(changeCounter, _changeCounter)) != B_OK) if (error != B_OK)
return error; return error;
// get the partition // get the partition
KDiskDeviceManager* manager = KDiskDeviceManager::Default(); KDiskDeviceManager* manager = KDiskDeviceManager::Default();
KPartition* partition = manager->WriteLockPartition(partitionID); KPartition* partition = manager->WriteLockPartition(partitionID);
if (!partition) if (partition == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar1(partition, true); PartitionRegistrar registrar1(partition, true);
@@ -556,7 +566,7 @@ _user_defragment_partition(partition_id partitionID, int32* _changeCounter)
// the partition must be initialized // the partition must be initialized
KDiskSystem* diskSystem = partition->DiskSystem(); KDiskSystem* diskSystem = partition->DiskSystem();
if (!diskSystem) if (diskSystem == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
// mark the partition busy and unlock // mark the partition busy and unlock
@@ -575,12 +585,7 @@ _user_defragment_partition(partition_id partitionID, int32* _changeCounter)
return error; return error;
// return change counter // return change counter
if ((error = copy_to_user_value(_changeCounter, partition->ChangeCounter())) return copy_to_user_value(_changeCounter, partition->ChangeCounter());
!= B_OK) {
return error;
}
return B_OK;
} }
@@ -591,14 +596,14 @@ _user_repair_partition(partition_id partitionID, int32* _changeCounter,
// copy parameters in // copy parameters in
int32 changeCounter; int32 changeCounter;
status_t error; status_t error = copy_from_user_value(changeCounter, _changeCounter);
if ((error = copy_from_user_value(changeCounter, _changeCounter)) != B_OK) if (error != B_OK)
return error; return error;
// get the partition // get the partition
KDiskDeviceManager* manager = KDiskDeviceManager::Default(); KDiskDeviceManager* manager = KDiskDeviceManager::Default();
KPartition* partition = manager->WriteLockPartition(partitionID); KPartition* partition = manager->WriteLockPartition(partitionID);
if (!partition) if (partition == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar1(partition, true); PartitionRegistrar registrar1(partition, true);
@@ -611,7 +616,7 @@ _user_repair_partition(partition_id partitionID, int32* _changeCounter,
// the partition must be initialized // the partition must be initialized
KDiskSystem* diskSystem = partition->DiskSystem(); KDiskSystem* diskSystem = partition->DiskSystem();
if (!diskSystem) if (diskSystem == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
// mark the partition busy and unlock // mark the partition busy and unlock
@@ -630,12 +635,7 @@ _user_repair_partition(partition_id partitionID, int32* _changeCounter,
return error; return error;
// return change counter // return change counter
if ((error = copy_to_user_value(_changeCounter, partition->ChangeCounter())) return copy_to_user_value(_changeCounter, partition->ChangeCounter());
!= B_OK) {
return error;
}
return B_OK;
} }
@@ -648,17 +648,16 @@ _user_resize_partition(partition_id partitionID, int32* _changeCounter,
int32 changeCounter; int32 changeCounter;
int32 childChangeCounter; int32 childChangeCounter;
status_t error; status_t error = copy_from_user_value(changeCounter, _changeCounter);
if ((error = copy_from_user_value(changeCounter, _changeCounter)) != B_OK if (error == B_OK)
|| (error = copy_from_user_value(childChangeCounter, error = copy_from_user_value(childChangeCounter, _childChangeCounter);
_childChangeCounter)) != B_OK) { if (error != B_OK)
return error; return error;
}
// get the partition // get the partition
KDiskDeviceManager* manager = KDiskDeviceManager::Default(); KDiskDeviceManager* manager = KDiskDeviceManager::Default();
KPartition* partition = manager->WriteLockPartition(partitionID); KPartition* partition = manager->WriteLockPartition(partitionID);
if (!partition) if (partition == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar1(partition, true); PartitionRegistrar registrar1(partition, true);
@@ -667,7 +666,7 @@ _user_resize_partition(partition_id partitionID, int32* _changeCounter,
// register child // register child
KPartition* child = manager->RegisterPartition(childID); KPartition* child = manager->RegisterPartition(childID);
if (!child) if (child == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar3(child, true); PartitionRegistrar registrar3(child, true);
@@ -680,7 +679,7 @@ _user_resize_partition(partition_id partitionID, int32* _changeCounter,
// the partition must be initialized // the partition must be initialized
KDiskSystem* diskSystem = partition->DiskSystem(); KDiskSystem* diskSystem = partition->DiskSystem();
if (!diskSystem) if (diskSystem == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
// child must indeed be a child of partition // child must indeed be a child of partition
@@ -723,16 +722,12 @@ _user_resize_partition(partition_id partitionID, int32* _changeCounter,
return error; return error;
// return change counters // return change counters
if ((error = copy_to_user_value(_changeCounter, partition->ChangeCounter())) error = copy_to_user_value(_changeCounter, partition->ChangeCounter());
!= B_OK if (error == B_OK)
|| (error = copy_to_user_value(_childChangeCounter, error = copy_to_user_value(_childChangeCounter, child->ChangeCounter());
child->ChangeCounter())) != B_OK) {
return error; return error;
} }
return B_OK;
}
status_t status_t
_user_move_partition(partition_id partitionID, int32* changeCounter, _user_move_partition(partition_id partitionID, int32* changeCounter,
@@ -784,19 +779,18 @@ _user_set_partition_name(partition_id partitionID, int32* _changeCounter,
int32 changeCounter; int32 changeCounter;
int32 childChangeCounter; int32 childChangeCounter;
status_t error; status_t error = name.Init(_name, B_DISK_DEVICE_NAME_LENGTH);
if ((error = name.Init(_name, B_DISK_DEVICE_NAME_LENGTH)) != B_OK if (error == B_OK)
|| (error = copy_from_user_value(changeCounter, _changeCounter)) error = copy_from_user_value(changeCounter, _changeCounter);
!= B_OK if (error == B_OK)
|| (error = copy_from_user_value(childChangeCounter, error = copy_from_user_value(childChangeCounter, _childChangeCounter);
_childChangeCounter)) != B_OK) { if (error != B_OK)
return error; return error;
}
// get the partition // get the partition
KDiskDeviceManager* manager = KDiskDeviceManager::Default(); KDiskDeviceManager* manager = KDiskDeviceManager::Default();
KPartition* partition = manager->WriteLockPartition(partitionID); KPartition* partition = manager->WriteLockPartition(partitionID);
if (!partition) if (partition == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar1(partition, true); PartitionRegistrar registrar1(partition, true);
@@ -805,7 +799,7 @@ _user_set_partition_name(partition_id partitionID, int32* _changeCounter,
// register child // register child
KPartition* child = manager->RegisterPartition(childID); KPartition* child = manager->RegisterPartition(childID);
if (!child) if (child == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar3(child, true); PartitionRegistrar registrar3(child, true);
@@ -818,7 +812,7 @@ _user_set_partition_name(partition_id partitionID, int32* _changeCounter,
// the partition must be initialized // the partition must be initialized
KDiskSystem* diskSystem = partition->DiskSystem(); KDiskSystem* diskSystem = partition->DiskSystem();
if (!diskSystem) if (diskSystem == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
// child must indeed be a child of partition // child must indeed be a child of partition
@@ -844,16 +838,12 @@ _user_set_partition_name(partition_id partitionID, int32* _changeCounter,
return error; return error;
// return change counters // return change counters
if ((error = copy_to_user_value(_changeCounter, partition->ChangeCounter())) error = copy_to_user_value(_changeCounter, partition->ChangeCounter());
!= B_OK if (error == B_OK)
|| (error = copy_to_user_value(_childChangeCounter, error = copy_to_user_value(_childChangeCounter, child->ChangeCounter());
child->ChangeCounter())) != B_OK) {
return error; return error;
} }
return B_OK;
}
status_t status_t
_user_set_partition_content_name(partition_id partitionID, _user_set_partition_content_name(partition_id partitionID,
@@ -863,17 +853,16 @@ _user_set_partition_content_name(partition_id partitionID,
UserStringParameter<true> name; UserStringParameter<true> name;
int32 changeCounter; int32 changeCounter;
status_t error; status_t error = name.Init(_name, B_DISK_DEVICE_NAME_LENGTH);
if ((error = name.Init(_name, B_DISK_DEVICE_NAME_LENGTH)) != B_OK if (error == B_OK)
|| (error = copy_from_user_value(changeCounter, _changeCounter)) error = copy_from_user_value(changeCounter, _changeCounter);
!= B_OK) { if (error != B_OK)
return error; return error;
}
// get the partition // get the partition
KDiskDeviceManager* manager = KDiskDeviceManager::Default(); KDiskDeviceManager* manager = KDiskDeviceManager::Default();
KPartition* partition = manager->WriteLockPartition(partitionID); KPartition* partition = manager->WriteLockPartition(partitionID);
if (!partition) if (partition == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar1(partition, true); PartitionRegistrar registrar1(partition, true);
@@ -886,7 +875,7 @@ _user_set_partition_content_name(partition_id partitionID,
// the partition must be initialized // the partition must be initialized
KDiskSystem* diskSystem = partition->DiskSystem(); KDiskSystem* diskSystem = partition->DiskSystem();
if (!diskSystem) if (diskSystem == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
// mark the partition busy and unlock // mark the partition busy and unlock
@@ -905,12 +894,7 @@ _user_set_partition_content_name(partition_id partitionID,
return error; return error;
// return change counter // return change counter
if ((error = copy_to_user_value(_changeCounter, partition->ChangeCounter())) return copy_to_user_value(_changeCounter, partition->ChangeCounter());
!= B_OK) {
return error;
}
return B_OK;
} }
@@ -923,19 +907,19 @@ _user_set_partition_type(partition_id partitionID, int32* _changeCounter,
int32 changeCounter; int32 changeCounter;
int32 childChangeCounter; int32 childChangeCounter;
status_t error; status_t error = type.Init(_type, B_DISK_DEVICE_TYPE_LENGTH);
if ((error = type.Init(_type, B_DISK_DEVICE_TYPE_LENGTH)) != B_OK if (error == B_OK)
|| (error = copy_from_user_value(changeCounter, _changeCounter)) error = copy_from_user_value(changeCounter, _changeCounter);
!= B_OK if (error == B_OK)
|| (error = copy_from_user_value(childChangeCounter, error = copy_from_user_value(childChangeCounter, _childChangeCounter);
_childChangeCounter)) != B_OK) { if (error != B_OK)
return error; return error;
}
// get the partition // get the partition
KDiskDeviceManager* manager = KDiskDeviceManager::Default(); KDiskDeviceManager* manager = KDiskDeviceManager::Default();
KPartition* partition = manager->WriteLockPartition(partitionID); KPartition* partition = manager->WriteLockPartition(partitionID);
if (!partition) if (partition == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar1(partition, true); PartitionRegistrar registrar1(partition, true);
@@ -944,7 +928,7 @@ _user_set_partition_type(partition_id partitionID, int32* _changeCounter,
// register child // register child
KPartition* child = manager->RegisterPartition(childID); KPartition* child = manager->RegisterPartition(childID);
if (!child) if (child == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar3(child, true); PartitionRegistrar registrar3(child, true);
@@ -957,7 +941,7 @@ _user_set_partition_type(partition_id partitionID, int32* _changeCounter,
// the partition must be initialized // the partition must be initialized
KDiskSystem* diskSystem = partition->DiskSystem(); KDiskSystem* diskSystem = partition->DiskSystem();
if (!diskSystem) if (diskSystem == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
// child must indeed be a child of partition // child must indeed be a child of partition
@@ -983,16 +967,12 @@ _user_set_partition_type(partition_id partitionID, int32* _changeCounter,
return error; return error;
// return change counters // return change counters
if ((error = copy_to_user_value(_changeCounter, partition->ChangeCounter())) error = copy_to_user_value(_changeCounter, partition->ChangeCounter());
!= B_OK if (error == B_OK)
|| (error = copy_to_user_value(_childChangeCounter, error = copy_to_user_value(_childChangeCounter, child->ChangeCounter());
child->ChangeCounter())) != B_OK) {
return error; return error;
} }
return B_OK;
}
status_t status_t
_user_set_partition_parameters(partition_id partitionID, int32* _changeCounter, _user_set_partition_parameters(partition_id partitionID, int32* _changeCounter,
@@ -1003,20 +983,19 @@ _user_set_partition_parameters(partition_id partitionID, int32* _changeCounter,
int32 changeCounter; int32 changeCounter;
int32 childChangeCounter; int32 childChangeCounter;
status_t error; status_t error
if ((error = parameters.Init(_parameters, B_DISK_DEVICE_MAX_PARAMETER_SIZE)) = parameters.Init(_parameters, B_DISK_DEVICE_MAX_PARAMETER_SIZE);
!= B_OK if (error == B_OK)
|| (error = copy_from_user_value(changeCounter, _changeCounter)) error = copy_from_user_value(changeCounter, _changeCounter);
!= B_OK if (error == B_OK)
|| (error = copy_from_user_value(childChangeCounter, error = copy_from_user_value(childChangeCounter, _childChangeCounter);
_childChangeCounter)) != B_OK) { if (error != B_OK)
return error; return error;
}
// get the partition // get the partition
KDiskDeviceManager* manager = KDiskDeviceManager::Default(); KDiskDeviceManager* manager = KDiskDeviceManager::Default();
KPartition* partition = manager->WriteLockPartition(partitionID); KPartition* partition = manager->WriteLockPartition(partitionID);
if (!partition) if (partition == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar1(partition, true); PartitionRegistrar registrar1(partition, true);
@@ -1025,7 +1004,7 @@ _user_set_partition_parameters(partition_id partitionID, int32* _changeCounter,
// register child // register child
KPartition* child = manager->RegisterPartition(childID); KPartition* child = manager->RegisterPartition(childID);
if (!child) if (child == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar3(child, true); PartitionRegistrar registrar3(child, true);
@@ -1038,7 +1017,7 @@ _user_set_partition_parameters(partition_id partitionID, int32* _changeCounter,
// the partition must be initialized // the partition must be initialized
KDiskSystem* diskSystem = partition->DiskSystem(); KDiskSystem* diskSystem = partition->DiskSystem();
if (!diskSystem) if (diskSystem == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
// child must indeed be a child of partition // child must indeed be a child of partition
@@ -1064,16 +1043,12 @@ _user_set_partition_parameters(partition_id partitionID, int32* _changeCounter,
return error; return error;
// return change counters // return change counters
if ((error = copy_to_user_value(_changeCounter, partition->ChangeCounter())) error = copy_to_user_value(_changeCounter, partition->ChangeCounter());
!= B_OK if (error == B_OK)
|| (error = copy_to_user_value(_childChangeCounter, error = copy_to_user_value(_childChangeCounter, child->ChangeCounter());
child->ChangeCounter())) != B_OK) {
return error; return error;
} }
return B_OK;
}
status_t status_t
_user_set_partition_content_parameters(partition_id partitionID, _user_set_partition_content_parameters(partition_id partitionID,
@@ -1085,15 +1060,15 @@ _user_set_partition_content_parameters(partition_id partitionID,
status_t error status_t error
= parameters.Init(_parameters, B_DISK_DEVICE_MAX_PARAMETER_SIZE); = parameters.Init(_parameters, B_DISK_DEVICE_MAX_PARAMETER_SIZE);
if (error != B_OK || (error = copy_from_user_value(changeCounter, if (error == B_OK)
_changeCounter)) != B_OK) { error = copy_from_user_value(changeCounter, _changeCounter);
if (error != B_OK)
return error; return error;
}
// get the partition // get the partition
KDiskDeviceManager* manager = KDiskDeviceManager::Default(); KDiskDeviceManager* manager = KDiskDeviceManager::Default();
KPartition* partition = manager->WriteLockPartition(partitionID); KPartition* partition = manager->WriteLockPartition(partitionID);
if (!partition) if (partition == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar1(partition, true); PartitionRegistrar registrar1(partition, true);
@@ -1106,7 +1081,7 @@ _user_set_partition_content_parameters(partition_id partitionID,
// the partition must be initialized // the partition must be initialized
KDiskSystem* diskSystem = partition->DiskSystem(); KDiskSystem* diskSystem = partition->DiskSystem();
if (!diskSystem) if (diskSystem == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
// mark the partition busy and unlock // mark the partition busy and unlock
@@ -1126,12 +1101,7 @@ _user_set_partition_content_parameters(partition_id partitionID,
return error; return error;
// return change counter // return change counter
if ((error = copy_to_user_value(_changeCounter, partition->ChangeCounter())) return copy_to_user_value(_changeCounter, partition->ChangeCounter());
!= B_OK) {
return error;
}
return B_OK;
} }
@@ -1145,21 +1115,21 @@ _user_initialize_partition(partition_id partitionID, int32* _changeCounter,
UserStringParameter<true> parameters; UserStringParameter<true> parameters;
int32 changeCounter; int32 changeCounter;
status_t error; status_t error
if ((error = diskSystemName.Init(_diskSystemName, = diskSystemName.Init(_diskSystemName, B_DISK_SYSTEM_NAME_LENGTH);
B_DISK_SYSTEM_NAME_LENGTH)) != B_OK if (error == B_OK)
|| (error = name.Init(_name, B_DISK_DEVICE_NAME_LENGTH)) != B_OK error = name.Init(_name, B_DISK_DEVICE_NAME_LENGTH);
|| (error = parameters.Init(_parameters, if (error == B_OK)
B_DISK_DEVICE_MAX_PARAMETER_SIZE)) != B_OK error = parameters.Init(_parameters, B_DISK_DEVICE_MAX_PARAMETER_SIZE);
|| (error = copy_from_user_value(changeCounter, _changeCounter)) if (error == B_OK)
!= B_OK) { error = copy_from_user_value(changeCounter, _changeCounter);
if (error != B_OK)
return error; return error;
}
// get the partition // get the partition
KDiskDeviceManager* manager = KDiskDeviceManager::Default(); KDiskDeviceManager* manager = KDiskDeviceManager::Default();
KPartition* partition = manager->WriteLockPartition(partitionID); KPartition* partition = manager->WriteLockPartition(partitionID);
if (!partition) if (partition == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar1(partition, true); PartitionRegistrar registrar1(partition, true);
@@ -1171,13 +1141,13 @@ _user_initialize_partition(partition_id partitionID, int32* _changeCounter,
return B_BAD_VALUE; return B_BAD_VALUE;
// the partition must be uninitialized // the partition must be uninitialized
if (partition->DiskSystem()) if (partition->DiskSystem() != NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
// load the new disk system // load the new disk system
KDiskSystem *diskSystem = manager->LoadDiskSystem(diskSystemName.value, KDiskSystem *diskSystem = manager->LoadDiskSystem(diskSystemName.value,
true); true);
if (!diskSystem) if (diskSystem == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
DiskSystemLoader loader(diskSystem, true); DiskSystemLoader loader(diskSystem, true);
@@ -1209,11 +1179,7 @@ _user_initialize_partition(partition_id partitionID, int32* _changeCounter,
partition->SetDiskSystem(diskSystem); partition->SetDiskSystem(diskSystem);
// return change counter // return change counter
error = copy_to_user_value(_changeCounter, partition->ChangeCounter()); return copy_to_user_value(_changeCounter, partition->ChangeCounter());
if (error != B_OK)
return error;
return B_OK;
} }
@@ -1226,17 +1192,16 @@ _user_uninitialize_partition(partition_id partitionID, int32* _changeCounter,
int32 parentChangeCounter; int32 parentChangeCounter;
bool haveParent = parentID >= 0; bool haveParent = parentID >= 0;
status_t error; status_t error = copy_from_user_value(changeCounter, _changeCounter);
if ((error = copy_from_user_value(changeCounter, _changeCounter)) != B_OK) if (haveParent && error == B_OK)
return error; error = copy_from_user_value(parentChangeCounter, _parentChangeCounter);
if (haveParent && (error = copy_from_user_value(parentChangeCounter, if (error != B_OK)
_parentChangeCounter)) != B_OK)
return error; return error;
// get the partition // get the partition
KDiskDeviceManager* manager = KDiskDeviceManager::Default(); KDiskDeviceManager* manager = KDiskDeviceManager::Default();
KPartition* partition = manager->WriteLockPartition(partitionID); KPartition* partition = manager->WriteLockPartition(partitionID);
if (!partition) if (partition == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar1(partition, true); PartitionRegistrar registrar1(partition, true);
@@ -1257,7 +1222,7 @@ _user_uninitialize_partition(partition_id partitionID, int32* _changeCounter,
return B_BAD_VALUE; return B_BAD_VALUE;
// the partition must be initialized // the partition must be initialized
if (!partition->DiskSystem()) if (partition->DiskSystem() == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
// check busy // check busy
@@ -1288,13 +1253,9 @@ _user_uninitialize_partition(partition_id partitionID, int32* _changeCounter,
// return change counter // return change counter
error = copy_to_user_value(_changeCounter, partition->ChangeCounter()); error = copy_to_user_value(_changeCounter, partition->ChangeCounter());
if (error != B_OK) if (haveParent && error == B_OK)
error = copy_to_user_value(_parentChangeCounter, parent->ChangeCounter());
return error; return error;
if (haveParent && (error = copy_to_user_value(_parentChangeCounter,
parent->ChangeCounter())) != B_OK)
return error;
return B_OK;
} }
@@ -1309,20 +1270,20 @@ _user_create_child_partition(partition_id partitionID, int32* _changeCounter,
UserStringParameter<true> parameters; UserStringParameter<true> parameters;
int32 changeCounter; int32 changeCounter;
status_t error; status_t error = type.Init(_type, B_DISK_DEVICE_TYPE_LENGTH);
if ((error = type.Init(_type, B_DISK_DEVICE_TYPE_LENGTH)) != B_OK if (error == B_OK)
|| (error = name.Init(_name, B_DISK_DEVICE_NAME_LENGTH)) != B_OK error = name.Init(_name, B_DISK_DEVICE_NAME_LENGTH);
|| (error = parameters.Init(_parameters, if (error == B_OK)
B_DISK_DEVICE_MAX_PARAMETER_SIZE)) != B_OK error = parameters.Init(_parameters, B_DISK_DEVICE_MAX_PARAMETER_SIZE);
|| (error = copy_from_user_value(changeCounter, _changeCounter)) if (error == B_OK)
!= B_OK) { error = copy_from_user_value(changeCounter, _changeCounter);
if (error != B_OK)
return error; return error;
}
// get the partition // get the partition
KDiskDeviceManager* manager = KDiskDeviceManager::Default(); KDiskDeviceManager* manager = KDiskDeviceManager::Default();
KPartition* partition = manager->WriteLockPartition(partitionID); KPartition* partition = manager->WriteLockPartition(partitionID);
if (!partition) if (partition == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar1(partition, true); PartitionRegistrar registrar1(partition, true);
@@ -1335,7 +1296,7 @@ _user_create_child_partition(partition_id partitionID, int32* _changeCounter,
// the partition must be initialized // the partition must be initialized
KDiskSystem* diskSystem = partition->DiskSystem(); KDiskSystem* diskSystem = partition->DiskSystem();
if (!diskSystem) if (diskSystem == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
// mark the partition busy and unlock // mark the partition busy and unlock
@@ -1361,15 +1322,12 @@ _user_create_child_partition(partition_id partitionID, int32* _changeCounter,
child->UnmarkBusy(true); child->UnmarkBusy(true);
// return change counter and child ID // return change counter and child ID
if ((error = copy_to_user_value(_changeCounter, partition->ChangeCounter())) error = copy_to_user_value(_changeCounter, partition->ChangeCounter());
!= B_OK if (error == B_OK)
|| (error = copy_to_user_value(childID, child->ID())) != B_OK) { error = copy_to_user_value(childID, child->ID());
return error; return error;
} }
return B_OK;
}
status_t status_t
_user_delete_child_partition(partition_id partitionID, int32* _changeCounter, _user_delete_child_partition(partition_id partitionID, int32* _changeCounter,
@@ -1385,7 +1343,7 @@ _user_delete_child_partition(partition_id partitionID, int32* _changeCounter,
// get the partition // get the partition
KDiskDeviceManager* manager = KDiskDeviceManager::Default(); KDiskDeviceManager* manager = KDiskDeviceManager::Default();
KPartition* partition = manager->WriteLockPartition(partitionID); KPartition* partition = manager->WriteLockPartition(partitionID);
if (!partition) if (partition == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar1(partition, true); PartitionRegistrar registrar1(partition, true);
@@ -1394,7 +1352,7 @@ _user_delete_child_partition(partition_id partitionID, int32* _changeCounter,
// register child // register child
KPartition* child = manager->RegisterPartition(childID); KPartition* child = manager->RegisterPartition(childID);
if (!child) if (child == NULL)
return B_ENTRY_NOT_FOUND; return B_ENTRY_NOT_FOUND;
PartitionRegistrar registrar3(child, true); PartitionRegistrar registrar3(child, true);
@@ -1407,7 +1365,7 @@ _user_delete_child_partition(partition_id partitionID, int32* _changeCounter,
// the partition must be initialized // the partition must be initialized
KDiskSystem* diskSystem = partition->DiskSystem(); KDiskSystem* diskSystem = partition->DiskSystem();
if (!diskSystem) if (diskSystem == NULL)
return B_BAD_VALUE; return B_BAD_VALUE;
// child must indeed be a child of partition // child must indeed be a child of partition