X-Git-Url: http://git.tuebingen.mpg.de/?p=adu.git;a=blobdiff_plain;f=select.c;h=7537f4cae3b638bccf6c51c23282ccb68b9b693c;hp=f200953266f1bc2eeed3e224c3b4714e09d93fff;hb=06ed2a8ccb7f05eacb530a34f222a1d900f35bf5;hpb=097617e14184af884c2839dd9b82c26aee5d7316

diff --git a/select.c b/select.c
index f200953..7537f4c 100644
--- a/select.c
+++ b/select.c
@@ -117,6 +117,10 @@ struct user_summary_info {
 	int osl_errno;
 };
 
+struct user_summary_line_info {
+	struct format_info *fi;
+	uint32_t count;
+};
 
 static FILE *output_file;
 
@@ -348,7 +352,7 @@ static int compute_user_summary(struct user_info *ui, __a_unused void *data)
 
 static int print_user_summary_line(struct user_info *ui, void *data)
 {
-	struct format_info *fi = data;
+	struct user_summary_line_info *usli = data;
 	union atom_value values[] = {
 		[usa_pw_name] = {.string_value = ui->pw_name?
 			ui->pw_name : "?"},
@@ -358,11 +362,15 @@ static int print_user_summary_line(struct user_info *ui, void *data)
 		[usa_size] = {.num_value =  (long long unsigned)ui->bytes}
 	};
 	char *buf;
-	int ret;
+	int ret = -E_LOOP_COMPLETE;
 
-	buf = format_items(fi, values);
+	if (!usli->count)
+		return ret;
+
+	buf = format_items(usli->fi, values);
 	ret = output("%s", buf);
 	free(buf);
+	usli->count--;
 	return ret;
 }
 
@@ -402,13 +410,17 @@ static int print_user_summary(struct format_info *fi)
 {
 	int ret;
 	int (*comp)(struct user_info *a, struct user_info *b);
+	struct user_summary_line_info usli = {
+		.fi = fi,
+		.count = select_conf.limit_arg
+	};
 
 	if (!select_conf.no_headers_given) {
 		ret = output("User summary\n");
 		if (ret < 0)
 			return ret;
 	}
-	ret = for_each_admissible_user(compute_user_summary, fi);
+	ret = for_each_admissible_user(compute_user_summary, NULL);
 	if (ret < 0)
 		return ret;
 	switch (select_conf.user_summary_sort_arg) {
@@ -429,7 +441,10 @@ static int print_user_summary(struct format_info *fi)
 		break;
 	}
 	sort_hash_table(comp);
-	return for_each_admissible_user(print_user_summary_line, fi);
+	ret = for_each_admissible_user(print_user_summary_line, &usli);
+	if (ret == -E_LOOP_COMPLETE)
+		ret = 1;
+	return ret;
 }
 
 static int user_list_loop_function(struct osl_row *row, void *data)