Check return of send_package and exit on error

1 files changed, 6 insertions, 1 deletions

diff --git a/qarsh.c b/qarsh.c
index ecaf951..f582c92 100644
--- a/qarsh.c
+++ b/qarsh.c
@@ -156,11 +156,16 @@ void
 set_remote_user(char *user, char *group)
 {
 	struct qa_packet *qp;
+	int ret;
 
 	qp = make_qp_setuser(user, group);
 	qp->qp_seq = 1;
-	send_packet(qarsh_fd, qp);
+	ret = send_packet(qarsh_fd, qp);
 	qpfree(qp);
+	if (ret <= 0) {
+		fprintf(stderr, "Failed to send set user message: %s\n", strerror(errno));
+		exit(125);
+	}
 	qp = recv_packet(qarsh_fd);
 	if (!qp) {
 		fprintf(stderr, "Failed to receive response to set user\n");