Add an explicit link count to the fstree nodes

Gets initialized to 2 for directories, 1 for all other types. The count
of the parent node is automatically incremented.

Signed-off-by: David Oberhollenzer <david.oberhollenzer@sigma-star.at>

6 files changed, 41 insertions, 0 deletions

diff --git a/tests/add_by_path.c b/tests/add_by_path.c
index af8dd88..6abf05f 100644
--- a/tests/add_by_path.c
+++ b/tests/add_by_path.c
@@ -29,6 +29,8 @@ int main(void)
 	sb.st_uid = 1000;
 	sb.st_gid = 100;
 
+	assert(fs.root->link_count == 2);
+
 	a = fstree_add_generic(&fs, "dir", &sb, NULL);
 	assert(a != NULL);
 	assert(strcmp(a->name, "dir") == 0);
@@ -36,8 +38,10 @@ int main(void)
 	assert(a->uid == sb.st_uid);
 	assert(a->gid == sb.st_gid);
 	assert(a->parent == fs.root);
+	assert(a->link_count == 2);
 	assert(a->next == NULL);
 	assert(fs.root->data.dir.children == a);
+	assert(fs.root->link_count == 3);
 	assert(!a->data.dir.created_implicitly);
 
 	memset(&sb, 0, sizeof(sb));
@@ -52,8 +56,10 @@ int main(void)
 	assert(b->uid == sb.st_uid);
 	assert(b->gid == sb.st_gid);
 	assert(b->parent == fs.root);
+	assert(b->link_count == 1);
 	assert(b->data.devno == sb.st_rdev);
 	assert(b->next == a);
+	assert(fs.root->link_count == 4);
 	assert(fs.root->data.dir.children == b);
 
 	assert(fstree_add_generic(&fs, "blkdev/foo", &sb, NULL) == NULL);
@@ -74,21 +80,29 @@ int main(void)
 	assert(b->mode == sb.st_mode);
 	assert(b->uid == sb.st_uid);
 	assert(b->gid == sb.st_gid);
+	assert(b->link_count == 1);
 	assert(b->parent == a);
 	assert(b->data.devno == sb.st_rdev);
 	assert(b->next == NULL);
 	assert(a->data.dir.children == b);
 
+	assert(a->link_count == 3);
+	assert(fs.root->link_count == 4);
+
 	b = fstree_add_generic(&fs, "dir/foo/chrdev", &sb, NULL);
 	assert(b != NULL);
 	assert(b->next == NULL);
 	assert(b->mode == sb.st_mode);
 	assert(b->uid == sb.st_uid);
 	assert(b->gid == sb.st_gid);
+	assert(b->link_count == 1);
 	assert(b->parent != a);
 	assert(b->parent->parent == a);
 	assert(b->data.devno == sb.st_rdev);
 	assert(b->next == NULL);
+
+	assert(a->link_count == 4);
+	assert(fs.root->link_count == 4);
 	assert(a->data.dir.children != b);
 
 	b = b->parent;
@@ -96,6 +110,7 @@ int main(void)
 	assert(b->mode == (S_IFDIR | 0755));
 	assert(b->uid == 21);
 	assert(b->gid == 42);
+	assert(b->link_count == 3);
 
 	memset(&sb, 0, sizeof(sb));
 	sb.st_mode = S_IFDIR | 0750;
@@ -109,6 +124,10 @@ int main(void)
 	assert(a->mode == sb.st_mode);
 	assert(a->uid == sb.st_uid);
 	assert(a->gid == sb.st_gid);
+	assert(a->link_count == 3);
+
+	assert(a->parent->link_count == 4);
+	assert(fs.root->link_count == 4);
 
 	assert(fstree_add_generic(&fs, "dir/foo", &sb, NULL) == NULL);
 	assert(errno == EEXIST);
diff --git a/tests/fstree_from_file.c b/tests/fstree_from_file.c
index e9b56be..34215c9 100644
--- a/tests/fstree_from_file.c
+++ b/tests/fstree_from_file.c
@@ -45,9 +45,12 @@ int main(void)
 	fstree_post_process(&fs);
 	n = fs.root->data.dir.children;
 
+	assert(fs.root->link_count == 9);
+
 	assert(n->mode == (S_IFBLK | 0600));
 	assert(n->uid == 8);
 	assert(n->gid == 9);
+	assert(n->link_count == 1);
 	assert(strcmp(n->name, "blkdev") == 0);
 	assert(n->data.devno == makedev(42, 21));
 
@@ -55,6 +58,7 @@ int main(void)
 	assert(n->mode == (S_IFCHR | 0600));
 	assert(n->uid == 6);
 	assert(n->gid == 7);
+	assert(n->link_count == 1);
 	assert(strcmp(n->name, "chardev") == 0);
 	assert(n->data.devno == makedev(13, 37));
 
@@ -62,6 +66,7 @@ int main(void)
 	assert(n->mode == (S_IFDIR | 0755));
 	assert(n->uid == 4);
 	assert(n->gid == 5);
+	assert(n->link_count == 2);
 	assert(strcmp(n->name, "dir") == 0);
 	assert(n->data.dir.children == NULL);
 
@@ -69,6 +74,7 @@ int main(void)
 	assert(n->mode == (S_IFDIR | 0755));
 	assert(n->uid == 0);
 	assert(n->gid == 0);
+	assert(n->link_count == 3);
 	assert(strcmp(n->name, "foo bar") == 0);
 	assert(n->data.dir.children != NULL);
 
@@ -76,6 +82,7 @@ int main(void)
 	assert(n->data.dir.children->mode == (S_IFDIR | 0755));
 	assert(n->data.dir.children->uid == 0);
 	assert(n->data.dir.children->gid == 0);
