Annotation of researchv10no/sys/inet/udp_device.c, revision 1.1.1.1

1.1       root        1: /*
                      2:  * udp device driver.
                      3:  */
                      4: 
                      5: #include "sys/param.h"
                      6: #include "sys/stream.h"
                      7: #include "sys/conf.h"
                      8: #include "sys/inet/in.h"
                      9: #include "sys/inet/ip.h"
                     10: #include "sys/inet/ip_var.h"
                     11: #include "sys/inet/udp.h"
                     12: #include "sys/inet/udp_var.h"
                     13: #include "sys/inet/udp_user.h"
                     14: 
                     15: int    udpprintfs;
                     16: long   udpdopen();
                     17: int    udpdclose(), udpdput();
                     18: int    udpdosrv();
                     19: int    udpdisrv();
                     20: static struct qinit udpdrinit = { noput, udpdisrv, udpdopen, udpdclose, 8*1024, 64 };
                     21: static struct qinit udpdwinit = { udpdput, udpdosrv, udpdopen, udpdclose,
                     22:                                   UDP_BODY_LIMIT, 64 };
                     23: struct streamtab udpdinfo = { &udpdrinit, &udpdwinit };
                     24: 
                     25: struct cdevsw udpcdev = cstrinit(&udpdinfo);
                     26: 
                     27: extern int udpcnt;     /* for netstat */
                     28: extern struct udp udpconn[];
                     29: 
                     30: in_addr udpbaddst;
                     31: 
                     32: long
                     33: udpdopen(q, dev)
                     34: register struct queue *q;
                     35: dev_t dev;
                     36: {
                     37:        register struct udp *udp;
                     38: 
                     39:        dev = minor(dev);
                     40:        if(dev >= udpcnt)
                     41:                return(0);
                     42:        udp = &udpconn[dev];
                     43:        if(q->ptr)
                     44:                return(0);
                     45:        udp->rq = q;
                     46:        udp->flags = UDP_INIT;
                     47:        udp->head = udp->tail = NULL;
                     48:        q->ptr = (caddr_t)udp;
                     49:        q->flag |= QDELIM;
                     50:        WR(q)->flag |= QBIGB;
                     51:        WR(q)->ptr = (caddr_t)udp;
                     52:        return(1);
                     53: }
                     54: 
                     55: udpdclose(q)
                     56: register struct queue *q;
                     57: {
                     58:        struct udp *udp;
                     59: 
                     60:        udp = (struct udp *)q->ptr;
                     61:        udp->rq = (struct queue *)0;
                     62:        q->ptr = (caddr_t)0;
                     63: }
                     64: 
                     65: udpdput(q, bp)
                     66: register struct queue *q;
                     67: register struct block *bp;
                     68: {
                     69: 
                     70:        switch(bp->type){
                     71:        case M_IOCTL:
                     72:                bp->type = M_IOCNAK;
                     73:                bp->wptr = bp->rptr;
                     74:                qreply(q, bp);
                     75:                return;
                     76:        case M_DATA:
                     77:                putq(q, bp);
                     78:                if (bp->class&S_DELIM)
                     79:                        qenable(q);
                     80:                return;
                     81:        default:
                     82:                freeb(bp);
                     83:                return;
                     84:        }
                     85: }
                     86: 
                     87: udpdosrv(q)
                     88: register struct queue *q;
                     89: {
                     90:        register struct block *bp;
                     91:        register struct udp *udp;
                     92: 
                     93:        udp = (struct udp *)q->ptr;
                     94:        while(bp = getq(q)){
                     95:                if (bp->type != M_DATA)
                     96:                        panic("udpdosrv");
                     97:                bp->next = NULL;
                     98:                if (udp->head == NULL)
                     99:                        udp->head = bp;
                    100:                else
                    101:                        udp->tail->next = bp;
                    102:                udp->tail = bp;
                    103:                if ((bp->class & S_DELIM) == 0)
                    104:                        continue;
                    105:                /*
                    106:                 * have entire datagram
                    107:                 */
                    108:                bp->class &=~ S_DELIM;
                    109:                if (udp->flags & UDP_INIT)
                    110:                        udpduser(udp->head, udp);
                    111:                else {
                    112:                        if ((udp->flags & UDP_CONNECTED) == 0
                    113:                        &&  udpdaddr(udp) == 0)
                    114:                                continue;       /* couldn't get an addr */
                    115:                        udp_output(udp->head, udp);
                    116:                }
                    117:                udp->head = udp->tail = NULL;
                    118:        }
                    119: }
                    120: 
                    121: /*
                    122:  * extract address from an outbound datagram
                    123:  * return 0 if can't find one
                    124:  *     -- if return 0, have freed bad message and cleared udp->head
                    125:  */
                    126: udpdaddr(udp)
                    127: register struct udp *udp;
                    128: {
                    129:        register struct udpaddr *uap;
                    130: 
                    131:        /* get destination */
                    132:        if((udp->head = bp_pullup(udp->head, sizeof(struct udpaddr))) == NULL)
                    133:                return(0);
                    134:        uap = (struct udpaddr *)udp->head->rptr;
                    135:        udp->dst = uap->host;
                    136:        udp->dport = uap->port;
                    137:        udp->head->rptr += sizeof(struct udpaddr);
                    138: 
                    139:        /* host address for datagrams depends on route */
                    140:        if ((udp->flags & UDP_CONNECTED) == 0) {
                    141:                udp->src = ip_hoston(udp->dst);
                    142:                if(udp->src == 0) {
                    143:                        udpbaddst = udp->dst;
                    144:                        bp_free(udp->head);
                    145:                        udp->head = NULL;
                    146:                        return(0);
                    147:                }
                    148:        }
                    149:        return(1);
                    150: }
                    151: 
                    152: udpdisrv(q)
                    153: register struct queue *q;
                    154: {
                    155:        register struct block *bp;
                    156: 
                    157:        while((q->next->flag & QFULL)==0){
                    158:                bp = getq(q);
                    159:                if(bp == 0)
                    160:                        break;
                    161:                (*q->next->qinfo->putp)(q->next, bp);
                    162:        }
                    163: }
                    164: 
                    165: udpdrint(bp, ui)
                    166: register struct block *bp;
                    167: struct udpiphdr *ui;
                    168: {
                    169:        register struct block *bp1;
                    170:        register struct queue *q;
                    171:        register struct udp *udp;
                    172:        register struct udpaddr *uap;
                    173:        register struct udp *udpend;
                    174:        int     ps = spl6();
                    175: 
                    176:        /* find the udp port */
                    177:        udpend = &udpconn[udpcnt];
                    178:        for (udp = udpconn; udp < udpend; ++udp) {
                    179:                if (udp->rq == 0)
                    180:                        continue;
                    181:                if (udp->sport != ui->ui_dport)
                    182:                        continue;
                    183:                if (udp->flags & UDP_LISTEN)
                    184:                        break;
                    185:                if ((udp->flags & UDP_CONNECTED) == 0)
                    186:                        break;
                    187:                if (udp->dport == ui->ui_sport && udp->dst == ui->ui_src)
                    188:                        break;
                    189:        }
                    190: 
                    191:        if (udp >= udpend) {
                    192:                if(udpprintfs)
                    193:                        printf("no udpconn: %d\n", ui->ui_dport);
                    194:                udpstat.udps_badport++;
                    195:                bp_free(bp);
                    196:                splx(ps);
                    197:                return;
                    198:        }
                    199:        if (udp->flags & UDP_LISTEN) {
                    200:                udp->dport = ui->ui_sport;
                    201:                udp->dst = ui->ui_src;
                    202:                if (udp->src == 0) {
                    203:                        /* use host name connecting end called us */
                    204:                        udp->src = ui->ui_dst;
                    205:                }
                    206:                udp->flags |= UDP_CONNECTED;
                    207:                udp->flags &= ~UDP_LISTEN;
                    208:                udpreply(udp);
                    209:        }
                    210:        splx(ps);
                    211: 
                    212:        if((q = udp->rq) == NULL) {
                    213:                printf("udpdrint but no read queue\n");
                    214:                bp_free(bp);
                    215:                return;
                    216:        }
                    217:        if(q->flag & QFULL) {
                    218:                if(udpprintfs)
                    219:                        printf("udpq: %x.%d %x.%d\n", ui->ui_src, ui->ui_sport,
                    220:                                        ui->ui_dst, ui->ui_dport);
                    221:                udpstat.udps_inqfull++;
                    222:                bp_free(bp);
                    223:                return;
                    224:        }
                    225:        if ((udp->flags & UDP_CONNECTED) == 0) {
                    226:                /* if not acting as a connected protocol, pass user source addr */
                    227:                bp1 = allocb(sizeof(struct udpaddr));
                    228:                if (bp1 == NULL) {
                    229:                        printf("udpdrint: out of blocks\n");
                    230:                        bp_free(bp);
                    231:                        return;
                    232:                }
                    233:                bp1->wptr += sizeof(struct udpaddr);
                    234:                uap = (struct udpaddr *)bp1->rptr;
                    235:                uap->host = ui->ui_src;
                    236:                uap->port = ui->ui_sport;
                    237:                putq(q, bp1);
                    238:        }
                    239:        while(bp){
                    240:                bp1 = bp->next;
                    241:                if (bp1==NULL)
                    242:                        bp->class |= S_DELIM;
                    243:                putq(q, bp);
                    244:                bp = bp1;
                    245:        }
                    246:        qenable(q);
                    247: }
                    248: 
                    249: /*
                    250:  * first write on this device:
                    251:  * data is a command from the user
                    252:  */
                    253: udpduser(bp, udp)
                    254: register struct block *bp;
                    255: register struct udp *udp;
                    256: {
                    257:        register struct udpuser *uu;
                    258: 
                    259:        if ((bp = bp_pullup(bp, sizeof(struct udpuser))) == 0) {
                    260:                printf("udpuser short\n");
                    261:                bp_free(bp);
                    262:                return;
                    263:        }
                    264:        uu = (struct udpuser *)bp->rptr;
                    265:        switch(uu->cmd){
                    266:        case UDPC_DATAGRAM:
                    267:                if (udpbind(udp, uu->sport))
                    268:                        goto bad;
                    269:                udp->flags &= ~UDP_INIT;
                    270:                udpreply(udp);
                    271:                break;
                    272:        case UDPC_CONNECT:
                    273:                if (udpbind(udp, uu->sport))
                    274:                        goto bad;
                    275:                udp->dport = uu->dport;
                    276:                udp->dst = uu->dst;
                    277:                udp->flags |= UDP_CONNECTED;
                    278:                udp->flags &= ~UDP_INIT;
                    279:                udpreply(udp);
                    280:                /* pick a local address related to the destination */
                    281:                udp->src = ip_hoston(udp->dst);
                    282:                if(udp->src == 0)
                    283:                        goto bad;
                    284:                break;
                    285:        case UDPC_LISTEN:
                    286:                if (udpbind(udp, uu->sport))
                    287:                        goto bad;
                    288:                udp->flags |= UDP_LISTEN;
                    289:                udp->flags &= ~UDP_INIT;
                    290:                break;
                    291:        default:
                    292: bad:           putctl(udp->rq->next, M_HANGUP);
                    293:                return;
                    294:        }
                    295:        bp_free(bp);
                    296: }
                    297: 
                    298: udpreply(udp)
                    299: register struct udp *udp;
                    300: {
                    301:        register struct queue *q;
                    302:        struct udpreply ur;
                    303: 
                    304:        ur.reply = UDPR_OK;
                    305:        ur.dport = udp->dport;
                    306:        ur.dst = udp->dst;
                    307:        ur.udpdev = udp - udpconn;
                    308: 
                    309:        q = udp->rq;
                    310:        putcpy(q->next, &ur, sizeof(ur));
                    311:        putctld(q->next, M_DATA);
                    312: }
                    313: 
                    314: udpbind(udp, port)
                    315: register struct udp *udp;
                    316: register int port;
                    317: {
                    318:        register struct udp *udp2;
                    319:        static udp_port udpport = 600;
                    320: 
                    321:        udp->sport = 0;
                    322:        if(port){
                    323:                for(udp2 = &udpconn[0]; udp2 < &udpconn[udpcnt]; udp2++){
                    324:                        if(udp2->rq == 0)
                    325:                                continue;
                    326:                        if(udp2->sport == port)
                    327:                                return(1);
                    328:                }
                    329:                udp->sport = port;
                    330:                return(0);
                    331:        }
                    332: 
                    333:        /* pick one for him */
                    334:        if(udpport >= 1024)
                    335:                udpport = 600;
                    336:        port = udpport;
                    337:        while(1){
                    338:                if(udpbind(udp, udpport) == 0){
                    339:                        udpport++;
                    340:                        return(0);
                    341:                }
                    342:                udpport++;
                    343:                if(udpport >= 1024)
                    344:                        udpport = 600;
                    345:                if(udpport == port)     /* tried them all */
                    346:                        break;
                    347:        }
                    348:        return(1);
                    349: }

unix.superglobalmegacorp.com

This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.