ref: d9d2925fa024c2eceeea18c9b3afda30af96301e
dir: /sys/src/libventi/fcall.c/
#include <u.h> #include <libc.h> #include <venti.h> Packet* vtfcallpack(VtFcall *f) { uchar buf[4]; Packet *p; p = packetalloc(); buf[0] = f->msgtype; buf[1] = f->tag; packetappend(p, buf, 2); switch(f->msgtype){ default: werrstr("vtfcallpack: unknown packet type %d", f->msgtype); goto Err; case VtRerror: if(vtputstring(p, f->error) < 0) goto Err; break; case VtTping: break; case VtRping: break; case VtThello: if(vtputstring(p, f->version) < 0 || vtputstring(p, f->uid) < 0) goto Err; buf[0] = f->strength; buf[1] = f->ncrypto; packetappend(p, buf, 2); packetappend(p, f->crypto, f->ncrypto); buf[0] = f->ncodec; packetappend(p, buf, 1); packetappend(p, f->codec, f->ncodec); break; case VtRhello: if(vtputstring(p, f->sid) < 0) goto Err; buf[0] = f->rcrypto; buf[1] = f->rcodec; packetappend(p, buf, 2); break; case VtTgoodbye: break; case VtTread: packetappend(p, f->score, VtScoreSize); buf[0] = vttodisktype(f->blocktype); if(~buf[0] == 0) goto Err; buf[1] = 0; buf[2] = f->count >> 8; buf[3] = f->count; packetappend(p, buf, 4); break; case VtRread: packetconcat(p, f->data); break; case VtTwrite: buf[0] = vttodisktype(f->blocktype); if(~buf[0] == 0) goto Err; buf[1] = 0; buf[2] = 0; buf[3] = 0; packetappend(p, buf, 4); packetconcat(p, f->data); break; case VtRwrite: packetappend(p, f->score, VtScoreSize); break; case VtTsync: break; case VtRsync: break; } return p; Err: packetfree(p); return nil; } int vtfcallunpack(VtFcall *f, Packet *p) { uchar buf[4]; memset(f, 0, sizeof *f); if(packetconsume(p, buf, 2) < 0) return -1; f->msgtype = buf[0]; f->tag = buf[1]; switch(f->msgtype){ default: werrstr("vtfcallunpack: unknown bad packet type %d", f->msgtype); vtfcallclear(f); return -1; case VtRerror: if(vtgetstring(p, &f->error) < 0) goto Err; break; case VtTping: break; case VtRping: break; case VtThello: if(vtgetstring(p, &f->version) < 0 || vtgetstring(p, &f->uid) < 0 || packetconsume(p, buf, 2) < 0) goto Err; f->strength = buf[0]; f->ncrypto = buf[1]; if(f->ncrypto){ f->crypto = vtmalloc(f->ncrypto); if(packetconsume(p, buf, f->ncrypto) < 0) goto Err; } if(packetconsume(p, buf, 1) < 0) goto Err; f->ncodec = buf[0]; if(f->ncodec){ f->codec = vtmalloc(f->ncodec); if(packetconsume(p, buf, f->ncodec) < 0) goto Err; } break; case VtRhello: if(vtgetstring(p, &f->sid) < 0 || packetconsume(p, buf, 2) < 0) goto Err; f->rcrypto = buf[0]; f->rcodec = buf[1]; break; case VtTgoodbye: break; case VtTread: if(packetconsume(p, f->score, VtScoreSize) < 0 || packetconsume(p, buf, 4) < 0) goto Err; f->blocktype = vtfromdisktype(buf[0]); if(~f->blocktype == 0) goto Err; f->count = (buf[2] << 8) | buf[3]; break; case VtRread: f->data = packetalloc(); packetconcat(f->data, p); break; case VtTwrite: if(packetconsume(p, buf, 4) < 0) goto Err; f->blocktype = vtfromdisktype(buf[0]); if(~f->blocktype == 0) goto Err; f->data = packetalloc(); packetconcat(f->data, p); break; case VtRwrite: if(packetconsume(p, f->score, VtScoreSize) < 0) goto Err; break; case VtTsync: break; case VtRsync: break; } if(packetsize(p) != 0) goto Err; return 0; Err: werrstr("bad packet"); vtfcallclear(f); return -1; } void vtfcallclear(VtFcall *f) { vtfree(f->error); f->error = nil; vtfree(f->uid); f->uid = nil; vtfree(f->sid); f->sid = nil; vtfree(f->version); f->version = nil; vtfree(f->crypto); f->crypto = nil; vtfree(f->codec); f->codec = nil; vtfree(f->auth); f->auth = nil; packetfree(f->data); f->data = nil; }