From owner-svn-src-head@freebsd.org  Tue Jun 13 13:26:51 2017
Return-Path: <owner-svn-src-head@freebsd.org>
Delivered-To: svn-src-head@mailman.ysv.freebsd.org
Received: from mx1.freebsd.org (mx1.freebsd.org
 [IPv6:2001:1900:2254:206a::19:1])
 by mailman.ysv.freebsd.org (Postfix) with ESMTP id 48FDBBF4734;
 Tue, 13 Jun 2017 13:26:51 +0000 (UTC)
 (envelope-from gahr@FreeBSD.org)
Received: from repo.freebsd.org (repo.freebsd.org
 [IPv6:2610:1c1:1:6068::e6a:0])
 (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits))
 (Client did not present a certificate)
 by mx1.freebsd.org (Postfix) with ESMTPS id 102487BBCA;
 Tue, 13 Jun 2017 13:26:50 +0000 (UTC)
 (envelope-from gahr@FreeBSD.org)
Received: from repo.freebsd.org ([127.0.1.37])
 by repo.freebsd.org (8.15.2/8.15.2) with ESMTP id v5DDQokf097492;
 Tue, 13 Jun 2017 13:26:50 GMT (envelope-from gahr@FreeBSD.org)
Received: (from gahr@localhost)
 by repo.freebsd.org (8.15.2/8.15.2/Submit) id v5DDQorC097491;
 Tue, 13 Jun 2017 13:26:50 GMT (envelope-from gahr@FreeBSD.org)
Message-Id: <201706131326.v5DDQorC097491@repo.freebsd.org>
X-Authentication-Warning: repo.freebsd.org: gahr set sender to
 gahr@FreeBSD.org using -f
From: Pietro Cerutti <gahr@FreeBSD.org>
Date: Tue, 13 Jun 2017 13:26:50 +0000 (UTC)
To: src-committers@freebsd.org, svn-src-all@freebsd.org,
 svn-src-head@freebsd.org
Subject: svn commit: r319898 - head/usr.bin/yes
X-SVN-Group: head
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
X-BeenThere: svn-src-head@freebsd.org
X-Mailman-Version: 2.1.23
Precedence: list
List-Id: SVN commit messages for the src tree for head/-current
 <svn-src-head.freebsd.org>
List-Unsubscribe: <https://lists.freebsd.org/mailman/options/svn-src-head>,
 <mailto:svn-src-head-request@freebsd.org?subject=unsubscribe>
List-Archive: <http://lists.freebsd.org/pipermail/svn-src-head/>
List-Post: <mailto:svn-src-head@freebsd.org>
List-Help: <mailto:svn-src-head-request@freebsd.org?subject=help>
List-Subscribe: <https://lists.freebsd.org/mailman/listinfo/svn-src-head>,
 <mailto:svn-src-head-request@freebsd.org?subject=subscribe>
X-List-Received-Date: Tue, 13 Jun 2017 13:26:51 -0000

Author: gahr (ports committer)
Date: Tue Jun 13 13:26:50 2017
New Revision: 319898
URL: https://svnweb.freebsd.org/changeset/base/319898

Log:
  Handle partial writes
  
  Reported by:	ed
  Reviewed by:	cognet
  Approved by:	cognet

Modified:
  head/usr.bin/yes/yes.c

Modified: head/usr.bin/yes/yes.c
==============================================================================
--- head/usr.bin/yes/yes.c	Tue Jun 13 12:35:01 2017	(r319897)
+++ head/usr.bin/yes/yes.c	Tue Jun 13 13:26:50 2017	(r319898)
@@ -55,6 +55,8 @@ main(int argc, char **argv)
 	char * exp = y;
 	size_t buflen = 0;
 	size_t explen = sizeof(y);
+	size_t more;
+	ssize_t ret;
 
 	if (caph_limit_stdio() < 0 || (cap_enter() < 0 && errno != ENOSYS))
 		err(1, "capsicum");
@@ -77,8 +79,10 @@ main(int argc, char **argv)
 		explen = buflen;
 	}
 
-	while (write(STDOUT_FILENO, exp, explen) > 0)
-		;
+	more = explen;
+	while ((ret = write(STDOUT_FILENO, exp + (explen - more), more)) > 0)
+		if ((more -= ret) == 0)
+			more = explen;
 
 	err(1, "stdout");
 	/*NOTREACHED*/