* Copyright 2008, Axel Dörfler, axeld@pinc-software.de.
* This file may be used under the terms of the MIT License.
*/
#include <dirent.h>
#include <unistd.h>
#include <stdio.h>
#include <string.h>
#include <Entry.h>
#include <File.h>
#include <fs_attr.h>
#include <TypeConstants.h>
const char* kTempFile = "/tmp/bfs_attribute_iterator_test";
bool gVerbose;
bool
is_marker(const char* name)
{
return strstr(name, ":marker") != NULL;
}
int32
attribute_index(const char* name)
{
int32 number;
if (sscanf(name, "test:%ld", &number) == 1)
return number;
return 0;
}
void
add_attributes(BFile& file, int32 start, int32 count)
{
for (int32 index = start; index < start + count; index++) {
char name[B_ATTR_NAME_LENGTH];
snprintf(name, sizeof(name), "test:%ld", index);
file.WriteAttr(name, B_INT32_TYPE, 0, &index, sizeof(int32));
}
}
void
remove_attributes(BFile& file, int32 start, int32 count)
{
for (int32 index = start; index < start + count; index++) {
char name[B_ATTR_NAME_LENGTH];
snprintf(name, sizeof(name), "test:%ld", index);
file.RemoveAttr(name);
}
}
void
add_marker_attribute(BFile& file, int32 index)
{
char name[B_ATTR_NAME_LENGTH];
snprintf(name, sizeof(name), "test:%ld:marker", index);
file.WriteAttr(name, B_INT32_TYPE, 0, &index, sizeof(int32));
}
void
remove_marker_attribute(BFile& file, int32 index)
{
char name[B_ATTR_NAME_LENGTH];
snprintf(name, sizeof(name), "test:%ld:marker", index);
file.RemoveAttr(name);
}
void
test_remove_attributes(BFile& file, int32 start, int32 count, int32 removeAt,
int32 removeIndex1, int32 removeIndex2)
{
int fd = file.Dup();
if (fd < 0)
return;
int32 index = 0;
bool seen[4096] = {0};
if (gVerbose)
printf("test removeAt %ld\n", removeAt);
DIR* dir = fs_fopen_attr_dir(fd);
while (struct dirent* entry = fs_read_attr_dir(dir)) {
if (gVerbose)
printf(" %ld. %s\n", index, entry->d_name);
if (index == removeAt) {
if (removeIndex1 > 0)
remove_marker_attribute(file, removeIndex1);
if (removeIndex2 > 0)
remove_marker_attribute(file, removeIndex2);
}
index++;
if (is_marker(entry->d_name))
continue;
int32 attributeIndex = attribute_index(entry->d_name);
if (attributeIndex > 0) {
if (seen[attributeIndex]) {
printf("attribute index %ld already listed!\n", attributeIndex);
exit(1);
} else
seen[attributeIndex] = true;
}
}
fs_close_attr_dir(dir);
close(fd);
for (int32 i = start; i < start + count; i++) {
if (!seen[i]) {
printf("attribute index %ld not listed, saw only %ld/%ld!\n", i,
index, count);
exit(1);
}
}
}
void
test_add_attributes(BFile& file, int32 start, int32 count, int32 addAt,
int32 addIndex1, int32 addIndex2)
{
int fd = file.Dup();
if (fd < 0)
return;
int32 index = 0;
bool seen[4096] = {0};
if (gVerbose)
printf("test addAt %ld\n", addAt);
DIR* dir = fs_fopen_attr_dir(fd);
while (struct dirent* entry = fs_read_attr_dir(dir)) {
if (gVerbose)
printf(" %ld. %s\n", index, entry->d_name);
if (index == addAt) {
if (addIndex1 > 0)
add_marker_attribute(file, addIndex1);
if (addIndex2 > 0)
add_marker_attribute(file, addIndex2);
}
index++;
if (is_marker(entry->d_name))
continue;
int32 attributeIndex = attribute_index(entry->d_name);
if (attributeIndex > 0) {
if (seen[attributeIndex]) {
printf("attribute index %ld already listed!\n", attributeIndex);
exit(1);
} else
seen[attributeIndex] = true;
}
}
fs_close_attr_dir(dir);
close(fd);
for (int32 i = start; i < start + count; i++) {
if (!seen[i]) {
printf("attribute index %ld not listed, saw only %ld/%ld!\n", i,
index, count);
exit(1);
}
}
}
int
main(int argc, char** argv)
{
if (argc > 1 && !strcmp(argv[1], "-v"))
gVerbose = true;
BFile file;
status_t status = file.SetTo(kTempFile, B_CREATE_FILE | B_READ_WRITE);
if (status != B_OK) {
fprintf(stderr, "Could not create temporary file: %s\n",
strerror(status));
return 1;
}
puts("--------- Remove Tests ----------");
puts("Test 1...");
for (int32 count = 5; count <= 100; count += 5) {
add_attributes(file, 1, count);
add_marker_attribute(file, count);
add_attributes(file, count + 1, count);
test_remove_attributes(file, 1, count,
count & 1 ? count - 1 : count + 1, count, -1);
remove_attributes(file, 1, count * 2);
}
puts("Test 2...");
for (int32 i = 1; i < 100; i++) {
add_attributes(file, 1, 50);
add_marker_attribute(file, 51);
add_attributes(file, 51, 50);
test_remove_attributes(file, 1, 100, i, 51, -1);
remove_attributes(file, 1, 100);
}
puts("Test 3...");
for (int32 i = 1; i < 100; i++) {
add_attributes(file, 1, 33);
add_marker_attribute(file, 33);
add_attributes(file, 34, 34);
add_marker_attribute(file, 67);
add_attributes(file, 68, 33);
test_remove_attributes(file, 1, 100, i, 33, 67);
remove_attributes(file, 1, 100);
}
puts("--------- Add Tests ----------");
puts("Test 4...");
for (int32 count = 10; count <= 200; count += 10) {
add_attributes(file, 1, count);
int32 half = count / 2;
test_add_attributes(file, 1, count,
half & 1 ? half - 1 : half + 1, half - 2, half + 2);
remove_attributes(file, 1, count);
}
puts("Test 5...");
for (int32 i = 1; i < 100; i++) {
add_attributes(file, 1, 100);
test_add_attributes(file, 1, 100, i, 50, -1);
remove_attributes(file, 1, 100);
}
puts("Test 6...");
for (int32 i = 1; i < 100; i++) {
add_attributes(file, 1, 100);
test_add_attributes(file, 1, 100, 50, i, -1);
remove_attributes(file, 1, 100);
}
puts("Test 7...");
for (int32 i = 1; i < 100; i++) {
add_attributes(file, 1, 100);
test_add_attributes(file, 1, 100, i, 33, 67);
remove_attributes(file, 1, 100);
}
puts("Test 8...");
for (int32 i = 1; i < 100; i++) {
add_attributes(file, 1, 100);
test_add_attributes(file, 1, 100, 50, i - 1, i + 1);
remove_attributes(file, 1, 100);
}
BEntry entry;
if (entry.SetTo(kTempFile) == B_OK)
entry.Remove();
return 0;
}
