Annotation of embedaddon/libpdel/io/boundary_fp.c, revision 1.1.1.1

1.1       misho       1: 
                      2: /*
                      3:  * Copyright (c) 2001-2002 Packet Design, LLC.
                      4:  * All rights reserved.
                      5:  * 
                      6:  * Subject to the following obligations and disclaimer of warranty,
                      7:  * use and redistribution of this software, in source or object code
                      8:  * forms, with or without modifications are expressly permitted by
                      9:  * Packet Design; provided, however, that:
                     10:  * 
                     11:  *    (i)  Any and all reproductions of the source or object code
                     12:  *         must include the copyright notice above and the following
                     13:  *         disclaimer of warranties; and
                     14:  *    (ii) No rights are granted, in any manner or form, to use
                     15:  *         Packet Design trademarks, including the mark "PACKET DESIGN"
                     16:  *         on advertising, endorsements, or otherwise except as such
                     17:  *         appears in the above copyright notice or in the software.
                     18:  * 
                     19:  * THIS SOFTWARE IS BEING PROVIDED BY PACKET DESIGN "AS IS", AND
                     20:  * TO THE MAXIMUM EXTENT PERMITTED BY LAW, PACKET DESIGN MAKES NO
                     21:  * REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED, REGARDING
                     22:  * THIS SOFTWARE, INCLUDING WITHOUT LIMITATION, ANY AND ALL IMPLIED
                     23:  * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE,
                     24:  * OR NON-INFRINGEMENT.  PACKET DESIGN DOES NOT WARRANT, GUARANTEE,
                     25:  * OR MAKE ANY REPRESENTATIONS REGARDING THE USE OF, OR THE RESULTS
                     26:  * OF THE USE OF THIS SOFTWARE IN TERMS OF ITS CORRECTNESS, ACCURACY,
                     27:  * RELIABILITY OR OTHERWISE.  IN NO EVENT SHALL PACKET DESIGN BE
                     28:  * LIABLE FOR ANY DAMAGES RESULTING FROM OR ARISING OUT OF ANY USE
                     29:  * OF THIS SOFTWARE, INCLUDING WITHOUT LIMITATION, ANY DIRECT,
                     30:  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, PUNITIVE, OR CONSEQUENTIAL
                     31:  * DAMAGES, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES, LOSS OF
                     32:  * USE, DATA OR PROFITS, HOWEVER CAUSED AND UNDER ANY THEORY OF
                     33:  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
                     34:  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
                     35:  * THE USE OF THIS SOFTWARE, EVEN IF PACKET DESIGN IS ADVISED OF
                     36:  * THE POSSIBILITY OF SUCH DAMAGE.
                     37:  *
                     38:  * Author: Archie Cobbs <archie@freebsd.org>
                     39:  */
                     40: 
                     41: #include <sys/types.h>
                     42: #include <sys/param.h>
                     43: 
                     44: #include <stdio.h>
                     45: #include <stdlib.h>
                     46: #include <stdarg.h>
                     47: #include <string.h>
                     48: #include <assert.h>
                     49: #include <errno.h>
                     50: 
                     51: #include "structs/structs.h"
                     52: #include "structs/type/array.h"
                     53: 
                     54: #include "util/typed_mem.h"
                     55: #include "io/boundary_fp.h"
                     56: 
                     57: #define MEM_TYPE               "boundary_fp"
                     58: 
                     59: #define MAX_BOUNDARY           255             /* must be less than 256 */
                     60: 
                     61: /* Internal state per stream */
                     62: struct boundary_fp {
                     63:        FILE    *fp;
                     64:        char    bdry[MAX_BOUNDARY];             /* boundary string */
                     65:        u_char  fail[MAX_BOUNDARY];             /* failure function */
                     66:        u_char  buf[MAX_BOUNDARY];              /* input buffer data */
                     67:        int     closeit;                        /* close underlying stream */
                     68:        int     blen;                           /* boundary string length */
                     69:        int     nbuf;                           /* number chars in buffer */
                     70:        int     nmatch;                         /* length of the longest suffix
                     71:                                                   of buffer data that matches
                     72:                                                   a prefix of the boundary */
                     73: };
                     74: 
                     75: /*
                     76:  * Internal functions
                     77:  */
                     78: static int     boundary_read(void *cookie, char *buf, int len);
                     79: static int     boundary_close(void *cookie);
                     80: 
                     81: /*
                     82:  * Create a FILE * that reads from another FILE *, stopping
                     83:  * just after it reads in the supplied boundary string.
                     84:  */
                     85: FILE *
                     86: boundary_fopen(FILE *fp, const char *boundary, int closeit)
                     87: {
                     88:        struct boundary_fp *m;
                     89:        int blen;
                     90:        int i, j;
                     91: 
                     92:        /* Sanity check */
                     93:        if ((blen = strlen(boundary)) > MAX_BOUNDARY) {
                     94:                errno = EINVAL;
                     95:                return (NULL);
                     96:        }
                     97: 
                     98:        /* Create state structure */
                     99:        if ((m = MALLOC(MEM_TYPE, sizeof(*m))) == NULL)
                    100:                return (NULL);
                    101:        memset(m, 0, sizeof(*m));
                    102:        m->fp = fp;
                    103:        m->blen = blen;
                    104:        m->closeit = closeit;
                    105:        memcpy(m->bdry, boundary, m->blen);
                    106: 
                    107:        /* Generate failure function (linear algorithms exist; this isn't) */
                    108:        for (i = 2; i < m->blen; i++) {
                    109:                for (j = 1; j < i && strncmp(m->bdry, m->bdry + j, i - j); j++);
                    110:                m->fail[i] = i - j;
                    111:        }
                    112: 
                    113:        /* Create new stream using methods below */
                    114:        if ((fp = funopen(m, boundary_read,
                    115:            NULL, NULL, boundary_close)) == NULL) {
                    116:                FREE(MEM_TYPE, m);
                    117:                return (NULL);
                    118:        }
                    119: 
                    120:        /* Done */
                    121:        return (fp);
                    122: }
                    123: 
                    124: /*
                    125:  * Read method.
                    126:  */
                    127: static int
                    128: boundary_read(void *cookie, char *buf, int len)
                    129: {
                    130:        struct boundary_fp *const m = cookie;
                    131:        int nr;
                    132:        int i;
                    133: 
                    134:        /* Sanity */
                    135:        assert(m->nmatch <= m->nbuf);
                    136: 
                    137:        /* Have we got a complete boundary sitting in the buffer? */
                    138:        if (m->nmatch == m->blen)
                    139:                return (0);
                    140: 
                    141:        /* Return available data, if any (and shift buffer) */
                    142:        if (m->nbuf > m->nmatch) {
                    143:                len = MIN(len, m->nbuf - m->nmatch);
                    144:                memcpy(buf, m->buf, len);
                    145:                m->nbuf -= len;
                    146:                memmove(m->buf, m->buf + len, m->nbuf);
                    147:                return (len);
                    148:        }
                    149: 
                    150:        /*
                    151:         * Try to fill up the buffer from underlying stream. It is not
                    152:         * possible to read past the end of a boundary at this point.
                    153:         */
                    154:        nr = fread(m->buf + m->nbuf, 1, m->blen - m->nbuf, m->fp);
                    155: 
                    156:        /* Check for EOF/error on underlying stream */
                    157:        if (nr == 0) {
                    158:                if (!ferror(m->fp))
                    159:                        errno = EFTYPE;         /* no final boundary seen */
                    160:                return (-1);
                    161:        }
                    162: 
                    163:        /* Recompute match counter using failure links and new data */
                    164:        for (i = m->nbuf; i < m->nbuf + nr; i++) {
                    165:                while (m->nmatch > 0 && m->buf[i] != m->bdry[m->nmatch])
                    166:                        m->nmatch = m->fail[m->nmatch];
                    167:                if (m->buf[i] == m->bdry[m->nmatch])
                    168:                        m->nmatch++;
                    169:        }
                    170:        m->nbuf += nr;
                    171: 
                    172:        /* Try again */
                    173:        return (boundary_read(cookie, buf, len));
                    174: }
                    175: 
                    176: /*
                    177:  * Close method.
                    178:  */
                    179: static int
                    180: boundary_close(void *cookie)
                    181: {
                    182:        struct boundary_fp *const m = cookie;
                    183: 
                    184:        if (m->closeit)
                    185:                fclose(m->fp);
                    186:        FREE(MEM_TYPE, m);
                    187:        return (0);
                    188: }
                    189: 

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>