lxc-attach: switch to attach_run_wait

src/lxc/tools/lxc_attach.c

@@ -354,9 +354,12 @@ int main(int argc, char *argv[])
 			goto out;
 	}
 
-	if (command.program)
-		ret = c->attach(c, lxc_attach_run_command, &command, &attach_options, &pid);
-	else
+	if (command.program) {
+		ret = c->attach_run_wait(c, &attach_options, command.program,
+					 (const char **)command.argv);
+		if (ret < 0)
+			goto out;
+	} else {
 		ret = c->attach(c, lxc_attach_run_shell, NULL, &attach_options, &pid);
 		if (ret < 0)
 			goto out;
@@ -367,6 +370,7 @@ int main(int argc, char *argv[])
 
 		if (WIFEXITED(ret))
 			wexit = WEXITSTATUS(ret);
+	}
 
 out:
 	lxc_container_put(c);