+	assert(n->data.dir.children->link_count == 2);
 	assert(strcmp(n->data.dir.children->name, " test \"") == 0);
 	assert(n->data.dir.children->data.dir.children == NULL);
 
@@ -83,12 +90,14 @@ int main(void)
 	assert(n->mode == (S_IFIFO | 0644));
 	assert(n->uid == 10);
 	assert(n->gid == 11);
+	assert(n->link_count == 1);
 	assert(strcmp(n->name, "pipe") == 0);
 
 	n = n->next;
 	assert(n->mode == (S_IFLNK | 0777));
 	assert(n->uid == 2);
 	assert(n->gid == 3);
+	assert(n->link_count == 1);
 	assert(strcmp(n->name, "slink") == 0);
 	fprintf(stderr, "'%s'\n", n->data.target);
 	assert(strcmp(n->data.target, "slinktarget") == 0);
@@ -97,6 +106,7 @@ int main(void)
 	assert(n->mode == (S_IFSOCK | 0555));
 	assert(n->uid == 12);
 	assert(n->gid == 13);
+	assert(n->link_count == 1);
 	assert(strcmp(n->name, "sock") == 0);
 	assert(n->next == NULL);
 
diff --git a/tests/mknode_dir.c b/tests/mknode_dir.c
index 3b6aeda..8f2131c 100644
--- a/tests/mknode_dir.c
+++ b/tests/mknode_dir.c
@@ -30,6 +30,7 @@ int main(void)
 	assert(root->uid == sb.st_uid);
 	assert(root->gid == sb.st_gid);
 	assert(root->mode == sb.st_mode);
+	assert(root->link_count == 2);
 	assert((char *)root->name >= (char *)root->payload);
 	assert(root->name >= (char *)root->payload);
 	assert(strcmp(root->name, "rootdir") == 0);
@@ -40,6 +41,8 @@ int main(void)
 	a = fstree_mknode(root, "adir", 4, NULL, &sb);
 	assert(a->parent == root);
 	assert(a->next == NULL);
+	assert(a->link_count == 2);
+	assert(root->link_count == 3);
 	assert(root->data.dir.children == a);
 	assert(root->parent == NULL);
 	assert(root->next == NULL);
@@ -47,7 +50,9 @@ int main(void)
 	b = fstree_mknode(root, "bdir", 4, NULL, &sb);
 	assert(a->parent == root);
 	assert(b->parent == root);
+	assert(b->link_count == 2);
 	assert(root->data.dir.children == b);
+	assert(root->link_count == 4);
 	assert(b->next == a);
 	assert(a->next == NULL);
 	assert(root->parent == NULL);
diff --git a/tests/mknode_reg.c b/tests/mknode_reg.c
index 26e42b7..40966f8 100644
--- a/tests/mknode_reg.c
+++ b/tests/mknode_reg.c
@@ -32,6 +32,7 @@ int main(void)
 	assert(node->gid == sb.st_gid);
 	assert(node->mode == sb.st_mode);
 	assert(node->parent == NULL);
+	assert(node->link_count == 1);
 	assert((char *)node->name >= (char *)node->payload);
 	assert(node->data.file.input_file >= (char *)node->payload);
 	assert(node->data.file.input_file >= node->name + 8);
diff --git a/tests/mknode_simple.c b/tests/mknode_simple.c
index 4b4e4aa..ef2a466 100644
--- a/tests/mknode_simple.c
+++ b/tests/mknode_simple.c
@@ -32,6 +32,7 @@ int main(void)
 	assert(node->uid == sb.st_uid);
 	assert(node->gid == sb.st_gid);
 	assert(node->mode == sb.st_mode);
+	assert(node->link_count == 1);
 	assert(node->parent == NULL);
 	assert(node->data.target == NULL);
 	assert(node->data.devno == 0);
@@ -51,6 +52,7 @@ int main(void)
 	assert(node->uid == sb.st_uid);
 	assert(node->gid == sb.st_gid);
 	assert(node->mode == sb.st_mode);
+	assert(node->link_count == 1);
 	assert(node->parent == NULL);
 	assert(node->data.target == NULL);
 	assert(node->data.devno == 0);
@@ -70,6 +72,7 @@ int main(void)
 	assert(node->uid == sb.st_uid);
 	assert(node->gid == sb.st_gid);
 	assert(node->mode == sb.st_mode);
+	assert(node->link_count == 1);
 	assert(node->data.devno == sb.st_rdev);
 	assert(node->parent == NULL);
 	free(node);
@@ -88,6 +91,7 @@ int main(void)
 	assert(node->uid == sb.st_uid);
 	assert(node->gid == sb.st_gid);
 	assert(node->mode == sb.st_mode);
+	assert(node->link_count == 1);
 	assert(node->data.devno == sb.st_rdev);
 	assert(node->parent == NULL);
 	free(node);
diff --git a/tests/mknode_slink.c b/tests/mknode_slink.c
index a520e70..bcb28f0 100644
--- a/tests/mknode_slink.c
+++ b/tests/mknode_slink.c
@@ -30,6 +30,7 @@ int main(void)
 	assert(node->uid == sb.st_uid);
 	assert(node->gid == sb.st_gid);
 	assert(node->mode == (S_IFLNK | 0777));
+	assert(node->link_count == 1);
 	assert(node->parent == NULL);
 	assert((char *)node->name >= (char *)node->payload);
 	assert(node->data.target >= (char *)node->payload);
@@ -42,6 +43,7 @@ int main(void)
 	assert(node->uid == sb.st_uid);
 	assert(node->gid == sb.st_gid);
 	assert(node->mode == (S_IFLNK | 0777));
+	assert(node->link_count == 1);
 	assert(node->parent == NULL);
 	assert((char *)node->name >= (char *)node->payload);
 	assert(node->data.target >= (char *)node->payload);