[PATCH 2/2] Move more common code from wait_task_*() to copy_wait_opts_to_user()

[Vitaly Mayatskikh <v.mayatskih@gmail.com>]

Roland McGrath pointed me to more common checks and put_user() around
invoke of copy_wait_opts_to_user() in wait_task_*() functions.
This common code is moved to copy_wait_opts_to_user().

Signed-off-by: Vitaly Mayatskikh <v.mayatskih@gmail.com>
---
 kernel/exit.c |   45 +++++++++++++++++----------------------------
 1 files changed, 17 insertions(+), 28 deletions(-)

diff --git a/kernel/exit.c b/kernel/exit.c
index 3952385..1d42627 100644
--- a/kernel/exit.c
+++ b/kernel/exit.c
@@ -1124,7 +1124,8 @@ static int eligible_child(struct wait_opts *wo, struct task_struct *p)
 }
 
 static int copy_wait_opts_to_user(struct wait_opts *wo, struct task_struct *p,
-				  pid_t pid, uid_t uid, int why, int status, int signal)
+				  pid_t pid, uid_t uid, int why, int status,
+				  int signal, int stat)
 {
 	struct siginfo __user *infop = wo->wo_info;
 	int retval = wo->wo_rusage
@@ -1143,6 +1144,8 @@ static int copy_wait_opts_to_user(struct wait_opts *wo, struct task_struct *p,
 		if (!retval)
 			retval = put_user(status, &infop->si_status);
 	}
+	if (!retval && wo->wo_stat)
+		retval = put_user(stat, wo->wo_stat);
 	if (!retval)
 		retval = pid;
 	return retval;
@@ -1157,7 +1160,7 @@ static int copy_wait_opts_to_user(struct wait_opts *wo, struct task_struct *p,
 static int wait_task_zombie(struct wait_opts *wo, struct task_struct *p)
 {
 	unsigned long state;
-	int retval, why, status, traced;
+	int retval, why, status, traced, stat;
 	pid_t pid = task_pid_vnr(p);
 	uid_t uid = __task_cred(p)->uid;
 
@@ -1177,7 +1180,7 @@ static int wait_task_zombie(struct wait_opts *wo, struct task_struct *p)
 			status = exit_code & 0x7f;
 		}
 		retval = copy_wait_opts_to_user(wo, p, pid, uid,
-						why, status, SIGCHLD);
+						why, status, SIGCHLD, 0);
 		put_task_struct(p);
 		return retval;
 	}
@@ -1260,23 +1263,18 @@ static int wait_task_zombie(struct wait_opts *wo, struct task_struct *p)
 	 */
 	read_unlock(&tasklist_lock);
 
-	status = (p->signal->flags & SIGNAL_GROUP_EXIT)
+	stat = (p->signal->flags & SIGNAL_GROUP_EXIT)
 		? p->signal->group_exit_code : p->exit_code;
-	if (wo->wo_stat)
-		retval = put_user(status, wo->wo_stat);
 
-	if ((status & 0x7f) == 0) {
+	if ((stat & 0x7f) == 0) {
 		why = CLD_EXITED;
-		status >>= 8;
+		status = stat >> 8;
 	} else {
-		why = (status & 0x80) ? CLD_DUMPED : CLD_KILLED;
-		status &= 0x7f;
+		why = (stat & 0x80) ? CLD_DUMPED : CLD_KILLED;
+		status = stat & 0x7f;
 	}
 
-	retval = copy_wait_opts_to_user(wo, p, pid, uid, why, status, SIGCHLD);
-
-	if (!retval)
-		retval = pid;
+	retval = copy_wait_opts_to_user(wo, p, pid, uid, why, status, SIGCHLD, stat);
 
 	if (traced) {
 		write_lock_irq(&tasklist_lock);
@@ -1366,12 +1364,8 @@ unlock_sig:
 	why = ptrace ? CLD_TRAPPED : CLD_STOPPED;
 	read_unlock(&tasklist_lock);
 
-	retval = copy_wait_opts_to_user(wo, p, pid, uid, why, exit_code, SIGCHLD);
-
-	if (!retval && wo->wo_stat)
-		retval = put_user((exit_code << 8) | 0x7f, wo->wo_stat);
-	if (!retval)
-		retval = pid;
+	retval = copy_wait_opts_to_user(wo, p, pid, uid, why, exit_code, SIGCHLD,
+					(exit_code << 8) | 0x7f);
 	put_task_struct(p);
 
 	BUG_ON(!retval);
@@ -1411,15 +1405,10 @@ static int wait_task_continued(struct wait_opts *wo, struct task_struct *p)
 	get_task_struct(p);
 	read_unlock(&tasklist_lock);
 
-	retval = copy_wait_opts_to_user(wo, p, pid, uid,
-					CLD_CONTINUED, SIGCONT, SIGCHLD);
+	retval = copy_wait_opts_to_user(wo, p, pid, uid, CLD_CONTINUED, SIGCONT,
+					SIGCHLD, 0xffff);
 	put_task_struct(p);
 
-	if (!retval && wo->wo_stat)
-		retval = put_user(0xffff, wo->wo_stat);
-	if (!retval)
-		retval = pid;
-
 	BUG_ON(retval == 0);
 
 	return retval;
@@ -1583,7 +1572,7 @@ end:
 			 * we would set so the user can easily tell the
 			 * difference.
 			 */
-			retval = copy_wait_opts_to_user(wo, 0, 0, 0, 0, 0, 0);
+			retval = copy_wait_opts_to_user(wo, 0, 0, 0, 0, 0, 0, 0);
 		}
 	}
 	return retval;
-- 
1.6.2.2