sjero.net Git - wget/blob - src/rbuf.c

   1 /* Buffering read.
   2    Copyright (C) 1995, 1996, 1997, 1998 Free Software Foundation, Inc.
   3
   4 This program is free software; you can redistribute it and/or modify
   5 it under the terms of the GNU General Public License as published by
   6 the Free Software Foundation; either version 2 of the License, or
   7 (at your option) any later version.
   8
   9 This program is distributed in the hope that it will be useful,
  10 but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 GNU General Public License for more details.
  13
  14 You should have received a copy of the GNU General Public License
  15 along with this program; if not, write to the Free Software
  16 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
  17
  18 /* This is a simple implementation of buffering IO-read functions.  */
  19
  20 #include <config.h>
  21
  22 #include <stdio.h>
  23
  24 #include "wget.h"
  25 #include "rbuf.h"
  26 #include "connect.h"
  27
  28 #ifdef HAVE_SSL
  29 #include <openssl/bio.h>
  30 #include <openssl/crypto.h>
  31 #include <openssl/x509.h>
  32 #include <openssl/ssl.h>
  33 #include <openssl/err.h>
  34 #include <openssl/pem.h>
  35 #endif /* HAVE_SSL */
  36
  37 void
  38 rbuf_initialize (struct rbuf *rbuf, int fd)
  39 {
  40   rbuf->fd = fd;
  41 #ifdef HAVE_SSL
  42 /* pointing ssl to NULL results in an unchanged behaviour */
  43   rbuf->ssl = NULL;
  44 #endif /* HAVE_SSL */
  45   rbuf->buffer_pos = rbuf->buffer;
  46   rbuf->buffer_left = 0;
  47 }
  48
  49 int
  50 rbuf_initialized_p (struct rbuf *rbuf)
  51 {
  52   return rbuf->fd != -1;
  53 }
  54
  55 void
  56 rbuf_uninitialize (struct rbuf *rbuf)
  57 {
  58   rbuf->fd = -1;
  59 }
  60
  61 /* Currently unused -- see RBUF_READCHAR.  */
  62 #if 0
  63 /* Function version of RBUF_READCHAR.  */
  64 int
  65 rbuf_readchar (struct rbuf *rbuf, char *store)
  66 {
  67   return RBUF_READCHAR (rbuf, store);
  68 }
  69 #endif
  70
  71 /* Like rbuf_readchar(), only don't move the buffer position.  */
  72 int
  73 rbuf_peek (struct rbuf *rbuf, char *store)
  74 {
  75   if (!rbuf->buffer_left)
  76     {
  77       int res;
  78       rbuf->buffer_pos = rbuf->buffer;
  79       rbuf->buffer_left = 0;
  80 #ifdef HAVE_SSL
  81                 if (rbuf->ssl != NULL) {
  82                 res = ssl_iread (rbuf->ssl, rbuf->buffer, sizeof (rbuf->buffer));
  83                 } else {
  84 #endif /* HAVE_SSL */
  85                 res = iread (rbuf->fd, rbuf->buffer, sizeof (rbuf->buffer));
  86 #ifdef HAVE_SSL
  87       }
  88 #endif /* HAVE_SSL */
  89       if (res <= 0)
  90         return res;
  91       rbuf->buffer_left = res;
  92     }
  93   *store = *rbuf->buffer_pos;
  94   return 1;
  95 }
  96
  97 /* Flush RBUF's buffer to WHERE.  Flush MAXSIZE bytes at most.
  98    Returns the number of bytes actually copied.  If the buffer is
  99    empty, 0 is returned.  */
 100 int
 101 rbuf_flush (struct rbuf *rbuf, char *where, int maxsize)
 102 {
 103   if (!rbuf->buffer_left)
 104     return 0;
 105   else
 106     {
 107       int howmuch = MINVAL (rbuf->buffer_left, maxsize);
 108
 109       if (where)
 110         memcpy (where, rbuf->buffer_pos, howmuch);
 111       rbuf->buffer_left -= howmuch;
 112       rbuf->buffer_pos += howmuch;
 113       return howmuch;
 114     }
 115 }
 116
 117 /* Discard any cached data in RBUF.  */
 118 void
 119 rbuf_discard (struct rbuf *rbuf)
 120 {
 121   rbuf->buffer_left = 0;
 122   rbuf->buffer_pos = rbuf->buffer;
 123 }