[BACK]Return to ring.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / telnet

Annotation of src/usr.bin/telnet/ring.c, Revision 1.11

1.11    ! jsg         1: /*     $OpenBSD: ring.c,v 1.10 2014/07/20 10:32:23 jsg Exp $   */
1.2       niklas      2: /*     $NetBSD: ring.c,v 1.7 1996/02/28 21:04:07 thorpej Exp $ */
                      3:
1.1       deraadt     4: /*
                      5:  * Copyright (c) 1988, 1993
                      6:  *     The Regents of the University of California.  All rights reserved.
                      7:  *
                      8:  * Redistribution and use in source and binary forms, with or without
                      9:  * modification, are permitted provided that the following conditions
                     10:  * are met:
                     11:  * 1. Redistributions of source code must retain the above copyright
                     12:  *    notice, this list of conditions and the following disclaimer.
                     13:  * 2. Redistributions in binary form must reproduce the above copyright
                     14:  *    notice, this list of conditions and the following disclaimer in the
                     15:  *    documentation and/or other materials provided with the distribution.
1.5       millert    16:  * 3. Neither the name of the University nor the names of its contributors
1.1       deraadt    17:  *    may be used to endorse or promote products derived from this software
                     18:  *    without specific prior written permission.
                     19:  *
                     20:  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
                     21:  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
                     22:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
                     23:  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
                     24:  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
                     25:  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
                     26:  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
                     27:  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
                     28:  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
                     29:  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
                     30:  * SUCH DAMAGE.
                     31:  */
                     32:
1.8       guenther   33: #include <string.h>
                     34: #include "ring.h"
1.1       deraadt    35:
                     36: /*
                     37:  * This defines a structure for a ring buffer.
                     38:  *
                     39:  * The circular buffer has two parts:
                     40:  *(((
                     41:  *     full:   [consume, supply)
                     42:  *     empty:  [supply, consume)
                     43:  *]]]
                     44:  *
                     45:  */
                     46:
                     47: /* Internal macros */
                     48: #define        ring_subtract(d,a,b)    (((a)-(b) >= 0)? \
                     49:                                        (a)-(b): (((a)-(b))+(d)->size))
                     50:
                     51: #define        ring_increment(d,a,c)   (((a)+(c) < (d)->top)? \
                     52:                                        (a)+(c) : (((a)+(c))-(d)->size))
                     53:
                     54: #define        ring_decrement(d,a,c)   (((a)-(c) >= (d)->bottom)? \
                     55:                                        (a)-(c) : (((a)-(c))-(d)->size))
                     56:
                     57:
                     58: /*
                     59:  * The following is a clock, used to determine full, empty, etc.
                     60:  *
                     61:  * There is some trickiness here.  Since the ring buffers are initialized
                     62:  * to ZERO on allocation, we need to make sure, when interpreting the
                     63:  * clock, that when the times are EQUAL, then the buffer is FULL.
                     64:  */
1.8       guenther   65: static unsigned long ring_clock = 0;
1.1       deraadt    66:
                     67:
                     68: #define        ring_empty(d) (((d)->consume == (d)->supply) && \
                     69:                                ((d)->consumetime >= (d)->supplytime))
                     70: #define        ring_full(d) (((d)->supply == (d)->consume) && \
                     71:                                ((d)->supplytime > (d)->consumetime))
                     72:
                     73:
                     74: /* Buffer state transition routines */
                     75:
1.7       guenther   76: void
1.11    ! jsg        77: ring_init(Ring *ring, unsigned char *buffer, int count)
1.1       deraadt    78: {
1.7       guenther   79:     memset(ring, 0, sizeof *ring);
1.1       deraadt    80:
                     81:     ring->size = count;
                     82:
                     83:     ring->supply = ring->consume = ring->bottom = buffer;
                     84:
                     85:     ring->top = ring->bottom+ring->size;
                     86: }
                     87:
                     88: /* Mark routines */
                     89:
                     90: /*
                     91:  * Mark the most recently supplied byte.
                     92:  */
                     93:
1.11    ! jsg        94: void
        !            95: ring_mark(Ring *ring)
1.1       deraadt    96: {
                     97:     ring->mark = ring_decrement(ring, ring->supply, 1);
                     98: }
                     99:
                    100: /*
                    101:  * Is the ring pointing to the mark?
                    102:  */
                    103:
1.11    ! jsg       104: int
        !           105: ring_at_mark(Ring *ring)
1.1       deraadt   106: {
                    107:     if (ring->mark == ring->consume) {
                    108:        return 1;
                    109:     } else {
                    110:        return 0;
                    111:     }
                    112: }
                    113:
                    114: /*
                    115:  * Clear any mark set on the ring.
                    116:  */
                    117:
1.11    ! jsg       118: void
        !           119: ring_clear_mark(Ring *ring)
