comparison: src/server/netbuf.c
src/server/netbuf.c
- changeset 14
- b8bf95b39952
- parent 13
- 1fdbf4170ef4
- child 15
- cff9c4101dd7
equal
deleted
inserted
replaced
| 13:1fdbf4170ef4 | 14:b8bf95b39952 |
|---|---|
| 1 /* | |
| 2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. | |
| 3 * | |
| 4 * Copyright 2008 Sun Microsystems, Inc. All rights reserved. | |
| 5 * | |
| 6 * THE BSD LICENSE | |
| 7 * | |
| 8 * Redistribution and use in source and binary forms, with or without | |
| 9 * modification, are permitted provided that the following conditions are met: | |
| 10 * | |
| 11 * Redistributions of source code must retain the above copyright notice, this | |
| 12 * list of conditions and the following disclaimer. | |
| 13 * Redistributions in binary form must reproduce the above copyright notice, | |
| 14 * this list of conditions and the following disclaimer in the documentation | |
| 15 * and/or other materials provided with the distribution. | |
| 16 * | |
| 17 * Neither the name of the nor the names of its contributors may be | |
| 18 * used to endorse or promote products derived from this software without | |
| 19 * specific prior written permission. | |
| 20 * | |
| 21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS | |
| 22 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT | |
| 23 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR | |
| 24 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER | |
| 25 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, | |
| 26 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, | |
| 27 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; | |
| 28 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, | |
| 29 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR | |
| 30 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF | |
| 31 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
| 32 */ | |
| 33 | |
| 34 /* | |
| 35 * netbuf.c: Handles buffered I/O on network sockets | |
| 36 * | |
| 37 * Rob McCool | |
| 38 */ | |
| 39 | |
| 40 #include <stdio.h> | |
| 41 #include <stdlib.h> | |
| 42 #include <unistd.h> | |
| 43 | |
| 44 #include "nspr.h" | |
| 45 #include "nsapi.h" | |
| 46 #include "io.h" | |
| 47 | |
| 48 //include "buffer.h" | |
| 49 | |
| 50 | |
| 51 #define NETBUF_RDTIMEOUT 120 | |
| 52 | |
| 53 | |
| 54 /* ----------------------------- netbuf_open ------------------------------ */ | |
| 55 | |
| 56 | |
| 57 NSAPI_PUBLIC netbuf *netbuf_open(SYS_NETFD sd, int sz) { | |
| 58 netbuf *buf = (netbuf *) malloc(sizeof(netbuf)); | |
| 59 | |
| 60 buf->rdtimeout = NETBUF_RDTIMEOUT; | |
| 61 buf->pos = sz; | |
| 62 buf->cursize = sz; /* force buffer_next on first getc */ | |
| 63 buf->maxsize = sz; | |
| 64 buf->sd = sd; | |
| 65 | |
| 66 buf->inbuf = NULL; | |
| 67 buf->errmsg = NULL; | |
| 68 | |
| 69 return buf; | |
| 70 } | |
| 71 | |
| 72 | |
| 73 /* ----------------------------- netbuf_close ----------------------------- */ | |
| 74 | |
| 75 | |
| 76 NSAPI_PUBLIC void netbuf_close(netbuf *buf) { | |
| 77 if(buf->inbuf) | |
| 78 free(buf->inbuf); | |
| 79 free(buf); | |
| 80 } | |
| 81 | |
| 82 | |
| 83 /* ----------------------------- netbuf_replace --------------------------- */ | |
| 84 | |
| 85 | |
| 86 NSAPI_PUBLIC unsigned char * netbuf_replace(netbuf *buf, | |
| 87 unsigned char *inbuf, int pos, int cursize, int maxsize) | |
| 88 { | |
| 89 unsigned char *oldbuf = buf->inbuf; | |
| 90 | |
| 91 buf->inbuf = inbuf; | |
| 92 buf->pos = pos; | |
| 93 buf->cursize = cursize; | |
| 94 buf->maxsize = maxsize; | |
| 95 | |
| 96 return oldbuf; | |
| 97 } | |
| 98 | |
| 99 | |
| 100 /* ----------------------------- netbuf_next ------------------------------ */ | |
| 101 | |
| 102 | |
| 103 NSAPI_PUBLIC int netbuf_next(netbuf *buf, int advance) { | |
| 104 int n; | |
| 105 | |
| 106 if(!buf->inbuf) | |
| 107 buf->inbuf = (unsigned char *) malloc(buf->maxsize); | |
| 108 | |
| 109 while(1) { | |
| 110 switch(n = net_read(buf->sd, (char *)(buf->inbuf), buf->maxsize)) { | |
| 111 //case IO_EOF: | |
| 112 case 0: | |
| 113 return IO_EOF; | |
| 114 case -1: | |
| 115 return IO_ERROR; | |
| 116 //case IO_ERROR: | |
| 117 //buf->errmsg = system_errmsg(); | |
| 118 // return IO_ERROR; | |
| 119 default: | |
| 120 buf->pos = advance; | |
| 121 buf->cursize = n; | |
| 122 return buf->inbuf[0]; | |
| 123 } | |
| 124 } | |
| 125 } | |
| 126 | |
| 127 NSAPI_PUBLIC int netbuf_getbytes(netbuf *buf, char *buffer, int size) | |
| 128 { | |
| 129 int bytes; | |
| 130 | |
| 131 if (!buf->inbuf) { | |
| 132 buf->inbuf = (unsigned char *) malloc(buf->maxsize); | |
| 133 } else { | |
| 134 if (buf->pos < buf->cursize) { | |
| 135 int bytes_in_buffer = buf->cursize - buf->pos; | |
| 136 | |
| 137 if (bytes_in_buffer > size) | |
| 138 bytes_in_buffer = size; | |
| 139 | |
| 140 memcpy(buffer, &(buf->inbuf[buf->pos]), bytes_in_buffer); | |
| 141 | |
| 142 buf->pos += bytes_in_buffer; | |
| 143 return bytes_in_buffer; | |
| 144 } | |
| 145 } | |
| 146 | |
| 147 /* The netbuf is empty. Read data directly into the caller's buffer */ | |
| 148 bytes = net_read(buf->sd, buffer, size); | |
| 149 if (bytes == 0) | |
| 150 return NETBUF_EOF; | |
| 151 if (bytes < 0) { | |
| 152 //buf->errmsg = system_errmsg(); | |
| 153 return NETBUF_ERROR; | |
| 154 } | |
| 155 return bytes; | |
| 156 } | |
| 157 | |
| 158 | |
| 159 /* ----------------------------- netbuf_grab ------------------------------ */ | |
| 160 | |
| 161 | |
| 162 NSAPI_PUBLIC int netbuf_grab(netbuf *buf, int sz) { | |
| 163 int n; | |
| 164 | |
| 165 if(!buf->inbuf) { | |
| 166 buf->inbuf = (unsigned char *) malloc(sz); | |
| 167 buf->maxsize = sz; | |
| 168 } | |
| 169 else if(sz > buf->maxsize) { | |
| 170 buf->inbuf = (unsigned char *) realloc(buf->inbuf, sz); | |
| 171 buf->maxsize = sz; | |
| 172 } | |
| 173 | |
| 174 PR_ASSERT(buf->pos == buf->cursize); | |
| 175 buf->pos = 0; | |
| 176 buf->cursize = 0; | |
| 177 | |
| 178 while(1) { | |
| 179 switch(n = net_read(buf->sd, (char *)(buf->inbuf), sz)) { | |
| 180 case 0: | |
| 181 return IO_EOF; | |
| 182 case -1: { | |
| 183 //buf->errmsg = system_errmsg(); | |
| 184 return IO_ERROR; | |
| 185 } | |
| 186 default: | |
| 187 buf->cursize = n; | |
| 188 return n; | |
| 189 } | |
| 190 } | |
| 191 } |
