Andrew's git - gitweb.git/blob

39d52b88b754338f7ad86cb71c1b7998f110f272
   1/*
   2 * Licensed under a two-clause BSD-style license.
   3 * See LICENSE for details.
   4 */
   5
   6#include "git-compat-util.h"
   7#include "line_buffer.h"
   8#include "strbuf.h"
   9
  10#define COPY_BUFFER_LEN 4096
  11
  12int buffer_init(struct line_buffer *buf, const char *filename)
  13{
  14        buf->infile = filename ? fopen(filename, "r") : stdin;
  15        if (!buf->infile)
  16                return -1;
  17        return 0;
  18}
  19
  20int buffer_fdinit(struct line_buffer *buf, int fd)
  21{
  22        buf->infile = fdopen(fd, "r");
  23        if (!buf->infile)
  24                return -1;
  25        return 0;
  26}
  27
  28int buffer_tmpfile_init(struct line_buffer *buf)
  29{
  30        buf->infile = tmpfile();
  31        if (!buf->infile)
  32                return -1;
  33        return 0;
  34}
  35
  36int buffer_deinit(struct line_buffer *buf)
  37{
  38        int err;
  39        if (buf->infile == stdin)
  40                return ferror(buf->infile);
  41        err = ferror(buf->infile);
  42        err |= fclose(buf->infile);
  43        return err;
  44}
  45
  46FILE *buffer_tmpfile_rewind(struct line_buffer *buf)
  47{
  48        rewind(buf->infile);
  49        return buf->infile;
  50}
  51
  52long buffer_tmpfile_prepare_to_read(struct line_buffer *buf)
  53{
  54        long pos = ftell(buf->infile);
  55        if (pos < 0)
  56                return error("ftell error: %s", strerror(errno));
  57        if (fseek(buf->infile, 0, SEEK_SET))
  58                return error("seek error: %s", strerror(errno));
  59        return pos;
  60}
  61
  62int buffer_ferror(struct line_buffer *buf)
  63{
  64        return ferror(buf->infile);
  65}
  66
  67int buffer_read_char(struct line_buffer *buf)
  68{
  69        return fgetc(buf->infile);
  70}
  71
  72/* Read a line without trailing newline. */
  73char *buffer_read_line(struct line_buffer *buf)
  74{
  75        char *end;
  76        if (!fgets(buf->line_buffer, sizeof(buf->line_buffer), buf->infile))
  77                /* Error or data exhausted. */
  78                return NULL;
  79        end = buf->line_buffer + strlen(buf->line_buffer);
  80        if (end[-1] == '\n')
  81                end[-1] = '\0';
  82        else if (feof(buf->infile))
  83                ; /* No newline at end of file.  That's fine. */
  84        else
  85                /*
  86                 * Line was too long.
  87                 * There is probably a saner way to deal with this,
  88                 * but for now let's return an error.
  89                 */
  90                return NULL;
  91        return buf->line_buffer;
  92}
  93
  94char *buffer_read_string(struct line_buffer *buf, uint32_t len)
  95{
  96        strbuf_reset(&buf->blob_buffer);
  97        strbuf_fread(&buf->blob_buffer, len, buf->infile);
  98        return ferror(buf->infile) ? NULL : buf->blob_buffer.buf;
  99}
 100
 101void buffer_read_binary(struct line_buffer *buf,
 102                                struct strbuf *sb, uint32_t size)
 103{
 104        strbuf_fread(sb, size, buf->infile);
 105}
 106
 107void buffer_copy_bytes(struct line_buffer *buf, off_t len)
 108{
 109        char byte_buffer[COPY_BUFFER_LEN];
 110        uint32_t in;
 111        while (len > 0 && !feof(buf->infile) && !ferror(buf->infile)) {
 112                in = len < COPY_BUFFER_LEN ? len : COPY_BUFFER_LEN;
 113                in = fread(byte_buffer, 1, in, buf->infile);
 114                len -= in;
 115                fwrite(byte_buffer, 1, in, stdout);
 116                if (ferror(stdout)) {
 117                        buffer_skip_bytes(buf, len);
 118                        return;
 119                }
 120        }
 121}
 122
 123off_t buffer_skip_bytes(struct line_buffer *buf, off_t nbytes)
 124{
 125        char byte_buffer[COPY_BUFFER_LEN];
 126        off_t done = 0;
 127        while (done < nbytes && !feof(buf->infile) && !ferror(buf->infile)) {
 128                off_t len = nbytes - done;
 129                size_t in = len < COPY_BUFFER_LEN ? len : COPY_BUFFER_LEN;
 130                done += fread(byte_buffer, 1, in, buf->infile);
 131        }
 132        return done;
 133}
 134
 135void buffer_reset(struct line_buffer *buf)
 136{
 137        strbuf_release(&buf->blob_buffer);
 138}