1.1       deraadt   120: {
1.10      jsg       121:     ring->mark = NULL;
1.1       deraadt   122: }
                    123:
                    124: /*
                    125:  * Add characters from current segment to ring buffer.
                    126:  */
1.11    ! jsg       127: void
        !           128: ring_supplied(Ring *ring, int count)
1.1       deraadt   129: {
                    130:     ring->supply = ring_increment(ring, ring->supply, count);
                    131:     ring->supplytime = ++ring_clock;
                    132: }
                    133:
                    134: /*
                    135:  * We have just consumed "c" bytes.
                    136:  */
1.11    ! jsg       137: void
        !           138: ring_consumed(Ring *ring, int count)
1.1       deraadt   139: {
                    140:     if (count == 0)    /* don't update anything */
                    141:        return;
                    142:
                    143:     if (ring->mark &&
                    144:                (ring_subtract(ring, ring->mark, ring->consume) < count)) {
1.10      jsg       145:        ring->mark = NULL;
1.1       deraadt   146:     }
                    147:     ring->consume = ring_increment(ring, ring->consume, count);
                    148:     ring->consumetime = ++ring_clock;
                    149:     /*
                    150:      * Try to encourage "ring_empty_consecutive()" to be large.
                    151:      */
                    152:     if (ring_empty(ring)) {
                    153:        ring->consume = ring->supply = ring->bottom;
                    154:     }
                    155: }
                    156:
                    157:
                    158: /* Buffer state query routines */
                    159:
                    160:
                    161: /* Number of bytes that may be supplied */
1.11    ! jsg       162: int
        !           163: ring_empty_count(Ring *ring)
1.1       deraadt   164: {
                    165:     if (ring_empty(ring)) {    /* if empty */
                    166:            return ring->size;
                    167:     } else {
                    168:        return ring_subtract(ring, ring->consume, ring->supply);
                    169:     }
                    170: }
                    171:
                    172: /* number of CONSECUTIVE bytes that may be supplied */
1.11    ! jsg       173: int
        !           174: ring_empty_consecutive(Ring *ring)
1.1       deraadt   175: {
                    176:     if ((ring->consume < ring->supply) || ring_empty(ring)) {
                    177:                            /*
                    178:                             * if consume is "below" supply, or empty, then
                    179:                             * return distance to the top
                    180:                             */
                    181:        return ring_subtract(ring, ring->top, ring->supply);
                    182:     } else {
                    183:                                    /*
                    184:                                     * else, return what we may.
                    185:                                     */
                    186:        return ring_subtract(ring, ring->consume, ring->supply);
                    187:     }
                    188: }
                    189:
                    190: /* Return the number of bytes that are available for consuming
                    191:  * (but don't give more than enough to get to cross over set mark)
                    192:  */
                    193:
1.11    ! jsg       194: int
        !           195: ring_full_count(Ring *ring)
1.1       deraadt   196: {
1.10      jsg       197:     if ((ring->mark == NULL) || (ring->mark == ring->consume)) {
1.1       deraadt   198:        if (ring_full(ring)) {
                    199:            return ring->size;  /* nothing consumed, but full */
                    200:        } else {
                    201:            return ring_subtract(ring, ring->supply, ring->consume);
                    202:        }
                    203:     } else {
                    204:        return ring_subtract(ring, ring->mark, ring->consume);
                    205:     }
                    206: }
                    207:
                    208: /*
                    209:  * Return the number of CONSECUTIVE bytes available for consuming.
                    210:  * However, don't return more than enough to cross over set mark.
                    211:  */
1.11    ! jsg       212: int
        !           213: ring_full_consecutive(Ring *ring)
1.1       deraadt   214: {
1.10      jsg       215:     if ((ring->mark == NULL) || (ring->mark == ring->consume)) {
1.1       deraadt   216:        if ((ring->supply < ring->consume) || ring_full(ring)) {
                    217:            return ring_subtract(ring, ring->top, ring->consume);
                    218:        } else {
                    219:            return ring_subtract(ring, ring->supply, ring->consume);
                    220:        }
                    221:     } else {
                    222:        if (ring->mark < ring->consume) {
                    223:            return ring_subtract(ring, ring->top, ring->consume);
                    224:        } else {        /* Else, distance to mark */
                    225:            return ring_subtract(ring, ring->mark, ring->consume);
                    226:        }
                    227:     }
                    228: }
                    229:
                    230: /*
                    231:  * Move data into the "supply" portion of of the ring buffer.
                    232:  */
1.11    ! jsg       233: void
        !           234: ring_supply_data(Ring *ring, unsigned char *buffer, int count)
1.1       deraadt   235: {
                    236:     int i;
                    237:
                    238:     while (count) {
1.9       guenther  239:        i = ring_empty_consecutive(ring);
                    240:        if (i > count)
                    241:                i = count;
                    242:        memcpy(ring->supply, buffer, i);
1.1       deraadt   243:        ring_supplied(ring, i);
                    244:        count -= i;
                    245:        buffer += i;
                    246:     }
                    247: }