X-Git-Url: https://the.earth.li/gitweb/?a=blobdiff_plain;f=charfuncs.c;h=59c8ca39e55d5cb519a063fe0bff0f3a1a9515b1;hb=HEAD;hp=4b404c9300f16038a91db63ae34524c30939c7c1;hpb=d38e1f468376f8b19b208f2da4d20cb2919875dd;p=onak.git

diff --git a/charfuncs.c b/charfuncs.c
index 4b404c9..59c8ca3 100644
--- a/charfuncs.c
+++ b/charfuncs.c
@@ -13,16 +13,14 @@
  * more details.
  *
  * You should have received a copy of the GNU General Public License along with
- * this program; if not, write to the Free Software Foundation, Inc., 51
- * Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ * this program.  If not, see <https://www.gnu.org/licenses/>.
  */
 
 #include <stdio.h>
 #include <string.h>
-#include <sys/types.h>
-#include <sys/uio.h>
 #include <unistd.h>
 
+#include "build-config.h"
 #include "charfuncs.h"
 
 /*
@@ -31,20 +29,20 @@
  *	@count: The number of characters to get from the buffer.
  *	@c: Where to put the characters retrieved.
  */
-int buffer_fetchchar(void *ctx, size_t count, void *c)
+size_t buffer_fetchchar(void *ctx, size_t count, void *c)
 {
 	struct buffer_ctx *buf = NULL;
 	
 	buf = (struct buffer_ctx *) ctx;
 
 	if (buf->offset + count > buf->size) {
-		return 1;
+		count = buf->size - buf->offset;
 	}
-	
+
 	memcpy(c, &buf->buffer[buf->offset], count);
 	buf->offset += count;
 
-	return 0;
+	return count;
 }
 
 /*
@@ -57,7 +55,7 @@ int buffer_fetchchar(void *ctx, size_t count, void *c)
  *	fill it then we double the size of the current buffer and then add the
  *	rest.
  */
-int buffer_putchar(void *ctx, size_t count, void *c)
+size_t buffer_putchar(void *ctx, size_t count, void *c)
 {
 	struct buffer_ctx *buf = NULL;
 	size_t newsize = 0;
@@ -75,37 +73,41 @@ int buffer_putchar(void *ctx, size_t count, void *c)
 	memcpy(&buf->buffer[buf->offset], c, count);
 	buf->offset += count;
 	
-	return 1;
+	return count;
 }
 
 /*
  * Fetches a char from a file.
  */
-int file_fetchchar(void *fd, size_t count, void *c)
+size_t file_fetchchar(void *fd, size_t count, void *c)
 {
-	return !(read( *(int *) fd, c, count));
+	ssize_t ret = read( *(int *) fd, c, count);
+
+	return (ret > 0) ? ret : 0;
 }
 
 /*
  * Puts a char to a file.
  */
-int file_putchar(void *fd, size_t count, void *c)
+size_t file_putchar(void *fd, size_t count, void *c)
 {
-	return !(write( *(int *) fd, c, count));
+	size_t ret = write( *(int *) fd, c, count);
+
+	return (ret > 0) ? ret : 0;
 }
 
 /*
  * Gets a char from stdin.
  */
-int stdin_getchar(void *ctx, size_t count, void *c)
+size_t stdin_getchar(__unused void *ctx, size_t count, void *c)
 {
-	return (fread(c, 1, count, stdin) != count);
+	return fread(c, 1, count, stdin);
 }
 
 /*
  * Puts a char to stdout.
  */
-int stdout_putchar(void *ctx, size_t count, void *c)
+size_t stdout_putchar(__unused void *ctx, size_t count, void *c)
 {
-	return (fwrite(c, 1, count, stdout) != count);
+	return fwrite(c, 1, count, stdout);
 }