12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311 |
- /*
- * Copyright (c) 2002 - 2005 NetGroup, Politecnico di Torino (Italy)
- * Copyright (c) 2005 - 2008 CACE Technologies, Davis (California)
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- *
- * 1. Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- * 2. Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in the
- * documentation and/or other materials provided with the distribution.
- * 3. Neither the name of the Politecnico di Torino, CACE Technologies
- * nor the names of its contributors may be used to endorse or promote
- * products derived from this software without specific prior written
- * permission.
- *
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- *
- */
- #ifdef HAVE_CONFIG_H
- #include <config.h>
- #endif
- #include "ftmacros.h"
- #include <string.h> /* for strlen(), ... */
- #include <stdlib.h> /* for malloc(), free(), ... */
- #include <stdarg.h> /* for functions with variable number of arguments */
- #include <errno.h> /* for the errno variable */
- #include "sockutils.h"
- #include "pcap-int.h"
- #include "rpcap-protocol.h"
- #include "pcap-rpcap.h"
- /*
- * This file contains the pcap module for capturing from a remote machine's
- * interfaces using the RPCAP protocol.
- *
- * WARNING: All the RPCAP functions that are allowed to return a buffer
- * containing the error description can return max PCAP_ERRBUF_SIZE characters.
- * However there is no guarantees that the string will be zero-terminated.
- * Best practice is to define the errbuf variable as a char of size
- * 'PCAP_ERRBUF_SIZE+1' and to insert manually a NULL character at the end
- * of the buffer. This will guarantee that no buffer overflows occur even
- * if we use the printf() to show the error on the screen.
- *
- * XXX - actually, null-terminating the error string is part of the
- * contract for the pcap API; if there's any place in the pcap code
- * that doesn't guarantee null-termination, even at the expense of
- * cutting the message short, that's a bug and needs to be fixed.
- */
- #define PCAP_STATS_STANDARD 0 /* Used by pcap_stats_rpcap to see if we want standard or extended statistics */
- #ifdef _WIN32
- #define PCAP_STATS_EX 1 /* Used by pcap_stats_rpcap to see if we want standard or extended statistics */
- #endif
- /*
- * \brief Keeps a list of all the opened connections in the active mode.
- *
- * This structure defines a linked list of items that are needed to keep the info required to
- * manage the active mode.
- * In other words, when a new connection in active mode starts, this structure is updated so that
- * it reflects the list of active mode connections currently opened.
- * This structure is required by findalldevs() and open_remote() to see if they have to open a new
- * control connection toward the host, or they already have a control connection in place.
- */
- struct activehosts
- {
- struct sockaddr_storage host;
- SOCKET sockctrl;
- uint8 protocol_version;
- struct activehosts *next;
- };
- /* Keeps a list of all the opened connections in the active mode. */
- static struct activehosts *activeHosts;
- /*
- * Keeps the main socket identifier when we want to accept a new remote
- * connection (active mode only).
- * See the documentation of pcap_remoteact_accept() and
- * pcap_remoteact_cleanup() for more details.
- */
- static SOCKET sockmain;
- /*
- * Private data for capturing remotely using the rpcap protocol.
- */
- struct pcap_rpcap {
- /*
- * This is '1' if we're the network client; it is needed by several
- * functions (such as pcap_setfilter()) to know whether they have
- * to use the socket or have to open the local adapter.
- */
- int rmt_clientside;
- SOCKET rmt_sockctrl; /* socket ID of the socket used for the control connection */
- SOCKET rmt_sockdata; /* socket ID of the socket used for the data connection */
- int rmt_flags; /* we have to save flags, since they are passed by the pcap_open_live(), but they are used by the pcap_startcapture() */
- int rmt_capstarted; /* 'true' if the capture is already started (needed to knoe if we have to call the pcap_startcapture() */
- char *currentfilter; /* Pointer to a buffer (allocated at run-time) that stores the current filter. Needed when flag PCAP_OPENFLAG_NOCAPTURE_RPCAP is turned on. */
- uint8 protocol_version; /* negotiated protocol version */
- unsigned int TotNetDrops; /* keeps the number of packets that have been dropped by the network */
- /*
- * This keeps the number of packets that have been received by the
- * application.
- *
- * Packets dropped by the kernel buffer are not counted in this
- * variable. It is always equal to (TotAccepted - TotDrops),
- * except for the case of remote capture, in which we have also
- * packets in flight, i.e. that have been transmitted by the remote
- * host, but that have not been received (yet) from the client.
- * In this case, (TotAccepted - TotDrops - TotNetDrops) gives a
- * wrong result, since this number does not corresponds always to
- * the number of packet received by the application. For this reason,
- * in the remote capture we need another variable that takes into
- * account of the number of packets actually received by the
- * application.
- */
- unsigned int TotCapt;
- struct pcap_stat stat;
- /* XXX */
- struct pcap *next; /* list of open pcaps that need stuff cleared on close */
- };
- /****************************************************
- * *
- * Locally defined functions *
- * *
- ****************************************************/
- static struct pcap_stat *rpcap_stats_rpcap(pcap_t *p, struct pcap_stat *ps, int mode);
- static int pcap_pack_bpffilter(pcap_t *fp, char *sendbuf, int *sendbufidx, struct bpf_program *prog);
- static int pcap_createfilter_norpcappkt(pcap_t *fp, struct bpf_program *prog);
- static int pcap_updatefilter_remote(pcap_t *fp, struct bpf_program *prog);
- static void pcap_save_current_filter_rpcap(pcap_t *fp, const char *filter);
- static int pcap_setfilter_rpcap(pcap_t *fp, struct bpf_program *prog);
- static int pcap_setsampling_remote(pcap_t *fp);
- static int pcap_startcapture_remote(pcap_t *fp);
- static int rpcap_sendauth(SOCKET sock, uint8 *ver, struct pcap_rmtauth *auth, char *errbuf);
- static int rpcap_recv_msg_header(SOCKET sock, struct rpcap_header *header, char *errbuf);
- static int rpcap_check_msg_ver(SOCKET sock, uint8 expected_ver, struct rpcap_header *header, char *errbuf);
- static int rpcap_check_msg_type(SOCKET sock, uint8 request_type, struct rpcap_header *header, uint16 *errcode, char *errbuf);
- static int rpcap_process_msg_header(SOCKET sock, uint8 ver, uint8 request_type, struct rpcap_header *header, char *errbuf);
- static int rpcap_recv(SOCKET sock, void *buffer, size_t toread, uint32 *plen, char *errbuf);
- static void rpcap_msg_err(SOCKET sockctrl, uint32 plen, char *remote_errbuf);
- static int rpcap_discard(SOCKET sock, uint32 len, char *errbuf);
- static int rpcap_read_packet_msg(SOCKET sock, pcap_t *p, size_t size);
- /****************************************************
- * *
- * Function bodies *
- * *
- ****************************************************/
- /*
- * This function translates (i.e. de-serializes) a 'rpcap_sockaddr'
- * structure from the network byte order to a 'sockaddr_in" or
- * 'sockaddr_in6' structure in the host byte order.
- *
- * It accepts an 'rpcap_sockaddr' structure as it is received from the
- * network, and checks the address family field against various values
- * to see whether it looks like an IPv4 address, an IPv6 address, or
- * neither of those. It checks for multiple values in order to try
- * to handle older rpcap daemons that sent the native OS's 'sockaddr_in'
- * or 'sockaddr_in6' structures over the wire with some members
- * byte-swapped, and to handle the fact that AF_INET6 has different
- * values on different OSes.
- *
- * For IPv4 addresses, it converts the address family to host byte
- * order from network byte order and puts it into the structure,
- * sets the length if a sockaddr structure has a length, converts the
- * port number to host byte order from network byte order and puts
- * it into the structure, copies over the IPv4 address, and zeroes
- * out the zero padding.
- *
- * For IPv6 addresses, it converts the address family to host byte
- * order from network byte order and puts it into the structure,
- * sets the length if a sockaddr structure has a length, converts the
- * port number and flow information to host byte order from network
- * byte order and puts them into the structure, copies over the IPv6
- * address, and converts the scope ID to host byte order from network
- * byte order and puts it into the structure.
- *
- * The function will allocate the 'sockaddrout' variable according to the
- * address family in use. In case the address does not belong to the
- * AF_INET nor AF_INET6 families, 'sockaddrout' is not allocated and a
- * NULL pointer is returned. This usually happens because that address
- * does not exist on the other host, or is of an address family other
- * than AF_INET or AF_INET6, so the RPCAP daemon sent a 'sockaddr_storage'
- * structure containing all 'zero' values.
- *
- * Older RPCAPDs sent the addresses over the wire in the OS's native
- * structure format. For most OSes, this looks like the over-the-wire
- * format, but might have a different value for AF_INET6 than the value
- * on the machine receiving the reply. For OSes with the newer BSD-style
- * sockaddr structures, this has, instead of a 2-byte address family,
- * a 1-byte structure length followed by a 1-byte address family. The
- * RPCAPD code would put the address family in network byte order before
- * sending it; that would set it to 0 on a little-endian machine, as
- * htons() of any value between 1 and 255 would result in a value > 255,
- * with its lower 8 bits zero, so putting that back into a 1-byte field
- * would set it to 0.
- *
- * Therefore, for older RPCAPDs running on an OS with newer BSD-style
- * sockaddr structures, the family field, if treated as a big-endian
- * (network byte order) 16-bit field, would be:
- *
- * (length << 8) | family if sent by a big-endian machine
- * (length << 8) if sent by a little-endian machine
- *
- * For current RPCAPDs, and for older RPCAPDs running on an OS with
- * older BSD-style sockaddr structures, the family field, if treated
- * as a big-endian 16-bit field, would just contain the family.
- *
- * \param sockaddrin: a 'rpcap_sockaddr' pointer to the variable that has
- * to be de-serialized.
- *
- * \param sockaddrout: a 'sockaddr_storage' pointer to the variable that will contain
- * the de-serialized data. The structure returned can be either a 'sockaddr_in' or 'sockaddr_in6'.
- * This variable will be allocated automatically inside this function.
- *
- * \param errbuf: a pointer to a user-allocated buffer (of size PCAP_ERRBUF_SIZE)
- * that will contain the error message (in case there is one).
- *
- * \return '0' if everything is fine, '-1' if some errors occurred. Basically, the error
- * can be only the fact that the malloc() failed to allocate memory.
- * The error message is returned in the 'errbuf' variable, while the deserialized address
- * is returned into the 'sockaddrout' variable.
- *
- * \warning This function supports only AF_INET and AF_INET6 address families.
- *
- * \warning The sockaddrout (if not NULL) must be deallocated by the user.
- */
- /*
- * Possible IPv4 family values other than the designated over-the-wire value,
- * which is 2 (because everybody uses 2 for AF_INET4).
- */
- #define SOCKADDR_IN_LEN 16 /* length of struct sockaddr_in */
- #define SOCKADDR_IN6_LEN 28 /* length of struct sockaddr_in6 */
- #define NEW_BSD_AF_INET_BE ((SOCKADDR_IN_LEN << 8) | 2)
- #define NEW_BSD_AF_INET_LE (SOCKADDR_IN_LEN << 8)
- /*
- * Possible IPv6 family values other than the designated over-the-wire value,
- * which is 23 (because that's what Windows uses, and most RPCAP servers
- * out there are probably running Windows, as WinPcap includes the server
- * but few if any UN*Xes build and ship it).
- *
- * The new BSD sockaddr structure format was in place before 4.4-Lite, so
- * all the free-software BSDs use it.
- */
- #define NEW_BSD_AF_INET6_BSD_BE ((SOCKADDR_IN6_LEN << 8) | 24) /* NetBSD, OpenBSD, BSD/OS */
- #define NEW_BSD_AF_INET6_FREEBSD_BE ((SOCKADDR_IN6_LEN << 8) | 28) /* FreeBSD, DragonFly BSD */
- #define NEW_BSD_AF_INET6_DARWIN_BE ((SOCKADDR_IN6_LEN << 8) | 30) /* macOS, iOS, anything else Darwin-based */
- #define NEW_BSD_AF_INET6_LE (SOCKADDR_IN6_LEN << 8)
- #define LINUX_AF_INET6 10
- #define HPUX_AF_INET6 22
- #define AIX_AF_INET6 24
- #define SOLARIS_AF_INET6 26
- static int
- rpcap_deseraddr(struct rpcap_sockaddr *sockaddrin, struct sockaddr_storage **sockaddrout, char *errbuf)
- {
- /* Warning: we support only AF_INET and AF_INET6 */
- switch (ntohs(sockaddrin->family))
- {
- case RPCAP_AF_INET:
- case NEW_BSD_AF_INET_BE:
- case NEW_BSD_AF_INET_LE:
- {
- struct rpcap_sockaddr_in *sockaddrin_ipv4;
- struct sockaddr_in *sockaddrout_ipv4;
- (*sockaddrout) = (struct sockaddr_storage *) malloc(sizeof(struct sockaddr_in));
- if ((*sockaddrout) == NULL)
- {
- pcap_fmt_errmsg_for_errno(errbuf, PCAP_ERRBUF_SIZE,
- errno, "malloc() failed");
- return -1;
- }
- sockaddrin_ipv4 = (struct rpcap_sockaddr_in *) sockaddrin;
- sockaddrout_ipv4 = (struct sockaddr_in *) (*sockaddrout);
- sockaddrout_ipv4->sin_family = AF_INET;
- sockaddrout_ipv4->sin_port = ntohs(sockaddrin_ipv4->port);
- memcpy(&sockaddrout_ipv4->sin_addr, &sockaddrin_ipv4->addr, sizeof(sockaddrout_ipv4->sin_addr));
- memset(sockaddrout_ipv4->sin_zero, 0, sizeof(sockaddrout_ipv4->sin_zero));
- break;
- }
- #ifdef AF_INET6
- case RPCAP_AF_INET6:
- case NEW_BSD_AF_INET6_BSD_BE:
- case NEW_BSD_AF_INET6_FREEBSD_BE:
- case NEW_BSD_AF_INET6_DARWIN_BE:
- case NEW_BSD_AF_INET6_LE:
- case LINUX_AF_INET6:
- case HPUX_AF_INET6:
- case AIX_AF_INET6:
- case SOLARIS_AF_INET6:
- {
- struct rpcap_sockaddr_in6 *sockaddrin_ipv6;
- struct sockaddr_in6 *sockaddrout_ipv6;
- (*sockaddrout) = (struct sockaddr_storage *) malloc(sizeof(struct sockaddr_in6));
- if ((*sockaddrout) == NULL)
- {
- pcap_fmt_errmsg_for_errno(errbuf, PCAP_ERRBUF_SIZE,
- errno, "malloc() failed");
- return -1;
- }
- sockaddrin_ipv6 = (struct rpcap_sockaddr_in6 *) sockaddrin;
- sockaddrout_ipv6 = (struct sockaddr_in6 *) (*sockaddrout);
- sockaddrout_ipv6->sin6_family = AF_INET6;
- sockaddrout_ipv6->sin6_port = ntohs(sockaddrin_ipv6->port);
- sockaddrout_ipv6->sin6_flowinfo = ntohl(sockaddrin_ipv6->flowinfo);
- memcpy(&sockaddrout_ipv6->sin6_addr, &sockaddrin_ipv6->addr, sizeof(sockaddrout_ipv6->sin6_addr));
- sockaddrout_ipv6->sin6_scope_id = ntohl(sockaddrin_ipv6->scope_id);
- break;
- }
- #endif
- default:
- /*
- * It is neither AF_INET nor AF_INET6 (or, if the OS doesn't
- * support AF_INET6, it's not AF_INET).
- */
- *sockaddrout = NULL;
- break;
- }
- return 0;
- }
- /*
- * This function reads a packet from the network socket. It does not
- * deliver the packet to a pcap_dispatch()/pcap_loop() callback (hence
- * the "nocb" string into its name).
- *
- * This function is called by pcap_read_rpcap().
- *
- * WARNING: By choice, this function does not make use of semaphores. A smarter
- * implementation should put a semaphore into the data thread, and a signal will
- * be raised as soon as there is data into the socket buffer.
- * However this is complicated and it does not bring any advantages when reading
- * from the network, in which network delays can be much more important than
- * these optimizations. Therefore, we chose the following approach:
- * - the 'timeout' chosen by the user is split in two (half on the server side,
- * with the usual meaning, and half on the client side)
- * - this function checks for packets; if there are no packets, it waits for
- * timeout/2 and then it checks again. If packets are still missing, it returns,
- * otherwise it reads packets.
- */
- static int pcap_read_nocb_remote(pcap_t *p, struct pcap_pkthdr *pkt_header, u_char **pkt_data)
- {
- struct pcap_rpcap *pr = p->priv; /* structure used when doing a remote live capture */
- struct rpcap_header *header; /* general header according to the RPCAP format */
- struct rpcap_pkthdr *net_pkt_header; /* header of the packet, from the message */
- u_char *net_pkt_data; /* packet data from the message */
- uint32 plen;
- int retval; /* generic return value */
- int msglen;
- /* Structures needed for the select() call */
- struct timeval tv; /* maximum time the select() can block waiting for data */
- fd_set rfds; /* set of socket descriptors we have to check */
- /*
- * Define the packet buffer timeout, to be used in the select()
- * 'timeout', in pcap_t, is in milliseconds; we have to convert it into sec and microsec
- */
- tv.tv_sec = p->opt.timeout / 1000;
- tv.tv_usec = (p->opt.timeout - tv.tv_sec * 1000) * 1000;
- /* Watch out sockdata to see if it has input */
- FD_ZERO(&rfds);
- /*
- * 'fp->rmt_sockdata' has always to be set before calling the select(),
- * since it is cleared by the select()
- */
- FD_SET(pr->rmt_sockdata, &rfds);
- retval = select((int) pr->rmt_sockdata + 1, &rfds, NULL, NULL, &tv);
- if (retval == -1)
- {
- #ifndef _WIN32
- if (errno == EINTR)
- {
- /* Interrupted. */
- return 0;
- }
- #endif
- sock_geterror("select(): ", p->errbuf, PCAP_ERRBUF_SIZE);
- return -1;
- }
- /* There is no data waiting, so return '0' */
- if (retval == 0)
- return 0;
- /*
- * We have to define 'header' as a pointer to a larger buffer,
- * because in case of UDP we have to read all the message within a single call
- */
- header = (struct rpcap_header *) p->buffer;
- net_pkt_header = (struct rpcap_pkthdr *) ((char *)p->buffer + sizeof(struct rpcap_header));
- net_pkt_data = (u_char *)p->buffer + sizeof(struct rpcap_header) + sizeof(struct rpcap_pkthdr);
- if (pr->rmt_flags & PCAP_OPENFLAG_DATATX_UDP)
- {
- /* Read the entire message from the network */
- msglen = sock_recv_dgram(pr->rmt_sockdata, p->buffer,
- p->bufsize, p->errbuf, PCAP_ERRBUF_SIZE);
- if (msglen == -1)
- {
- /* Network error. */
- return -1;
- }
- if (msglen == -3)
- {
- /* Interrupted receive. */
- return 0;
- }
- if ((size_t)msglen < sizeof(struct rpcap_header))
- {
- /*
- * Message is shorter than an rpcap header.
- */
- pcap_snprintf(p->errbuf, PCAP_ERRBUF_SIZE,
- "UDP packet message is shorter than an rpcap header");
- return -1;
- }
- plen = ntohl(header->plen);
- if ((size_t)msglen < sizeof(struct rpcap_header) + plen)
- {
- /*
- * Message is shorter than the header claims it
- * is.
- */
- pcap_snprintf(p->errbuf, PCAP_ERRBUF_SIZE,
- "UDP packet message is shorter than its rpcap header claims");
- return -1;
- }
- }
- else
- {
- int status;
- if ((size_t)p->cc < sizeof(struct rpcap_header))
- {
- /*
- * We haven't read any of the packet header yet.
- * The size we should get is the size of the
- * packet header.
- */
- status = rpcap_read_packet_msg(pr->rmt_sockdata, p,
- sizeof(struct rpcap_header));
- if (status == -1)
- {
- /* Network error. */
- return -1;
- }
- if (status == -3)
- {
- /* Interrupted receive. */
- return 0;
- }
- }
- /*
- * We have the header, so we know how long the
- * message payload is. The size we should get
- * is the size of the packet header plus the
- * size of the payload.
- */
- plen = ntohl(header->plen);
- if (plen > p->bufsize - sizeof(struct rpcap_header))
- {
- /*
- * This is bigger than the largest
- * record we'd expect. (We do it by
- * subtracting in order to avoid an
- * overflow.)
- */
- pcap_snprintf(p->errbuf, PCAP_ERRBUF_SIZE,
- "Server sent us a message larger than the largest expected packet message");
- return -1;
- }
- status = rpcap_read_packet_msg(pr->rmt_sockdata, p,
- sizeof(struct rpcap_header) + plen);
- if (status == -1)
- {
- /* Network error. */
- return -1;
- }
- if (status == -3)
- {
- /* Interrupted receive. */
- return 0;
- }
- /*
- * We have the entire message; reset the buffer pointer
- * and count, as the next read should start a new
- * message.
- */
- p->bp = p->buffer;
- p->cc = 0;
- }
- /*
- * We have the entire message.
- */
- header->plen = plen;
- /*
- * Did the server specify the version we negotiated?
- */
- if (rpcap_check_msg_ver(pr->rmt_sockdata, pr->protocol_version,
- header, p->errbuf) == -1)
- {
- return 0; /* Return 'no packets received' */
- }
- /*
- * Is this a RPCAP_MSG_PACKET message?
- */
- if (header->type != RPCAP_MSG_PACKET)
- {
- return 0; /* Return 'no packets received' */
- }
- if (ntohl(net_pkt_header->caplen) > plen)
- {
- pcap_snprintf(p->errbuf, PCAP_ERRBUF_SIZE,
- "Packet's captured data goes past the end of the received packet message.");
- return -1;
- }
- /* Fill in packet header */
- pkt_header->caplen = ntohl(net_pkt_header->caplen);
- pkt_header->len = ntohl(net_pkt_header->len);
- pkt_header->ts.tv_sec = ntohl(net_pkt_header->timestamp_sec);
- pkt_header->ts.tv_usec = ntohl(net_pkt_header->timestamp_usec);
- /* Supply a pointer to the beginning of the packet data */
- *pkt_data = net_pkt_data;
- /*
- * I don't update the counter of the packets dropped by the network since we're using TCP,
- * therefore no packets are dropped. Just update the number of packets received correctly
- */
- pr->TotCapt++;
- if (pr->rmt_flags & PCAP_OPENFLAG_DATATX_UDP)
- {
- unsigned int npkt;
- /* We're using UDP, so we need to update the counter of the packets dropped by the network */
- npkt = ntohl(net_pkt_header->npkt);
- if (pr->TotCapt != npkt)
- {
- pr->TotNetDrops += (npkt - pr->TotCapt);
- pr->TotCapt = npkt;
- }
- }
- /* Packet read successfully */
- return 1;
- }
- /*
- * This function reads a packet from the network socket.
- *
- * This function relies on the pcap_read_nocb_remote to deliver packets. The
- * difference, here, is that as soon as a packet is read, it is delivered
- * to the application by means of a callback function.
- */
- static int pcap_read_rpcap(pcap_t *p, int cnt, pcap_handler callback, u_char *user)
- {
- struct pcap_rpcap *pr = p->priv; /* structure used when doing a remote live capture */
- struct pcap_pkthdr pkt_header;
- u_char *pkt_data;
- int n = 0;
- int ret;
- /*
- * If this is client-side, and we haven't already started
- * the capture, start it now.
- */
- if (pr->rmt_clientside)
- {
- /* We are on an remote capture */
- if (!pr->rmt_capstarted)
- {
- /*
- * The capture isn't started yet, so try to
- * start it.
- */
- if (pcap_startcapture_remote(p))
- return -1;
- }
- }
- while (n < cnt || PACKET_COUNT_IS_UNLIMITED(cnt))
- {
- /*
- * Has "pcap_breakloop()" been called?
- */
- if (p->break_loop) {
- /*
- * Yes - clear the flag that indicates that it
- * has, and return PCAP_ERROR_BREAK to indicate
- * that we were told to break out of the loop.
- */
- p->break_loop = 0;
- return (PCAP_ERROR_BREAK);
- }
- /*
- * Read some packets.
- */
- ret = pcap_read_nocb_remote(p, &pkt_header, &pkt_data);
- if (ret == 1)
- {
- /*
- * We got a packet. Hand it to the callback
- * and count it so we can return the count.
- */
- (*callback)(user, &pkt_header, pkt_data);
- n++;
- }
- else if (ret == -1)
- {
- /* Error. */
- return ret;
- }
- else
- {
- /*
- * No packet; this could mean that we timed
- * out, or that we got interrupted, or that
- * we got a bad packet.
- *
- * Were we told to break out of the loop?
- */
- if (p->break_loop) {
- /*
- * Yes.
- */
- p->break_loop = 0;
- return (PCAP_ERROR_BREAK);
- }
- /* No - return the number of packets we've processed. */
- return n;
- }
- }
- return n;
- }
- /*
- * This function sends a CLOSE command to the capture server.
- *
- * It is called when the user calls pcap_close(). It sends a command
- * to our peer that says 'ok, let's stop capturing'.
- *
- * WARNING: Since we're closing the connection, we do not check for errors.
- */
- static void pcap_cleanup_rpcap(pcap_t *fp)
- {
- struct pcap_rpcap *pr = fp->priv; /* structure used when doing a remote live capture */
- struct rpcap_header header; /* header of the RPCAP packet */
- struct activehosts *temp; /* temp var needed to scan the host list chain, to detect if we're in active mode */
- int active = 0; /* active mode or not? */
- /* detect if we're in active mode */
- temp = activeHosts;
- while (temp)
- {
- if (temp->sockctrl == pr->rmt_sockctrl)
- {
- active = 1;
- break;
- }
- temp = temp->next;
- }
- if (!active)
- {
- rpcap_createhdr(&header, pr->protocol_version,
- RPCAP_MSG_CLOSE, 0, 0);
- /*
- * Send the close request; don't report any errors, as
- * we're closing this pcap_t, and have no place to report
- * the error. No reply is sent to this message.
- */
- (void)sock_send(pr->rmt_sockctrl, (char *)&header,
- sizeof(struct rpcap_header), NULL, 0);
- }
- else
- {
- rpcap_createhdr(&header, pr->protocol_version,
- RPCAP_MSG_ENDCAP_REQ, 0, 0);
- /*
- * Send the end capture request; don't report any errors,
- * as we're closing this pcap_t, and have no place to
- * report the error.
- */
- if (sock_send(pr->rmt_sockctrl, (char *)&header,
- sizeof(struct rpcap_header), NULL, 0) == 0)
- {
- /*
- * Wait for the answer; don't report any errors,
- * as we're closing this pcap_t, and have no
- * place to report the error.
- */
- if (rpcap_process_msg_header(pr->rmt_sockctrl,
- pr->protocol_version, RPCAP_MSG_ENDCAP_REQ,
- &header, NULL) == 0)
- {
- (void)rpcap_discard(pr->rmt_sockctrl,
- header.plen, NULL);
- }
- }
- }
- if (pr->rmt_sockdata)
- {
- sock_close(pr->rmt_sockdata, NULL, 0);
- pr->rmt_sockdata = 0;
- }
- if ((!active) && (pr->rmt_sockctrl))
- sock_close(pr->rmt_sockctrl, NULL, 0);
- pr->rmt_sockctrl = 0;
- if (pr->currentfilter)
- {
- free(pr->currentfilter);
- pr->currentfilter = NULL;
- }
- /* To avoid inconsistencies in the number of sock_init() */
- sock_cleanup();
- }
- /*
- * This function retrieves network statistics from our peer;
- * it provides only the standard statistics.
- */
- static int pcap_stats_rpcap(pcap_t *p, struct pcap_stat *ps)
- {
- struct pcap_stat *retval;
- retval = rpcap_stats_rpcap(p, ps, PCAP_STATS_STANDARD);
- if (retval)
- return 0;
- else
- return -1;
- }
- #ifdef _WIN32
- /*
- * This function retrieves network statistics from our peer;
- * it provides the additional statistics supported by pcap_stats_ex().
- */
- static struct pcap_stat *pcap_stats_ex_rpcap(pcap_t *p, int *pcap_stat_size)
- {
- *pcap_stat_size = sizeof (p->stat);
- /* PCAP_STATS_EX (third param) means 'extended pcap_stats()' */
- return (rpcap_stats_rpcap(p, &(p->stat), PCAP_STATS_EX));
- }
- #endif
- /*
- * This function retrieves network statistics from our peer. It
- * is used by the two previous functions.
- *
- * It can be called in two modes:
- * - PCAP_STATS_STANDARD: if we want just standard statistics (i.e.,
- * for pcap_stats())
- * - PCAP_STATS_EX: if we want extended statistics (i.e., for
- * pcap_stats_ex())
- *
- * This 'mode' parameter is needed because in pcap_stats() the variable that
- * keeps the statistics is allocated by the user. On Windows, this structure
- * has been extended in order to keep new stats. However, if the user has a
- * smaller structure and it passes it to pcap_stats(), this function will
- * try to fill in more data than the size of the structure, so that memory
- * after the structure will be overwritten.
- *
- * So, we need to know it we have to copy just the standard fields, or the
- * extended fields as well.
- *
- * In case we want to copy the extended fields as well, the problem of
- * memory overflow no longer exists because the structure that's filled
- * in is part of the pcap_t, so that it can be guaranteed to be large
- * enough for the additional statistics.
- *
- * \param p: the pcap_t structure related to the current instance.
- *
- * \param ps: a pointer to a 'pcap_stat' structure, needed for compatibility
- * with pcap_stat(), where the structure is allocated by the user. In case
- * of pcap_stats_ex(), this structure and the function return value point
- * to the same variable.
- *
- * \param mode: one of PCAP_STATS_STANDARD or PCAP_STATS_EX.
- *
- * \return The structure that keeps the statistics, or NULL in case of error.
- * The error string is placed in the pcap_t structure.
- */
- static struct pcap_stat *rpcap_stats_rpcap(pcap_t *p, struct pcap_stat *ps, int mode)
- {
- struct pcap_rpcap *pr = p->priv; /* structure used when doing a remote live capture */
- struct rpcap_header header; /* header of the RPCAP packet */
- struct rpcap_stats netstats; /* statistics sent on the network */
- uint32 plen; /* data remaining in the message */
- #ifdef _WIN32
- if (mode != PCAP_STATS_STANDARD && mode != PCAP_STATS_EX)
- #else
- if (mode != PCAP_STATS_STANDARD)
- #endif
- {
- pcap_snprintf(p->errbuf, PCAP_ERRBUF_SIZE,
- "Invalid stats mode %d", mode);
- return NULL;
- }
- /*
- * If the capture has not yet started, we cannot request statistics
- * for the capture from our peer, so we return 0 for all statistics,
- * as nothing's been seen yet.
- */
- if (!pr->rmt_capstarted)
- {
- ps->ps_drop = 0;
- ps->ps_ifdrop = 0;
- ps->ps_recv = 0;
- #ifdef _WIN32
- if (mode == PCAP_STATS_EX)
- {
- ps->ps_capt = 0;
- ps->ps_sent = 0;
- ps->ps_netdrop = 0;
- }
- #endif /* _WIN32 */
- return ps;
- }
- rpcap_createhdr(&header, pr->protocol_version,
- RPCAP_MSG_STATS_REQ, 0, 0);
- /* Send the PCAP_STATS command */
- if (sock_send(pr->rmt_sockctrl, (char *)&header,
- sizeof(struct rpcap_header), p->errbuf, PCAP_ERRBUF_SIZE) < 0)
- return NULL; /* Unrecoverable network error */
- /* Receive and process the reply message header. */
- if (rpcap_process_msg_header(pr->rmt_sockctrl, pr->protocol_version,
- RPCAP_MSG_STATS_REQ, &header, p->errbuf) == -1)
- return NULL; /* Error */
- plen = header.plen;
- /* Read the reply body */
- if (rpcap_recv(pr->rmt_sockctrl, (char *)&netstats,
- sizeof(struct rpcap_stats), &plen, p->errbuf) == -1)
- goto error;
- ps->ps_drop = ntohl(netstats.krnldrop);
- ps->ps_ifdrop = ntohl(netstats.ifdrop);
- ps->ps_recv = ntohl(netstats.ifrecv);
- #ifdef _WIN32
- if (mode == PCAP_STATS_EX)
- {
- ps->ps_capt = pr->TotCapt;
- ps->ps_netdrop = pr->TotNetDrops;
- ps->ps_sent = ntohl(netstats.svrcapt);
- }
- #endif /* _WIN32 */
- /* Discard the rest of the message. */
- if (rpcap_discard(pr->rmt_sockctrl, plen, p->errbuf) == -1)
- goto error_nodiscard;
- return ps;
- error:
- /*
- * Discard the rest of the message.
- * We already reported an error; if this gets an error, just
- * drive on.
- */
- (void)rpcap_discard(pr->rmt_sockctrl, plen, NULL);
- error_nodiscard:
- return NULL;
- }
- /*
- * This function returns the entry in the list of active hosts for this
- * active connection (active mode only), or NULL if there is no
- * active connection or an error occurred. It is just for internal
- * use.
- *
- * \param host: a string that keeps the host name of the host for which we
- * want to get the socket ID for that active connection.
- *
- * \param error: a pointer to an int that is set to 1 if an error occurred
- * and 0 otherwise.
- *
- * \param errbuf: a pointer to a user-allocated buffer (of size
- * PCAP_ERRBUF_SIZE) that will contain the error message (in case
- * there is one).
- *
- * \return the entry for this host in the list of active connections
- * if found, NULL if it's not found or there's an error.
- */
- static struct activehosts *
- rpcap_remoteact_getsock(const char *host, int *error, char *errbuf)
- {
- struct activehosts *temp; /* temp var needed to scan the host list chain */
- struct addrinfo hints, *addrinfo, *ai_next; /* temp var needed to translate between hostname to its address */
- int retval;
- /* retrieve the network address corresponding to 'host' */
- addrinfo = NULL;
- memset(&hints, 0, sizeof(struct addrinfo));
- hints.ai_family = PF_UNSPEC;
- hints.ai_socktype = SOCK_STREAM;
- retval = getaddrinfo(host, "0", &hints, &addrinfo);
- if (retval != 0)
- {
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "getaddrinfo() %s",
- gai_strerror(retval));
- *error = 1;
- return NULL;
- }
- temp = activeHosts;
- while (temp)
- {
- ai_next = addrinfo;
- while (ai_next)
- {
- if (sock_cmpaddr(&temp->host, (struct sockaddr_storage *) ai_next->ai_addr) == 0)
- {
- *error = 0;
- freeaddrinfo(addrinfo);
- return temp;
- }
- ai_next = ai_next->ai_next;
- }
- temp = temp->next;
- }
- if (addrinfo)
- freeaddrinfo(addrinfo);
- /*
- * The host for which you want to get the socket ID does not have an
- * active connection.
- */
- *error = 0;
- return NULL;
- }
- /*
- * This function starts a remote capture.
- *
- * This function is required since the RPCAP protocol decouples the 'open'
- * from the 'start capture' functions.
- * This function takes all the parameters needed (which have been stored
- * into the pcap_t structure) and sends them to the server.
- *
- * \param fp: the pcap_t descriptor of the device currently open.
- *
- * \return '0' if everything is fine, '-1' otherwise. The error message
- * (if one) is returned into the 'errbuf' field of the pcap_t structure.
- */
- static int pcap_startcapture_remote(pcap_t *fp)
- {
- struct pcap_rpcap *pr = fp->priv; /* structure used when doing a remote live capture */
- char sendbuf[RPCAP_NETBUF_SIZE]; /* temporary buffer in which data to be sent is buffered */
- int sendbufidx = 0; /* index which keeps the number of bytes currently buffered */
- char portdata[PCAP_BUF_SIZE]; /* temp variable needed to keep the network port for the data connection */
- uint32 plen;
- int active = 0; /* '1' if we're in active mode */
- struct activehosts *temp; /* temp var needed to scan the host list chain, to detect if we're in active mode */
- char host[INET6_ADDRSTRLEN + 1]; /* numeric name of the other host */
- /* socket-related variables*/
- struct addrinfo hints; /* temp, needed to open a socket connection */
- struct addrinfo *addrinfo; /* temp, needed to open a socket connection */
- SOCKET sockdata = 0; /* socket descriptor of the data connection */
- struct sockaddr_storage saddr; /* temp, needed to retrieve the network data port chosen on the local machine */
- socklen_t saddrlen; /* temp, needed to retrieve the network data port chosen on the local machine */
- int ai_family; /* temp, keeps the address family used by the control connection */
- /* RPCAP-related variables*/
- struct rpcap_header header; /* header of the RPCAP packet */
- struct rpcap_startcapreq *startcapreq; /* start capture request message */
- struct rpcap_startcapreply startcapreply; /* start capture reply message */
- /* Variables related to the buffer setting */
- int res;
- socklen_t itemp;
- int sockbufsize = 0;
- uint32 server_sockbufsize;
- /*
- * Let's check if sampling has been required.
- * If so, let's set it first
- */
- if (pcap_setsampling_remote(fp) != 0)
- return -1;
- /* detect if we're in active mode */
- temp = activeHosts;
- while (temp)
- {
- if (temp->sockctrl == pr->rmt_sockctrl)
- {
- active = 1;
- break;
- }
- temp = temp->next;
- }
- addrinfo = NULL;
- /*
- * Gets the complete sockaddr structure used in the ctrl connection
- * This is needed to get the address family of the control socket
- * Tip: I cannot save the ai_family of the ctrl sock in the pcap_t struct,
- * since the ctrl socket can already be open in case of active mode;
- * so I would have to call getpeername() anyway
- */
- saddrlen = sizeof(struct sockaddr_storage);
- if (getpeername(pr->rmt_sockctrl, (struct sockaddr *) &saddr, &saddrlen) == -1)
- {
- sock_geterror("getsockname(): ", fp->errbuf, PCAP_ERRBUF_SIZE);
- goto error_nodiscard;
- }
- ai_family = ((struct sockaddr_storage *) &saddr)->ss_family;
- /* Get the numeric address of the remote host we are connected to */
- if (getnameinfo((struct sockaddr *) &saddr, saddrlen, host,
- sizeof(host), NULL, 0, NI_NUMERICHOST))
- {
- sock_geterror("getnameinfo(): ", fp->errbuf, PCAP_ERRBUF_SIZE);
- goto error_nodiscard;
- }
- /*
- * Data connection is opened by the server toward the client if:
- * - we're using TCP, and the user wants us to be in active mode
- * - we're using UDP
- */
- if ((active) || (pr->rmt_flags & PCAP_OPENFLAG_DATATX_UDP))
- {
- /*
- * We have to create a new socket to receive packets
- * We have to do that immediately, since we have to tell the other
- * end which network port we picked up
- */
- memset(&hints, 0, sizeof(struct addrinfo));
- /* TEMP addrinfo is NULL in case of active */
- hints.ai_family = ai_family; /* Use the same address family of the control socket */
- hints.ai_socktype = (pr->rmt_flags & PCAP_OPENFLAG_DATATX_UDP) ? SOCK_DGRAM : SOCK_STREAM;
- hints.ai_flags = AI_PASSIVE; /* Data connection is opened by the server toward the client */
- /* Let's the server pick up a free network port for us */
- if (sock_initaddress(NULL, "0", &hints, &addrinfo, fp->errbuf, PCAP_ERRBUF_SIZE) == -1)
- goto error_nodiscard;
- if ((sockdata = sock_open(addrinfo, SOCKOPEN_SERVER,
- 1 /* max 1 connection in queue */, fp->errbuf, PCAP_ERRBUF_SIZE)) == INVALID_SOCKET)
- goto error_nodiscard;
- /* addrinfo is no longer used */
- freeaddrinfo(addrinfo);
- addrinfo = NULL;
- /* get the complete sockaddr structure used in the data connection */
- saddrlen = sizeof(struct sockaddr_storage);
- if (getsockname(sockdata, (struct sockaddr *) &saddr, &saddrlen) == -1)
- {
- sock_geterror("getsockname(): ", fp->errbuf, PCAP_ERRBUF_SIZE);
- goto error_nodiscard;
- }
- /* Get the local port the system picked up */
- if (getnameinfo((struct sockaddr *) &saddr, saddrlen, NULL,
- 0, portdata, sizeof(portdata), NI_NUMERICSERV))
- {
- sock_geterror("getnameinfo(): ", fp->errbuf, PCAP_ERRBUF_SIZE);
- goto error_nodiscard;
- }
- }
- /*
- * Now it's time to start playing with the RPCAP protocol
- * RPCAP start capture command: create the request message
- */
- if (sock_bufferize(NULL, sizeof(struct rpcap_header), NULL,
- &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, fp->errbuf, PCAP_ERRBUF_SIZE))
- goto error_nodiscard;
- rpcap_createhdr((struct rpcap_header *) sendbuf,
- pr->protocol_version, RPCAP_MSG_STARTCAP_REQ, 0,
- sizeof(struct rpcap_startcapreq) + sizeof(struct rpcap_filter) + fp->fcode.bf_len * sizeof(struct rpcap_filterbpf_insn));
- /* Fill the structure needed to open an adapter remotely */
- startcapreq = (struct rpcap_startcapreq *) &sendbuf[sendbufidx];
- if (sock_bufferize(NULL, sizeof(struct rpcap_startcapreq), NULL,
- &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, fp->errbuf, PCAP_ERRBUF_SIZE))
- goto error_nodiscard;
- memset(startcapreq, 0, sizeof(struct rpcap_startcapreq));
- /* By default, apply half the timeout on one side, half of the other */
- fp->opt.timeout = fp->opt.timeout / 2;
- startcapreq->read_timeout = htonl(fp->opt.timeout);
- /* portdata on the openreq is meaningful only if we're in active mode */
- if ((active) || (pr->rmt_flags & PCAP_OPENFLAG_DATATX_UDP))
- {
- sscanf(portdata, "%d", (int *)&(startcapreq->portdata)); /* cast to avoid a compiler warning */
- startcapreq->portdata = htons(startcapreq->portdata);
- }
- startcapreq->snaplen = htonl(fp->snapshot);
- startcapreq->flags = 0;
- if (pr->rmt_flags & PCAP_OPENFLAG_PROMISCUOUS)
- startcapreq->flags |= RPCAP_STARTCAPREQ_FLAG_PROMISC;
- if (pr->rmt_flags & PCAP_OPENFLAG_DATATX_UDP)
- startcapreq->flags |= RPCAP_STARTCAPREQ_FLAG_DGRAM;
- if (active)
- startcapreq->flags |= RPCAP_STARTCAPREQ_FLAG_SERVEROPEN;
- startcapreq->flags = htons(startcapreq->flags);
- /* Pack the capture filter */
- if (pcap_pack_bpffilter(fp, &sendbuf[sendbufidx], &sendbufidx, &fp->fcode))
- goto error_nodiscard;
- if (sock_send(pr->rmt_sockctrl, sendbuf, sendbufidx, fp->errbuf,
- PCAP_ERRBUF_SIZE) < 0)
- goto error_nodiscard;
- /* Receive and process the reply message header. */
- if (rpcap_process_msg_header(pr->rmt_sockctrl, pr->protocol_version,
- RPCAP_MSG_STARTCAP_REQ, &header, fp->errbuf) == -1)
- goto error_nodiscard;
- plen = header.plen;
- if (rpcap_recv(pr->rmt_sockctrl, (char *)&startcapreply,
- sizeof(struct rpcap_startcapreply), &plen, fp->errbuf) == -1)
- goto error;
- /*
- * In case of UDP data stream, the connection is always opened by the daemon
- * So, this case is already covered by the code above.
- * Now, we have still to handle TCP connections, because:
- * - if we're in active mode, we have to wait for a remote connection
- * - if we're in passive more, we have to start a connection
- *
- * We have to do he job in two steps because in case we're opening a TCP connection, we have
- * to tell the port we're using to the remote side; in case we're accepting a TCP
- * connection, we have to wait this info from the remote side.
- */
- if (!(pr->rmt_flags & PCAP_OPENFLAG_DATATX_UDP))
- {
- if (!active)
- {
- memset(&hints, 0, sizeof(struct addrinfo));
- hints.ai_family = ai_family; /* Use the same address family of the control socket */
- hints.ai_socktype = (pr->rmt_flags & PCAP_OPENFLAG_DATATX_UDP) ? SOCK_DGRAM : SOCK_STREAM;
- pcap_snprintf(portdata, PCAP_BUF_SIZE, "%d", ntohs(startcapreply.portdata));
- /* Let's the server pick up a free network port for us */
- if (sock_initaddress(host, portdata, &hints, &addrinfo, fp->errbuf, PCAP_ERRBUF_SIZE) == -1)
- goto error;
- if ((sockdata = sock_open(addrinfo, SOCKOPEN_CLIENT, 0, fp->errbuf, PCAP_ERRBUF_SIZE)) == INVALID_SOCKET)
- goto error;
- /* addrinfo is no longer used */
- freeaddrinfo(addrinfo);
- addrinfo = NULL;
- }
- else
- {
- SOCKET socktemp; /* We need another socket, since we're going to accept() a connection */
- /* Connection creation */
- saddrlen = sizeof(struct sockaddr_storage);
- socktemp = accept(sockdata, (struct sockaddr *) &saddr, &saddrlen);
- if (socktemp == INVALID_SOCKET)
- {
- sock_geterror("accept(): ", fp->errbuf, PCAP_ERRBUF_SIZE);
- goto error;
- }
- /* Now that I accepted the connection, the server socket is no longer needed */
- sock_close(sockdata, fp->errbuf, PCAP_ERRBUF_SIZE);
- sockdata = socktemp;
- }
- }
- /* Let's save the socket of the data connection */
- pr->rmt_sockdata = sockdata;
- /*
- * Set the size of the socket buffer for the data socket.
- * It has the same size as the local capture buffer used
- * on the other side of the connection.
- */
- server_sockbufsize = ntohl(startcapreply.bufsize);
- /* Let's get the actual size of the socket buffer */
- itemp = sizeof(sockbufsize);
- res = getsockopt(sockdata, SOL_SOCKET, SO_RCVBUF, (char *)&sockbufsize, &itemp);
- if (res == -1)
- {
- sock_geterror("pcap_startcapture_remote()", fp->errbuf, PCAP_ERRBUF_SIZE);
- SOCK_DEBUG_MESSAGE(fp->errbuf);
- }
- /*
- * Warning: on some kernels (e.g. Linux), the size of the user
- * buffer does not take into account the pcap_header and such,
- * and it is set equal to the snaplen.
- *
- * In my view, this is wrong (the meaning of the bufsize became
- * a bit strange). So, here bufsize is the whole size of the
- * user buffer. In case the bufsize returned is too small,
- * let's adjust it accordingly.
- */
- if (server_sockbufsize <= (u_int) fp->snapshot)
- server_sockbufsize += sizeof(struct pcap_pkthdr);
- /* if the current socket buffer is smaller than the desired one */
- if ((u_int) sockbufsize < server_sockbufsize)
- {
- /*
- * Loop until the buffer size is OK or the original
- * socket buffer size is larger than this one.
- */
- for (;;)
- {
- res = setsockopt(sockdata, SOL_SOCKET, SO_RCVBUF,
- (char *)&(server_sockbufsize),
- sizeof(server_sockbufsize));
- if (res == 0)
- break;
- /*
- * If something goes wrong, halve the buffer size
- * (checking that it does not become smaller than
- * the current one).
- */
- server_sockbufsize /= 2;
- if ((u_int) sockbufsize >= server_sockbufsize)
- {
- server_sockbufsize = sockbufsize;
- break;
- }
- }
- }
- /*
- * Let's allocate the packet; this is required in order to put
- * the packet somewhere when extracting data from the socket.
- * Since buffering has already been done in the socket buffer,
- * here we need just a buffer whose size is equal to the
- * largest possible packet message for the snapshot size,
- * namely the length of the message header plus the length
- * of the packet header plus the snapshot length.
- */
- fp->bufsize = sizeof(struct rpcap_header) + sizeof(struct rpcap_pkthdr) + fp->snapshot;
- fp->buffer = (u_char *)malloc(fp->bufsize);
- if (fp->buffer == NULL)
- {
- pcap_fmt_errmsg_for_errno(fp->errbuf, PCAP_ERRBUF_SIZE,
- errno, "malloc");
- goto error;
- }
- /*
- * The buffer is currently empty.
- */
- fp->bp = fp->buffer;
- fp->cc = 0;
- /* Discard the rest of the message. */
- if (rpcap_discard(pr->rmt_sockctrl, plen, fp->errbuf) == -1)
- goto error_nodiscard;
- /*
- * In case the user does not want to capture RPCAP packets, let's update the filter
- * We have to update it here (instead of sending it into the 'StartCapture' message
- * because when we generate the 'start capture' we do not know (yet) all the ports
- * we're currently using.
- */
- if (pr->rmt_flags & PCAP_OPENFLAG_NOCAPTURE_RPCAP)
- {
- struct bpf_program fcode;
- if (pcap_createfilter_norpcappkt(fp, &fcode) == -1)
- goto error;
- /* We cannot use 'pcap_setfilter_rpcap' because formally the capture has not been started yet */
- /* (the 'pr->rmt_capstarted' variable will be updated some lines below) */
- if (pcap_updatefilter_remote(fp, &fcode) == -1)
- goto error;
- pcap_freecode(&fcode);
- }
- pr->rmt_capstarted = 1;
- return 0;
- error:
- /*
- * When the connection has been established, we have to close it. So, at the
- * beginning of this function, if an error occur we return immediately with
- * a return NULL; when the connection is established, we have to come here
- * ('goto error;') in order to close everything properly.
- */
- /*
- * Discard the rest of the message.
- * We already reported an error; if this gets an error, just
- * drive on.
- */
- (void)rpcap_discard(pr->rmt_sockctrl, plen, NULL);
- error_nodiscard:
- if ((sockdata) && (sockdata != -1)) /* we can be here because sockdata said 'error' */
- sock_close(sockdata, NULL, 0);
- if (!active)
- sock_close(pr->rmt_sockctrl, NULL, 0);
- if (addrinfo != NULL)
- freeaddrinfo(addrinfo);
- /*
- * We do not have to call pcap_close() here, because this function is always called
- * by the user in case something bad happens
- */
- #if 0
- if (fp)
- {
- pcap_close(fp);
- fp= NULL;
- }
- #endif
- return -1;
- }
- /*
- * This function takes a bpf program and sends it to the other host.
- *
- * This function can be called in two cases:
- * - pcap_startcapture_remote() is called (we have to send the filter
- * along with the 'start capture' command)
- * - we want to udpate the filter during a capture (i.e. pcap_setfilter()
- * after the capture has been started)
- *
- * This function serializes the filter into the sending buffer ('sendbuf',
- * passed as a parameter) and return back. It does not send anything on
- * the network.
- *
- * \param fp: the pcap_t descriptor of the device currently opened.
- *
- * \param sendbuf: the buffer on which the serialized data has to copied.
- *
- * \param sendbufidx: it is used to return the abounf of bytes copied into the buffer.
- *
- * \param prog: the bpf program we have to copy.
- *
- * \return '0' if everything is fine, '-1' otherwise. The error message (if one)
- * is returned into the 'errbuf' field of the pcap_t structure.
- */
- static int pcap_pack_bpffilter(pcap_t *fp, char *sendbuf, int *sendbufidx, struct bpf_program *prog)
- {
- struct rpcap_filter *filter;
- struct rpcap_filterbpf_insn *insn;
- struct bpf_insn *bf_insn;
- struct bpf_program fake_prog; /* To be used just in case the user forgot to set a filter */
- unsigned int i;
- if (prog->bf_len == 0) /* No filters have been specified; so, let's apply a "fake" filter */
- {
- if (pcap_compile(fp, &fake_prog, NULL /* buffer */, 1, 0) == -1)
- return -1;
- prog = &fake_prog;
- }
- filter = (struct rpcap_filter *) sendbuf;
- if (sock_bufferize(NULL, sizeof(struct rpcap_filter), NULL, sendbufidx,
- RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, fp->errbuf, PCAP_ERRBUF_SIZE))
- return -1;
- filter->filtertype = htons(RPCAP_UPDATEFILTER_BPF);
- filter->nitems = htonl((int32)prog->bf_len);
- if (sock_bufferize(NULL, prog->bf_len * sizeof(struct rpcap_filterbpf_insn),
- NULL, sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, fp->errbuf, PCAP_ERRBUF_SIZE))
- return -1;
- insn = (struct rpcap_filterbpf_insn *) (filter + 1);
- bf_insn = prog->bf_insns;
- for (i = 0; i < prog->bf_len; i++)
- {
- insn->code = htons(bf_insn->code);
- insn->jf = bf_insn->jf;
- insn->jt = bf_insn->jt;
- insn->k = htonl(bf_insn->k);
- insn++;
- bf_insn++;
- }
- return 0;
- }
- /*
- * This function updates a filter on a remote host.
- *
- * It is called when the user wants to update a filter.
- * In case we're capturing from the network, it sends the filter to our
- * peer.
- * This function is *not* called automatically when the user calls
- * pcap_setfilter().
- * There will be two cases:
- * - the capture has been started: in this case, pcap_setfilter_rpcap()
- * calls pcap_updatefilter_remote()
- * - the capture has not started yet: in this case, pcap_setfilter_rpcap()
- * stores the filter into the pcap_t structure, and then the filter is
- * sent with pcap_startcap().
- *
- * WARNING This function *does not* clear the packet currently into the
- * buffers. Therefore, the user has to expect to receive some packets
- * that are related to the previous filter. If you want to discard all
- * the packets before applying a new filter, you have to close the
- * current capture session and start a new one.
- *
- * XXX - we really should have pcap_setfilter() always discard packets
- * received with the old filter, and have a separate pcap_setfilter_noflush()
- * function that doesn't discard any packets.
- */
- static int pcap_updatefilter_remote(pcap_t *fp, struct bpf_program *prog)
- {
- struct pcap_rpcap *pr = fp->priv; /* structure used when doing a remote live capture */
- char sendbuf[RPCAP_NETBUF_SIZE]; /* temporary buffer in which data to be sent is buffered */
- int sendbufidx = 0; /* index which keeps the number of bytes currently buffered */
- struct rpcap_header header; /* To keep the reply message */
- if (sock_bufferize(NULL, sizeof(struct rpcap_header), NULL, &sendbufidx,
- RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, fp->errbuf, PCAP_ERRBUF_SIZE))
- return -1;
- rpcap_createhdr((struct rpcap_header *) sendbuf,
- pr->protocol_version, RPCAP_MSG_UPDATEFILTER_REQ, 0,
- sizeof(struct rpcap_filter) + prog->bf_len * sizeof(struct rpcap_filterbpf_insn));
- if (pcap_pack_bpffilter(fp, &sendbuf[sendbufidx], &sendbufidx, prog))
- return -1;
- if (sock_send(pr->rmt_sockctrl, sendbuf, sendbufidx, fp->errbuf,
- PCAP_ERRBUF_SIZE) < 0)
- return -1;
- /* Receive and process the reply message header. */
- if (rpcap_process_msg_header(pr->rmt_sockctrl, pr->protocol_version,
- RPCAP_MSG_UPDATEFILTER_REQ, &header, fp->errbuf) == -1)
- return -1;
- /*
- * It shouldn't have any contents; discard it if it does.
- */
- if (rpcap_discard(pr->rmt_sockctrl, header.plen, fp->errbuf) == -1)
- return -1;
- return 0;
- }
- static void
- pcap_save_current_filter_rpcap(pcap_t *fp, const char *filter)
- {
- struct pcap_rpcap *pr = fp->priv; /* structure used when doing a remote live capture */
- /*
- * Check if:
- * - We are on an remote capture
- * - we do not want to capture RPCAP traffic
- *
- * If so, we have to save the current filter, because we have to
- * add some piece of stuff later
- */
- if (pr->rmt_clientside &&
- (pr->rmt_flags & PCAP_OPENFLAG_NOCAPTURE_RPCAP))
- {
- if (pr->currentfilter)
- free(pr->currentfilter);
- if (filter == NULL)
- filter = "";
- pr->currentfilter = strdup(filter);
- }
- }
- /*
- * This function sends a filter to a remote host.
- *
- * This function is called when the user wants to set a filter.
- * It sends the filter to our peer.
- * This function is called automatically when the user calls pcap_setfilter().
- *
- * Parameters and return values are exactly the same of pcap_setfilter().
- */
- static int pcap_setfilter_rpcap(pcap_t *fp, struct bpf_program *prog)
- {
- struct pcap_rpcap *pr = fp->priv; /* structure used when doing a remote live capture */
- if (!pr->rmt_capstarted)
- {
- /* copy filter into the pcap_t structure */
- if (install_bpf_program(fp, prog) == -1)
- return -1;
- return 0;
- }
- /* we have to update a filter during run-time */
- if (pcap_updatefilter_remote(fp, prog))
- return -1;
- return 0;
- }
- /*
- * This function updates the current filter in order not to capture rpcap
- * packets.
- *
- * This function is called *only* when the user wants exclude RPCAP packets
- * related to the current session from the captured packets.
- *
- * \return '0' if everything is fine, '-1' otherwise. The error message (if one)
- * is returned into the 'errbuf' field of the pcap_t structure.
- */
- static int pcap_createfilter_norpcappkt(pcap_t *fp, struct bpf_program *prog)
- {
- struct pcap_rpcap *pr = fp->priv; /* structure used when doing a remote live capture */
- int RetVal = 0;
- /* We do not want to capture our RPCAP traffic. So, let's update the filter */
- if (pr->rmt_flags & PCAP_OPENFLAG_NOCAPTURE_RPCAP)
- {
- struct sockaddr_storage saddr; /* temp, needed to retrieve the network data port chosen on the local machine */
- socklen_t saddrlen; /* temp, needed to retrieve the network data port chosen on the local machine */
- char myaddress[128];
- char myctrlport[128];
- char mydataport[128];
- char peeraddress[128];
- char peerctrlport[128];
- char *newfilter;
- const int newstringsize = 1024;
- size_t currentfiltersize;
- /* Get the name/port of our peer */
- saddrlen = sizeof(struct sockaddr_storage);
- if (getpeername(pr->rmt_sockctrl, (struct sockaddr *) &saddr, &saddrlen) == -1)
- {
- sock_geterror("getpeername(): ", fp->errbuf, PCAP_ERRBUF_SIZE);
- return -1;
- }
- if (getnameinfo((struct sockaddr *) &saddr, saddrlen, peeraddress,
- sizeof(peeraddress), peerctrlport, sizeof(peerctrlport), NI_NUMERICHOST | NI_NUMERICSERV))
- {
- sock_geterror("getnameinfo(): ", fp->errbuf, PCAP_ERRBUF_SIZE);
- return -1;
- }
- /* We cannot check the data port, because this is available only in case of TCP sockets */
- /* Get the name/port of the current host */
- if (getsockname(pr->rmt_sockctrl, (struct sockaddr *) &saddr, &saddrlen) == -1)
- {
- sock_geterror("getsockname(): ", fp->errbuf, PCAP_ERRBUF_SIZE);
- return -1;
- }
- /* Get the local port the system picked up */
- if (getnameinfo((struct sockaddr *) &saddr, saddrlen, myaddress,
- sizeof(myaddress), myctrlport, sizeof(myctrlport), NI_NUMERICHOST | NI_NUMERICSERV))
- {
- sock_geterror("getnameinfo(): ", fp->errbuf, PCAP_ERRBUF_SIZE);
- return -1;
- }
- /* Let's now check the data port */
- if (getsockname(pr->rmt_sockdata, (struct sockaddr *) &saddr, &saddrlen) == -1)
- {
- sock_geterror("getsockname(): ", fp->errbuf, PCAP_ERRBUF_SIZE);
- return -1;
- }
- /* Get the local port the system picked up */
- if (getnameinfo((struct sockaddr *) &saddr, saddrlen, NULL, 0, mydataport, sizeof(mydataport), NI_NUMERICSERV))
- {
- sock_geterror("getnameinfo(): ", fp->errbuf, PCAP_ERRBUF_SIZE);
- return -1;
- }
- currentfiltersize = pr->currentfilter ? strlen(pr->currentfilter) : 0;
- newfilter = (char *)malloc(currentfiltersize + newstringsize + 1);
- if (currentfiltersize)
- {
- pcap_snprintf(newfilter, currentfiltersize + newstringsize,
- "(%s) and not (host %s and host %s and port %s and port %s) and not (host %s and host %s and port %s)",
- pr->currentfilter, myaddress, peeraddress, myctrlport, peerctrlport, myaddress, peeraddress, mydataport);
- }
- else
- {
- pcap_snprintf(newfilter, currentfiltersize + newstringsize,
- "not (host %s and host %s and port %s and port %s) and not (host %s and host %s and port %s)",
- myaddress, peeraddress, myctrlport, peerctrlport, myaddress, peeraddress, mydataport);
- }
- newfilter[currentfiltersize + newstringsize] = 0;
- /*
- * This is only an hack to prevent the save_current_filter
- * routine, which will be called when we call pcap_compile(),
- * from saving the modified filter.
- */
- pr->rmt_clientside = 0;
- if (pcap_compile(fp, prog, newfilter, 1, 0) == -1)
- RetVal = -1;
- /* Undo the hack. */
- pr->rmt_clientside = 1;
- free(newfilter);
- }
- return RetVal;
- }
- /*
- * This function sets sampling parameters in the remote host.
- *
- * It is called when the user wants to set activate sampling on the
- * remote host.
- *
- * Sampling parameters are defined into the 'pcap_t' structure.
- *
- * \param p: the pcap_t descriptor of the device currently opened.
- *
- * \return '0' if everything is OK, '-1' is something goes wrong. The
- * error message is returned in the 'errbuf' member of the pcap_t structure.
- */
- static int pcap_setsampling_remote(pcap_t *fp)
- {
- struct pcap_rpcap *pr = fp->priv; /* structure used when doing a remote live capture */
- char sendbuf[RPCAP_NETBUF_SIZE];/* temporary buffer in which data to be sent is buffered */
- int sendbufidx = 0; /* index which keeps the number of bytes currently buffered */
- struct rpcap_header header; /* To keep the reply message */
- struct rpcap_sampling *sampling_pars; /* Structure that is needed to send sampling parameters to the remote host */
- /* If no samping is requested, return 'ok' */
- if (fp->rmt_samp.method == PCAP_SAMP_NOSAMP)
- return 0;
- /*
- * Check for sampling parameters that don't fit in a message.
- * We'll let the server complain about invalid parameters
- * that do fit into the message.
- */
- if (fp->rmt_samp.method < 0 || fp->rmt_samp.method > 255) {
- pcap_snprintf(fp->errbuf, PCAP_ERRBUF_SIZE,
- "Invalid sampling method %d", fp->rmt_samp.method);
- return -1;
- }
- if (fp->rmt_samp.value < 0 || fp->rmt_samp.value > 65535) {
- pcap_snprintf(fp->errbuf, PCAP_ERRBUF_SIZE,
- "Invalid sampling value %d", fp->rmt_samp.value);
- return -1;
- }
- if (sock_bufferize(NULL, sizeof(struct rpcap_header), NULL,
- &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, fp->errbuf, PCAP_ERRBUF_SIZE))
- return -1;
- rpcap_createhdr((struct rpcap_header *) sendbuf,
- pr->protocol_version, RPCAP_MSG_SETSAMPLING_REQ, 0,
- sizeof(struct rpcap_sampling));
- /* Fill the structure needed to open an adapter remotely */
- sampling_pars = (struct rpcap_sampling *) &sendbuf[sendbufidx];
- if (sock_bufferize(NULL, sizeof(struct rpcap_sampling), NULL,
- &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, fp->errbuf, PCAP_ERRBUF_SIZE))
- return -1;
- memset(sampling_pars, 0, sizeof(struct rpcap_sampling));
- sampling_pars->method = (uint8)fp->rmt_samp.method;
- sampling_pars->value = (uint16)htonl(fp->rmt_samp.value);
- if (sock_send(pr->rmt_sockctrl, sendbuf, sendbufidx, fp->errbuf,
- PCAP_ERRBUF_SIZE) < 0)
- return -1;
- /* Receive and process the reply message header. */
- if (rpcap_process_msg_header(pr->rmt_sockctrl, pr->protocol_version,
- RPCAP_MSG_SETSAMPLING_REQ, &header, fp->errbuf) == -1)
- return -1;
- /*
- * It shouldn't have any contents; discard it if it does.
- */
- if (rpcap_discard(pr->rmt_sockctrl, header.plen, fp->errbuf) == -1)
- return -1;
- return 0;
- }
- /*********************************************************
- * *
- * Miscellaneous functions *
- * *
- *********************************************************/
- /*
- * This function performs authentication and protocol version
- * negotiation. It first tries to authenticate with the maximum
- * version we support and, if that fails with an "I don't support
- * that version" error from the server, and the version number in
- * the reply from the server is one we support, tries again with
- * that version.
- *
- * \param sock: the socket we are currently using.
- *
- * \param ver: pointer to variable holding protocol version number to send
- * and to set to the protocol version number in the reply.
- *
- * \param auth: authentication parameters that have to be sent.
- *
- * \param errbuf: a pointer to a user-allocated buffer (of size
- * PCAP_ERRBUF_SIZE) that will contain the error message (in case there
- * is one). It could be a network problem or the fact that the authorization
- * failed.
- *
- * \return '0' if everything is fine, '-1' for an error. For errors,
- * an error message string is returned in the 'errbuf' variable.
- */
- static int rpcap_doauth(SOCKET sockctrl, uint8 *ver, struct pcap_rmtauth *auth, char *errbuf)
- {
- int status;
- /*
- * Send authentication to the remote machine.
- *
- * First try with the maximum version number we support.
- */
- *ver = RPCAP_MAX_VERSION;
- status = rpcap_sendauth(sockctrl, ver, auth, errbuf);
- if (status == 0)
- {
- //
- // Success.
- //
- return 0;
- }
- if (status == -1)
- {
- /* Unrecoverable error. */
- return -1;
- }
- /*
- * The server doesn't support the version we used in the initial
- * message, and it sent us back a reply either with the maximum
- * version they do support, or with the version we sent, and we
- * support that version. *ver has been set to that version; try
- * authenticating again with that version.
- */
- status = rpcap_sendauth(sockctrl, ver, auth, errbuf);
- if (status == 0)
- {
- //
- // Success.
- //
- return 0;
- }
- if (status == -1)
- {
- /* Unrecoverable error. */
- return -1;
- }
- if (status == -2)
- {
- /*
- * The server doesn't support that version, which
- * means there is no version we both support, so
- * this is a fatal error.
- */
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "The server doesn't support any protocol version that we support");
- return -1;
- }
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "rpcap_sendauth() returned %d", status);
- return -1;
- }
- /*
- * This function sends the authentication message.
- *
- * It sends the authentication parameters on the control socket.
- * It is required in order to open the connection with the other end party.
- *
- * \param sock: the socket we are currently using.
- *
- * \param ver: pointer to variable holding protocol version number to send
- * and to set to the protocol version number in the reply.
- *
- * \param auth: authentication parameters that have to be sent.
- *
- * \param errbuf: a pointer to a user-allocated buffer (of size
- * PCAP_ERRBUF_SIZE) that will contain the error message (in case there
- * is one). It could be a network problem or the fact that the authorization
- * failed.
- *
- * \return '0' if everything is fine, '-2' if the server didn't reply with
- * the protocol version we requested but replied with a version we do
- * support, or '-1' for other errors. For errors, an error message string
- * is returned in the 'errbuf' variable.
- */
- static int rpcap_sendauth(SOCKET sock, uint8 *ver, struct pcap_rmtauth *auth, char *errbuf)
- {
- char sendbuf[RPCAP_NETBUF_SIZE]; /* temporary buffer in which data that has to be sent is buffered */
- int sendbufidx = 0; /* index which keeps the number of bytes currently buffered */
- uint16 length; /* length of the payload of this message */
- uint16 errcode;
- struct rpcap_auth *rpauth;
- uint16 auth_type;
- struct rpcap_header header;
- size_t str_length;
- if (auth)
- {
- switch (auth->type)
- {
- case RPCAP_RMTAUTH_NULL:
- length = sizeof(struct rpcap_auth);
- break;
- case RPCAP_RMTAUTH_PWD:
- length = sizeof(struct rpcap_auth);
- if (auth->username)
- {
- str_length = strlen(auth->username);
- if (str_length > 65535)
- {
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "User name is too long (> 65535 bytes)");
- return -1;
- }
- length += (uint16)str_length;
- }
- if (auth->password)
- {
- str_length = strlen(auth->password);
- if (str_length > 65535)
- {
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "Password is too long (> 65535 bytes)");
- return -1;
- }
- length += (uint16)str_length;
- }
- break;
- default:
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "Authentication type not recognized.");
- return -1;
- }
- auth_type = (uint16)auth->type;
- }
- else
- {
- auth_type = RPCAP_RMTAUTH_NULL;
- length = sizeof(struct rpcap_auth);
- }
- if (sock_bufferize(NULL, sizeof(struct rpcap_header), NULL,
- &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, errbuf, PCAP_ERRBUF_SIZE))
- return -1;
- rpcap_createhdr((struct rpcap_header *) sendbuf, *ver,
- RPCAP_MSG_AUTH_REQ, 0, length);
- rpauth = (struct rpcap_auth *) &sendbuf[sendbufidx];
- if (sock_bufferize(NULL, sizeof(struct rpcap_auth), NULL,
- &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, errbuf, PCAP_ERRBUF_SIZE))
- return -1;
- memset(rpauth, 0, sizeof(struct rpcap_auth));
- rpauth->type = htons(auth_type);
- if (auth_type == RPCAP_RMTAUTH_PWD)
- {
- if (auth->username)
- rpauth->slen1 = (uint16)strlen(auth->username);
- else
- rpauth->slen1 = 0;
- if (sock_bufferize(auth->username, rpauth->slen1, sendbuf,
- &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_BUFFERIZE, errbuf, PCAP_ERRBUF_SIZE))
- return -1;
- if (auth->password)
- rpauth->slen2 = (uint16)strlen(auth->password);
- else
- rpauth->slen2 = 0;
- if (sock_bufferize(auth->password, rpauth->slen2, sendbuf,
- &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_BUFFERIZE, errbuf, PCAP_ERRBUF_SIZE))
- return -1;
- rpauth->slen1 = htons(rpauth->slen1);
- rpauth->slen2 = htons(rpauth->slen2);
- }
- if (sock_send(sock, sendbuf, sendbufidx, errbuf, PCAP_ERRBUF_SIZE) < 0)
- return -1;
- /* Receive the reply */
- if (rpcap_recv_msg_header(sock, &header, errbuf) == -1)
- return -1;
- if (rpcap_check_msg_type(sock, RPCAP_MSG_AUTH_REQ, &header,
- &errcode, errbuf) == -1)
- {
- /* Error message - or something else, which is a protocol error. */
- if (header.type == RPCAP_MSG_ERROR &&
- errcode == PCAP_ERR_WRONGVER)
- {
- /*
- * The server didn't support the version we sent,
- * and replied with the maximum version it supports
- * if our version was too big or with the version
- * we sent if out version was too small.
- *
- * Do we also support it?
- */
- if (!RPCAP_VERSION_IS_SUPPORTED(header.ver))
- {
- /*
- * No, so there's no version we both support.
- * This is an unrecoverable error.
- */
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "The server doesn't support any protocol version that we support");
- return -1;
- }
- /*
- * OK, use that version, and tell our caller to
- * try again.
- */
- *ver = header.ver;
- return -2;
- }
- /*
- * Other error - unrecoverable.
- */
- return -1;
- }
- /*
- * OK, it's an authentication reply, so they're OK with the
- * protocol version we sent.
- *
- * Discard the rest of it.
- */
- if (rpcap_discard(sock, header.plen, errbuf) == -1)
- return -1;
- return 0;
- }
- /* We don't currently support non-blocking mode. */
- static int
- pcap_getnonblock_rpcap(pcap_t *p)
- {
- pcap_snprintf(p->errbuf, PCAP_ERRBUF_SIZE,
- "Non-blocking mode isn't supported for capturing remotely with rpcap");
- return (-1);
- }
- static int
- pcap_setnonblock_rpcap(pcap_t *p, int nonblock _U_)
- {
- pcap_snprintf(p->errbuf, PCAP_ERRBUF_SIZE,
- "Non-blocking mode isn't supported for capturing remotely with rpcap");
- return (-1);
- }
- /*
- * This function opens a remote adapter by opening an RPCAP connection and
- * so on.
- *
- * It does the job of pcap_open_live() for a remote interface; it's called
- * by pcap_open() for remote interfaces.
- *
- * We do not start the capture until pcap_startcapture_remote() is called.
- *
- * This is because, when doing a remote capture, we cannot start capturing
- * data as soon as the 'open adapter' command is sent. Suppose the remote
- * adapter is already overloaded; if we start a capture (which, by default,
- * has a NULL filter) the new traffic can saturate the network.
- *
- * Instead, we want to "open" the adapter, then send a "start capture"
- * command only when we're ready to start the capture.
- * This function does this job: it sends an "open adapter" command
- * (according to the RPCAP protocol), but it does not start the capture.
- *
- * Since the other libpcap functions do not share this way of life, we
- * have to do some dirty things in order to make everything work.
- *
- * \param source: see pcap_open().
- * \param snaplen: see pcap_open().
- * \param flags: see pcap_open().
- * \param read_timeout: see pcap_open().
- * \param auth: see pcap_open().
- * \param errbuf: see pcap_open().
- *
- * \return a pcap_t pointer in case of success, NULL otherwise. In case of
- * success, the pcap_t pointer can be used as a parameter to the following
- * calls (pcap_compile() and so on). In case of problems, errbuf contains
- * a text explanation of error.
- *
- * WARNING: In case we call pcap_compile() and the capture has not yet
- * been started, the filter will be saved into the pcap_t structure,
- * and it will be sent to the other host later (when
- * pcap_startcapture_remote() is called).
- */
- pcap_t *pcap_open_rpcap(const char *source, int snaplen, int flags, int read_timeout, struct pcap_rmtauth *auth, char *errbuf)
- {
- pcap_t *fp;
- char *source_str;
- struct pcap_rpcap *pr; /* structure used when doing a remote live capture */
- char host[PCAP_BUF_SIZE], ctrlport[PCAP_BUF_SIZE], iface[PCAP_BUF_SIZE];
- struct activehosts *activeconn; /* active connection, if there is one */
- int error; /* '1' if rpcap_remoteact_getsock returned an error */
- SOCKET sockctrl;
- uint8 protocol_version; /* negotiated protocol version */
- int active;
- uint32 plen;
- char sendbuf[RPCAP_NETBUF_SIZE]; /* temporary buffer in which data to be sent is buffered */
- int sendbufidx = 0; /* index which keeps the number of bytes currently buffered */
- int retval; /* store the return value of the functions */
- /* RPCAP-related variables */
- struct rpcap_header header; /* header of the RPCAP packet */
- struct rpcap_openreply openreply; /* open reply message */
- fp = pcap_create_common(errbuf, sizeof (struct pcap_rpcap));
- if (fp == NULL)
- {
- return NULL;
- }
- source_str = strdup(source);
- if (source_str == NULL) {
- pcap_fmt_errmsg_for_errno(errbuf, PCAP_ERRBUF_SIZE,
- errno, "malloc");
- return NULL;
- }
- /*
- * Turn a negative snapshot value (invalid), a snapshot value of
- * 0 (unspecified), or a value bigger than the normal maximum
- * value, into the maximum allowed value.
- *
- * If some application really *needs* a bigger snapshot
- * length, we should just increase MAXIMUM_SNAPLEN.
- *
- * XXX - should we leave this up to the remote server to
- * do?
- */
- if (snaplen <= 0 || snaplen > MAXIMUM_SNAPLEN)
- snaplen = MAXIMUM_SNAPLEN;
- fp->opt.device = source_str;
- fp->snapshot = snaplen;
- fp->opt.timeout = read_timeout;
- pr = fp->priv;
- pr->rmt_flags = flags;
- /*
- * determine the type of the source (NULL, file, local, remote)
- * You must have a valid source string even if we're in active mode, because otherwise
- * the call to the following function will fail.
- */
- if (pcap_parsesrcstr(fp->opt.device, &retval, host, ctrlport, iface, errbuf) == -1)
- {
- pcap_close(fp);
- return NULL;
- }
- if (retval != PCAP_SRC_IFREMOTE)
- {
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "This function is able to open only remote interfaces");
- pcap_close(fp);
- return NULL;
- }
- /*
- * Warning: this call can be the first one called by the user.
- * For this reason, we have to initialize the WinSock support.
- */
- if (sock_init(errbuf, PCAP_ERRBUF_SIZE) == -1)
- {
- pcap_close(fp);
- return NULL;
- }
- /* Check for active mode */
- activeconn = rpcap_remoteact_getsock(host, &error, errbuf);
- if (activeconn != NULL)
- {
- sockctrl = activeconn->sockctrl;
- protocol_version = activeconn->protocol_version;
- active = 1;
- }
- else
- {
- struct addrinfo hints; /* temp, needed to open a socket connection */
- struct addrinfo *addrinfo; /* temp, needed to open a socket connection */
- if (error)
- {
- /*
- * Call failed.
- */
- pcap_close(fp);
- return NULL;
- }
- /*
- * We're not in active mode; let's try to open a new
- * control connection.
- */
- memset(&hints, 0, sizeof(struct addrinfo));
- hints.ai_family = PF_UNSPEC;
- hints.ai_socktype = SOCK_STREAM;
- if (ctrlport[0] == 0)
- {
- /* the user chose not to specify the port */
- if (sock_initaddress(host, RPCAP_DEFAULT_NETPORT, &hints, &addrinfo, errbuf, PCAP_ERRBUF_SIZE) == -1)
- {
- pcap_close(fp);
- return NULL;
- }
- }
- else
- {
- if (sock_initaddress(host, ctrlport, &hints, &addrinfo, errbuf, PCAP_ERRBUF_SIZE) == -1)
- {
- pcap_close(fp);
- return NULL;
- }
- }
- if ((sockctrl = sock_open(addrinfo, SOCKOPEN_CLIENT, 0, errbuf, PCAP_ERRBUF_SIZE)) == INVALID_SOCKET)
- {
- freeaddrinfo(addrinfo);
- pcap_close(fp);
- return NULL;
- }
- /* addrinfo is no longer used */
- freeaddrinfo(addrinfo);
- if (rpcap_doauth(sockctrl, &protocol_version, auth, errbuf) == -1)
- {
- sock_close(sockctrl, NULL, 0);
- pcap_close(fp);
- return NULL;
- }
- active = 0;
- }
- /*
- * Now it's time to start playing with the RPCAP protocol
- * RPCAP open command: create the request message
- */
- if (sock_bufferize(NULL, sizeof(struct rpcap_header), NULL,
- &sendbufidx, RPCAP_NETBUF_SIZE, SOCKBUF_CHECKONLY, errbuf, PCAP_ERRBUF_SIZE))
- goto error_nodiscard;
- rpcap_createhdr((struct rpcap_header *) sendbuf, protocol_version,
- RPCAP_MSG_OPEN_REQ, 0, (uint32) strlen(iface));
- if (sock_bufferize(iface, (int) strlen(iface), sendbuf, &sendbufidx,
- RPCAP_NETBUF_SIZE, SOCKBUF_BUFFERIZE, errbuf, PCAP_ERRBUF_SIZE))
- goto error_nodiscard;
- if (sock_send(sockctrl, sendbuf, sendbufidx, errbuf,
- PCAP_ERRBUF_SIZE) < 0)
- goto error_nodiscard;
- /* Receive and process the reply message header. */
- if (rpcap_process_msg_header(sockctrl, protocol_version,
- RPCAP_MSG_OPEN_REQ, &header, errbuf) == -1)
- goto error_nodiscard;
- plen = header.plen;
- /* Read the reply body */
- if (rpcap_recv(sockctrl, (char *)&openreply,
- sizeof(struct rpcap_openreply), &plen, errbuf) == -1)
- goto error;
- /* Discard the rest of the message, if there is any. */
- if (rpcap_discard(pr->rmt_sockctrl, plen, errbuf) == -1)
- goto error_nodiscard;
- /* Set proper fields into the pcap_t struct */
- fp->linktype = ntohl(openreply.linktype);
- fp->tzoff = ntohl(openreply.tzoff);
- pr->rmt_sockctrl = sockctrl;
- pr->protocol_version = protocol_version;
- pr->rmt_clientside = 1;
- /* This code is duplicated from the end of this function */
- fp->read_op = pcap_read_rpcap;
- fp->save_current_filter_op = pcap_save_current_filter_rpcap;
- fp->setfilter_op = pcap_setfilter_rpcap;
- fp->getnonblock_op = pcap_getnonblock_rpcap;
- fp->setnonblock_op = pcap_setnonblock_rpcap;
- fp->stats_op = pcap_stats_rpcap;
- #ifdef _WIN32
- fp->stats_ex_op = pcap_stats_ex_rpcap;
- #endif
- fp->cleanup_op = pcap_cleanup_rpcap;
- fp->activated = 1;
- return fp;
- error:
- /*
- * When the connection has been established, we have to close it. So, at the
- * beginning of this function, if an error occur we return immediately with
- * a return NULL; when the connection is established, we have to come here
- * ('goto error;') in order to close everything properly.
- */
- /*
- * Discard the rest of the message.
- * We already reported an error; if this gets an error, just
- * drive on.
- */
- (void)rpcap_discard(pr->rmt_sockctrl, plen, NULL);
- error_nodiscard:
- if (!active)
- sock_close(sockctrl, NULL, 0);
- pcap_close(fp);
- return NULL;
- }
- /* String identifier to be used in the pcap_findalldevs_ex() */
- #define PCAP_TEXT_SOURCE_ADAPTER "Network adapter"
- /* String identifier to be used in the pcap_findalldevs_ex() */
- #define PCAP_TEXT_SOURCE_ON_REMOTE_HOST "on remote node"
- static void
- freeaddr(struct pcap_addr *addr)
- {
- free(addr->addr);
- free(addr->netmask);
- free(addr->broadaddr);
- free(addr->dstaddr);
- free(addr);
- }
- int
- pcap_findalldevs_ex_remote(char *source, struct pcap_rmtauth *auth, pcap_if_t **alldevs, char *errbuf)
- {
- struct activehosts *activeconn; /* active connection, if there is one */
- int error; /* '1' if rpcap_remoteact_getsock returned an error */
- uint8 protocol_version; /* protocol version */
- SOCKET sockctrl; /* socket descriptor of the control connection */
- uint32 plen;
- struct rpcap_header header; /* structure that keeps the general header of the rpcap protocol */
- int i, j; /* temp variables */
- int nif; /* Number of interfaces listed */
- int active; /* 'true' if we the other end-party is in active mode */
- int type;
- char host[PCAP_BUF_SIZE], port[PCAP_BUF_SIZE];
- char tmpstring[PCAP_BUF_SIZE + 1]; /* Needed to convert names and descriptions from 'old' syntax to the 'new' one */
- pcap_if_t *lastdev; /* Last device in the pcap_if_t list */
- pcap_if_t *dev; /* Device we're adding to the pcap_if_t list */
- /* List starts out empty. */
- (*alldevs) = NULL;
- lastdev = NULL;
- /* Retrieve the needed data for getting adapter list */
- if (pcap_parsesrcstr(source, &type, host, port, NULL, errbuf) == -1)
- return -1;
- /* Warning: this call can be the first one called by the user. */
- /* For this reason, we have to initialize the WinSock support. */
- if (sock_init(errbuf, PCAP_ERRBUF_SIZE) == -1)
- return -1;
- /* Check for active mode */
- activeconn = rpcap_remoteact_getsock(host, &error, errbuf);
- if (activeconn != NULL)
- {
- sockctrl = activeconn->sockctrl;
- protocol_version = activeconn->protocol_version;
- active = 1;
- }
- else
- {
- struct addrinfo hints; /* temp variable needed to resolve hostnames into to socket representation */
- struct addrinfo *addrinfo; /* temp variable needed to resolve hostnames into to socket representation */
- if (error)
- {
- /*
- * Call failed.
- */
- return -1;
- }
- /*
- * We're not in active mode; let's try to open a new
- * control connection.
- */
- memset(&hints, 0, sizeof(struct addrinfo));
- hints.ai_family = PF_UNSPEC;
- hints.ai_socktype = SOCK_STREAM;
- if (port[0] == 0)
- {
- /* the user chose not to specify the port */
- if (sock_initaddress(host, RPCAP_DEFAULT_NETPORT, &hints, &addrinfo, errbuf, PCAP_ERRBUF_SIZE) == -1)
- return -1;
- }
- else
- {
- if (sock_initaddress(host, port, &hints, &addrinfo, errbuf, PCAP_ERRBUF_SIZE) == -1)
- return -1;
- }
- if ((sockctrl = sock_open(addrinfo, SOCKOPEN_CLIENT, 0, errbuf, PCAP_ERRBUF_SIZE)) == INVALID_SOCKET)
- {
- freeaddrinfo(addrinfo);
- return -1;
- }
- /* addrinfo is no longer used */
- freeaddrinfo(addrinfo);
- addrinfo = NULL;
- if (rpcap_doauth(sockctrl, &protocol_version, auth, errbuf) == -1)
- {
- sock_close(sockctrl, NULL, 0);
- return -1;
- }
- active = 0;
- }
- /* RPCAP findalldevs command */
- rpcap_createhdr(&header, protocol_version, RPCAP_MSG_FINDALLIF_REQ,
- 0, 0);
- if (sock_send(sockctrl, (char *)&header, sizeof(struct rpcap_header),
- errbuf, PCAP_ERRBUF_SIZE) < 0)
- goto error_nodiscard;
- /* Receive and process the reply message header. */
- if (rpcap_process_msg_header(sockctrl, protocol_version,
- RPCAP_MSG_FINDALLIF_REQ, &header, errbuf) == -1)
- goto error_nodiscard;
- plen = header.plen;
- /* read the number of interfaces */
- nif = ntohs(header.value);
- /* loop until all interfaces have been received */
- for (i = 0; i < nif; i++)
- {
- struct rpcap_findalldevs_if findalldevs_if;
- char tmpstring2[PCAP_BUF_SIZE + 1]; /* Needed to convert names and descriptions from 'old' syntax to the 'new' one */
- size_t stringlen;
- struct pcap_addr *addr, *prevaddr;
- tmpstring2[PCAP_BUF_SIZE] = 0;
- /* receive the findalldevs structure from remote host */
- if (rpcap_recv(sockctrl, (char *)&findalldevs_if,
- sizeof(struct rpcap_findalldevs_if), &plen, errbuf) == -1)
- goto error;
- findalldevs_if.namelen = ntohs(findalldevs_if.namelen);
- findalldevs_if.desclen = ntohs(findalldevs_if.desclen);
- findalldevs_if.naddr = ntohs(findalldevs_if.naddr);
- /* allocate the main structure */
- dev = (pcap_if_t *)malloc(sizeof(pcap_if_t));
- if (dev == NULL)
- {
- pcap_fmt_errmsg_for_errno(errbuf, PCAP_ERRBUF_SIZE,
- errno, "malloc() failed");
- goto error;
- }
- /* Initialize the structure to 'zero' */
- memset(dev, 0, sizeof(pcap_if_t));
- /* Append it to the list. */
- if (lastdev == NULL)
- {
- /*
- * List is empty, so it's also the first device.
- */
- *alldevs = dev;
- }
- else
- {
- /*
- * Append after the last device.
- */
- lastdev->next = dev;
- }
- /* It's now the last device. */
- lastdev = dev;
- /* allocate mem for name and description */
- if (findalldevs_if.namelen)
- {
- if (findalldevs_if.namelen >= sizeof(tmpstring))
- {
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "Interface name too long");
- goto error;
- }
- /* Retrieve adapter name */
- if (rpcap_recv(sockctrl, tmpstring,
- findalldevs_if.namelen, &plen, errbuf) == -1)
- goto error;
- tmpstring[findalldevs_if.namelen] = 0;
- /* Create the new device identifier */
- if (pcap_createsrcstr(tmpstring2, PCAP_SRC_IFREMOTE, host, port, tmpstring, errbuf) == -1)
- return -1;
- stringlen = strlen(tmpstring2);
- dev->name = (char *)malloc(stringlen + 1);
- if (dev->name == NULL)
- {
- pcap_fmt_errmsg_for_errno(errbuf,
- PCAP_ERRBUF_SIZE, errno, "malloc() failed");
- goto error;
- }
- /* Copy the new device name into the correct memory location */
- strlcpy(dev->name, tmpstring2, stringlen + 1);
- }
- if (findalldevs_if.desclen)
- {
- if (findalldevs_if.desclen >= sizeof(tmpstring))
- {
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "Interface description too long");
- goto error;
- }
- /* Retrieve adapter description */
- if (rpcap_recv(sockctrl, tmpstring,
- findalldevs_if.desclen, &plen, errbuf) == -1)
- goto error;
- tmpstring[findalldevs_if.desclen] = 0;
- pcap_snprintf(tmpstring2, sizeof(tmpstring2) - 1, "%s '%s' %s %s", PCAP_TEXT_SOURCE_ADAPTER,
- tmpstring, PCAP_TEXT_SOURCE_ON_REMOTE_HOST, host);
- stringlen = strlen(tmpstring2);
- dev->description = (char *)malloc(stringlen + 1);
- if (dev->description == NULL)
- {
- pcap_fmt_errmsg_for_errno(errbuf,
- PCAP_ERRBUF_SIZE, errno, "malloc() failed");
- goto error;
- }
- /* Copy the new device description into the correct memory location */
- strlcpy(dev->description, tmpstring2, stringlen + 1);
- }
- dev->flags = ntohl(findalldevs_if.flags);
- prevaddr = NULL;
- /* loop until all addresses have been received */
- for (j = 0; j < findalldevs_if.naddr; j++)
- {
- struct rpcap_findalldevs_ifaddr ifaddr;
- /* Retrieve the interface addresses */
- if (rpcap_recv(sockctrl, (char *)&ifaddr,
- sizeof(struct rpcap_findalldevs_ifaddr),
- &plen, errbuf) == -1)
- goto error;
- /*
- * Deserialize all the address components.
- */
- addr = (struct pcap_addr *) malloc(sizeof(struct pcap_addr));
- if (addr == NULL)
- {
- pcap_fmt_errmsg_for_errno(errbuf,
- PCAP_ERRBUF_SIZE, errno, "malloc() failed");
- goto error;
- }
- addr->next = NULL;
- addr->addr = NULL;
- addr->netmask = NULL;
- addr->broadaddr = NULL;
- addr->dstaddr = NULL;
- if (rpcap_deseraddr(&ifaddr.addr,
- (struct sockaddr_storage **) &addr->addr, errbuf) == -1)
- {
- freeaddr(addr);
- goto error;
- }
- if (rpcap_deseraddr(&ifaddr.netmask,
- (struct sockaddr_storage **) &addr->netmask, errbuf) == -1)
- {
- freeaddr(addr);
- goto error;
- }
- if (rpcap_deseraddr(&ifaddr.broadaddr,
- (struct sockaddr_storage **) &addr->broadaddr, errbuf) == -1)
- {
- freeaddr(addr);
- goto error;
- }
- if (rpcap_deseraddr(&ifaddr.dstaddr,
- (struct sockaddr_storage **) &addr->dstaddr, errbuf) == -1)
- {
- freeaddr(addr);
- goto error;
- }
- if ((addr->addr == NULL) && (addr->netmask == NULL) &&
- (addr->broadaddr == NULL) && (addr->dstaddr == NULL))
- {
- /*
- * None of the addresses are IPv4 or IPv6
- * addresses, so throw this entry away.
- */
- free(addr);
- }
- else
- {
- /*
- * Add this entry to the list.
- */
- if (prevaddr == NULL)
- {
- dev->addresses = addr;
- }
- else
- {
- prevaddr->next = addr;
- }
- prevaddr = addr;
- }
- }
- }
- /* Discard the rest of the message. */
- if (rpcap_discard(sockctrl, plen, errbuf) == 1)
- goto error_nodiscard;
- /* Control connection has to be closed only in case the remote machine is in passive mode */
- if (!active)
- {
- /* DO not send RPCAP_CLOSE, since we did not open a pcap_t; no need to free resources */
- if (sock_close(sockctrl, errbuf, PCAP_ERRBUF_SIZE))
- return -1;
- }
- /* To avoid inconsistencies in the number of sock_init() */
- sock_cleanup();
- return 0;
- error:
- /*
- * In case there has been an error, I don't want to overwrite it with a new one
- * if the following call fails. I want to return always the original error.
- *
- * Take care: this connection can already be closed when we try to close it.
- * This happens because a previous error in the rpcapd, which requested to
- * closed the connection. In that case, we already recognized that into the
- * rpspck_isheaderok() and we already acknowledged the closing.
- * In that sense, this call is useless here (however it is needed in case
- * the client generates the error).
- *
- * Checks if all the data has been read; if not, discard the data in excess
- */
- (void) rpcap_discard(sockctrl, plen, NULL);
- error_nodiscard:
- /* Control connection has to be closed only in case the remote machine is in passive mode */
- if (!active)
- sock_close(sockctrl, NULL, 0);
- /* To avoid inconsistencies in the number of sock_init() */
- sock_cleanup();
- /* Free whatever interfaces we've allocated. */
- pcap_freealldevs(*alldevs);
- return -1;
- }
- /*
- * Active mode routines.
- *
- * The old libpcap API is somewhat ugly, and makes active mode difficult
- * to implement; we provide some APIs for it that work only with rpcap.
- */
- SOCKET pcap_remoteact_accept(const char *address, const char *port, const char *hostlist, char *connectinghost, struct pcap_rmtauth *auth, char *errbuf)
- {
- /* socket-related variables */
- struct addrinfo hints; /* temporary struct to keep settings needed to open the new socket */
- struct addrinfo *addrinfo; /* keeps the addrinfo chain; required to open a new socket */
- struct sockaddr_storage from; /* generic sockaddr_storage variable */
- socklen_t fromlen; /* keeps the length of the sockaddr_storage variable */
- SOCKET sockctrl; /* keeps the main socket identifier */
- uint8 protocol_version; /* negotiated protocol version */
- struct activehosts *temp, *prev; /* temp var needed to scan he host list chain */
- *connectinghost = 0; /* just in case */
- /* Prepare to open a new server socket */
- memset(&hints, 0, sizeof(struct addrinfo));
- /* WARNING Currently it supports only ONE socket family among ipv4 and IPv6 */
- hints.ai_family = AF_INET; /* PF_UNSPEC to have both IPv4 and IPv6 server */
- hints.ai_flags = AI_PASSIVE; /* Ready to a bind() socket */
- hints.ai_socktype = SOCK_STREAM;
- /* Warning: this call can be the first one called by the user. */
- /* For this reason, we have to initialize the WinSock support. */
- if (sock_init(errbuf, PCAP_ERRBUF_SIZE) == -1)
- return (SOCKET)-1;
- /* Do the work */
- if ((port == NULL) || (port[0] == 0))
- {
- if (sock_initaddress(address, RPCAP_DEFAULT_NETPORT_ACTIVE, &hints, &addrinfo, errbuf, PCAP_ERRBUF_SIZE) == -1)
- {
- SOCK_DEBUG_MESSAGE(errbuf);
- return (SOCKET)-2;
- }
- }
- else
- {
- if (sock_initaddress(address, port, &hints, &addrinfo, errbuf, PCAP_ERRBUF_SIZE) == -1)
- {
- SOCK_DEBUG_MESSAGE(errbuf);
- return (SOCKET)-2;
- }
- }
- if ((sockmain = sock_open(addrinfo, SOCKOPEN_SERVER, 1, errbuf, PCAP_ERRBUF_SIZE)) == INVALID_SOCKET)
- {
- SOCK_DEBUG_MESSAGE(errbuf);
- freeaddrinfo(addrinfo);
- return (SOCKET)-2;
- }
- freeaddrinfo(addrinfo);
- /* Connection creation */
- fromlen = sizeof(struct sockaddr_storage);
- sockctrl = accept(sockmain, (struct sockaddr *) &from, &fromlen);
- /* We're not using sock_close, since we do not want to send a shutdown */
- /* (which is not allowed on a non-connected socket) */
- closesocket(sockmain);
- sockmain = 0;
- if (sockctrl == INVALID_SOCKET)
- {
- sock_geterror("accept(): ", errbuf, PCAP_ERRBUF_SIZE);
- return (SOCKET)-2;
- }
- /* Get the numeric for of the name of the connecting host */
- if (getnameinfo((struct sockaddr *) &from, fromlen, connectinghost, RPCAP_HOSTLIST_SIZE, NULL, 0, NI_NUMERICHOST))
- {
- sock_geterror("getnameinfo(): ", errbuf, PCAP_ERRBUF_SIZE);
- rpcap_senderror(sockctrl, 0, PCAP_ERR_REMOTEACCEPT, errbuf, NULL);
- sock_close(sockctrl, NULL, 0);
- return (SOCKET)-1;
- }
- /* checks if the connecting host is among the ones allowed */
- if (sock_check_hostlist((char *)hostlist, RPCAP_HOSTLIST_SEP, &from, errbuf, PCAP_ERRBUF_SIZE) < 0)
- {
- rpcap_senderror(sockctrl, 0, PCAP_ERR_REMOTEACCEPT, errbuf, NULL);
- sock_close(sockctrl, NULL, 0);
- return (SOCKET)-1;
- }
- /*
- * Send authentication to the remote machine.
- */
- if (rpcap_doauth(sockctrl, &protocol_version, auth, errbuf) == -1)
- {
- /* Unrecoverable error. */
- rpcap_senderror(sockctrl, 0, PCAP_ERR_REMOTEACCEPT, errbuf, NULL);
- sock_close(sockctrl, NULL, 0);
- return (SOCKET)-3;
- }
- /* Checks that this host does not already have a cntrl connection in place */
- /* Initialize pointers */
- temp = activeHosts;
- prev = NULL;
- while (temp)
- {
- /* This host already has an active connection in place, so I don't have to update the host list */
- if (sock_cmpaddr(&temp->host, &from) == 0)
- return sockctrl;
- prev = temp;
- temp = temp->next;
- }
- /* The host does not exist in the list; so I have to update the list */
- if (prev)
- {
- prev->next = (struct activehosts *) malloc(sizeof(struct activehosts));
- temp = prev->next;
- }
- else
- {
- activeHosts = (struct activehosts *) malloc(sizeof(struct activehosts));
- temp = activeHosts;
- }
- if (temp == NULL)
- {
- pcap_fmt_errmsg_for_errno(errbuf, PCAP_ERRBUF_SIZE,
- errno, "malloc() failed");
- rpcap_senderror(sockctrl, protocol_version, PCAP_ERR_REMOTEACCEPT, errbuf, NULL);
- sock_close(sockctrl, NULL, 0);
- return (SOCKET)-1;
- }
- memcpy(&temp->host, &from, fromlen);
- temp->sockctrl = sockctrl;
- temp->protocol_version = protocol_version;
- temp->next = NULL;
- return sockctrl;
- }
- int pcap_remoteact_close(const char *host, char *errbuf)
- {
- struct activehosts *temp, *prev; /* temp var needed to scan the host list chain */
- struct addrinfo hints, *addrinfo, *ai_next; /* temp var needed to translate between hostname to its address */
- int retval;
- temp = activeHosts;
- prev = NULL;
- /* retrieve the network address corresponding to 'host' */
- addrinfo = NULL;
- memset(&hints, 0, sizeof(struct addrinfo));
- hints.ai_family = PF_UNSPEC;
- hints.ai_socktype = SOCK_STREAM;
- retval = getaddrinfo(host, "0", &hints, &addrinfo);
- if (retval != 0)
- {
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "getaddrinfo() %s", gai_strerror(retval));
- return -1;
- }
- while (temp)
- {
- ai_next = addrinfo;
- while (ai_next)
- {
- if (sock_cmpaddr(&temp->host, (struct sockaddr_storage *) ai_next->ai_addr) == 0)
- {
- struct rpcap_header header;
- int status = 0;
- /* Close this connection */
- rpcap_createhdr(&header, temp->protocol_version,
- RPCAP_MSG_CLOSE, 0, 0);
- /*
- * Don't check for errors, since we're
- * just cleaning up.
- */
- if (sock_send(temp->sockctrl,
- (char *)&header,
- sizeof(struct rpcap_header), errbuf,
- PCAP_ERRBUF_SIZE) < 0)
- {
- /*
- * Let that error be the one we
- * report.
- */
- (void)sock_close(temp->sockctrl, NULL,
- 0);
- status = -1;
- }
- else
- {
- if (sock_close(temp->sockctrl, errbuf,
- PCAP_ERRBUF_SIZE) == -1)
- status = -1;
- }
- /*
- * Remove the host from the list of active
- * hosts.
- */
- if (prev)
- prev->next = temp->next;
- else
- activeHosts = temp->next;
- freeaddrinfo(addrinfo);
- free(temp);
- /* To avoid inconsistencies in the number of sock_init() */
- sock_cleanup();
- return status;
- }
- ai_next = ai_next->ai_next;
- }
- prev = temp;
- temp = temp->next;
- }
- if (addrinfo)
- freeaddrinfo(addrinfo);
- /* To avoid inconsistencies in the number of sock_init() */
- sock_cleanup();
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "The host you want to close the active connection is not known");
- return -1;
- }
- void pcap_remoteact_cleanup(void)
- {
- /* Very dirty, but it works */
- if (sockmain)
- {
- closesocket(sockmain);
- /* To avoid inconsistencies in the number of sock_init() */
- sock_cleanup();
- }
- }
- int pcap_remoteact_list(char *hostlist, char sep, int size, char *errbuf)
- {
- struct activehosts *temp; /* temp var needed to scan the host list chain */
- size_t len;
- char hoststr[RPCAP_HOSTLIST_SIZE + 1];
- temp = activeHosts;
- len = 0;
- *hostlist = 0;
- while (temp)
- {
- /*int sock_getascii_addrport(const struct sockaddr_storage *sockaddr, char *address, int addrlen, char *port, int portlen, int flags, char *errbuf, int errbuflen) */
- /* Get the numeric form of the name of the connecting host */
- if (sock_getascii_addrport((struct sockaddr_storage *) &temp->host, hoststr,
- RPCAP_HOSTLIST_SIZE, NULL, 0, NI_NUMERICHOST, errbuf, PCAP_ERRBUF_SIZE) != -1)
- /* if (getnameinfo( (struct sockaddr *) &temp->host, sizeof (struct sockaddr_storage), hoststr, */
- /* RPCAP_HOSTLIST_SIZE, NULL, 0, NI_NUMERICHOST) ) */
- {
- /* sock_geterror("getnameinfo(): ", errbuf, PCAP_ERRBUF_SIZE); */
- return -1;
- }
- len = len + strlen(hoststr) + 1 /* the separator */;
- if ((size < 0) || (len >= (size_t)size))
- {
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "The string you provided is not able to keep "
- "the hostnames for all the active connections");
- return -1;
- }
- strlcat(hostlist, hoststr, PCAP_ERRBUF_SIZE);
- hostlist[len - 1] = sep;
- hostlist[len] = 0;
- temp = temp->next;
- }
- return 0;
- }
- /*
- * Receive the header of a message.
- */
- static int rpcap_recv_msg_header(SOCKET sock, struct rpcap_header *header, char *errbuf)
- {
- int nrecv;
- nrecv = sock_recv(sock, (char *) header, sizeof(struct rpcap_header),
- SOCK_RECEIVEALL_YES|SOCK_EOF_IS_ERROR, errbuf,
- PCAP_ERRBUF_SIZE);
- if (nrecv == -1)
- {
- /* Network error. */
- return -1;
- }
- header->plen = ntohl(header->plen);
- return 0;
- }
- /*
- * Make sure the protocol version of a received message is what we were
- * expecting.
- */
- static int rpcap_check_msg_ver(SOCKET sock, uint8 expected_ver, struct rpcap_header *header, char *errbuf)
- {
- /*
- * Did the server specify the version we negotiated?
- */
- if (header->ver != expected_ver)
- {
- /*
- * Discard the rest of the message.
- */
- if (rpcap_discard(sock, header->plen, errbuf) == -1)
- return -1;
- /*
- * Tell our caller that it's not the negotiated version.
- */
- if (errbuf != NULL)
- {
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE,
- "Server sent us a message with version %u when we were expecting %u",
- header->ver, expected_ver);
- }
- return -1;
- }
- return 0;
- }
- /*
- * Check the message type of a received message, which should either be
- * the expected message type or RPCAP_MSG_ERROR.
- */
- static int rpcap_check_msg_type(SOCKET sock, uint8 request_type, struct rpcap_header *header, uint16 *errcode, char *errbuf)
- {
- const char *request_type_string;
- const char *msg_type_string;
- /*
- * What type of message is it?
- */
- if (header->type == RPCAP_MSG_ERROR)
- {
- /*
- * The server reported an error.
- * Hand that error back to our caller.
- */
- *errcode = ntohs(header->value);
- rpcap_msg_err(sock, header->plen, errbuf);
- return -1;
- }
- *errcode = 0;
- /*
- * For a given request type value, the expected reply type value
- * is the request type value with ORed with RPCAP_MSG_IS_REPLY.
- */
- if (header->type != (request_type | RPCAP_MSG_IS_REPLY))
- {
- /*
- * This isn't a reply to the request we sent.
- */
- /*
- * Discard the rest of the message.
- */
- if (rpcap_discard(sock, header->plen, errbuf) == -1)
- return -1;
- /*
- * Tell our caller about it.
- */
- request_type_string = rpcap_msg_type_string(request_type);
- msg_type_string = rpcap_msg_type_string(header->type);
- if (errbuf != NULL)
- {
- if (request_type_string == NULL)
- {
- /* This should not happen. */
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE,
- "rpcap_check_msg_type called for request message with type %u",
- request_type);
- return -1;
- }
- if (msg_type_string != NULL)
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE,
- "%s message received in response to a %s message",
- msg_type_string, request_type_string);
- else
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE,
- "Message of unknown type %u message received in response to a %s request",
- header->type, request_type_string);
- }
- return -1;
- }
- return 0;
- }
- /*
- * Receive and process the header of a message.
- */
- static int rpcap_process_msg_header(SOCKET sock, uint8 expected_ver, uint8 request_type, struct rpcap_header *header, char *errbuf)
- {
- uint16 errcode;
- if (rpcap_recv_msg_header(sock, header, errbuf) == -1)
- {
- /* Network error. */
- return -1;
- }
- /*
- * Did the server specify the version we negotiated?
- */
- if (rpcap_check_msg_ver(sock, expected_ver, header, errbuf) == -1)
- return -1;
- /*
- * Check the message type.
- */
- return rpcap_check_msg_type(sock, request_type, header,
- &errcode, errbuf);
- }
- /*
- * Read data from a message.
- * If we're trying to read more data that remains, puts an error
- * message into errmsgbuf and returns -2. Otherwise, tries to read
- * the data and, if that succeeds, subtracts the amount read from
- * the number of bytes of data that remains.
- * Returns 0 on success, logs a message and returns -1 on a network
- * error.
- */
- static int rpcap_recv(SOCKET sock, void *buffer, size_t toread, uint32 *plen, char *errbuf)
- {
- int nread;
- if (toread > *plen)
- {
- /* The server sent us a bad message */
- pcap_snprintf(errbuf, PCAP_ERRBUF_SIZE, "Message payload is too short");
- return -1;
- }
- nread = sock_recv(sock, buffer, toread,
- SOCK_RECEIVEALL_YES|SOCK_EOF_IS_ERROR, errbuf, PCAP_ERRBUF_SIZE);
- if (nread == -1)
- {
- return -1;
- }
- *plen -= nread;
- return 0;
- }
- /*
- * This handles the RPCAP_MSG_ERROR message.
- */
- static void rpcap_msg_err(SOCKET sockctrl, uint32 plen, char *remote_errbuf)
- {
- char errbuf[PCAP_ERRBUF_SIZE];
- if (plen >= PCAP_ERRBUF_SIZE)
- {
- /*
- * Message is too long; just read as much of it as we
- * can into the buffer provided, and discard the rest.
- */
- if (sock_recv(sockctrl, remote_errbuf, PCAP_ERRBUF_SIZE - 1,
- SOCK_RECEIVEALL_YES|SOCK_EOF_IS_ERROR, errbuf,
- PCAP_ERRBUF_SIZE) == -1)
- {
- // Network error.
- pcap_snprintf(remote_errbuf, PCAP_ERRBUF_SIZE, "Read of error message from client failed: %s", errbuf);
- return;
- }
- /*
- * Null-terminate it.
- */
- remote_errbuf[PCAP_ERRBUF_SIZE - 1] = '\0';
- /*
- * Throw away the rest.
- */
- (void)rpcap_discard(sockctrl, plen - (PCAP_ERRBUF_SIZE - 1), remote_errbuf);
- }
- else if (plen == 0)
- {
- /* Empty error string. */
- remote_errbuf[0] = '\0';
- }
- else
- {
- if (sock_recv(sockctrl, remote_errbuf, plen,
- SOCK_RECEIVEALL_YES|SOCK_EOF_IS_ERROR, errbuf,
- PCAP_ERRBUF_SIZE) == -1)
- {
- // Network error.
- pcap_snprintf(remote_errbuf, PCAP_ERRBUF_SIZE, "Read of error message from client failed: %s", errbuf);
- return;
- }
- /*
- * Null-terminate it.
- */
- remote_errbuf[plen] = '\0';
- }
- }
- /*
- * Discard data from a connection.
- * Mostly used to discard wrong-sized messages.
- * Returns 0 on success, logs a message and returns -1 on a network
- * error.
- */
- static int rpcap_discard(SOCKET sock, uint32 len, char *errbuf)
- {
- if (len != 0)
- {
- if (sock_discard(sock, len, errbuf, PCAP_ERRBUF_SIZE) == -1)
- {
- // Network error.
- return -1;
- }
- }
- return 0;
- }
- /*
- * Read bytes into the pcap_t's buffer until we have the specified
- * number of bytes read or we get an error or interrupt indication.
- */
- static int rpcap_read_packet_msg(SOCKET sock, pcap_t *p, size_t size)
- {
- u_char *bp;
- int cc;
- int bytes_read;
- bp = p->bp;
- cc = p->cc;
- /*
- * Loop until we have the amount of data requested or we get
- * an error or interrupt.
- */
- while ((size_t)cc < size)
- {
- /*
- * We haven't read all of the packet header yet.
- * Read what remains, which could be all of it.
- */
- bytes_read = sock_recv(sock, bp, size - cc,
- SOCK_RECEIVEALL_NO|SOCK_EOF_IS_ERROR, p->errbuf,
- PCAP_ERRBUF_SIZE);
- if (bytes_read == -1)
- {
- /*
- * Network error. Update the read pointer and
- * byte count, and return an error indication.
- */
- p->bp = bp;
- p->cc = cc;
- return -1;
- }
- if (bytes_read == -3)
- {
- /*
- * Interrupted receive. Update the read
- * pointer and byte count, and return
- * an interrupted indication.
- */
- p->bp = bp;
- p->cc = cc;
- return -3;
- }
- if (bytes_read == 0)
- {
- /*
- * EOF - server terminated the connection.
- * Update the read pointer and byte count, and
- * return an error indication.
- */
- pcap_snprintf(p->errbuf, PCAP_ERRBUF_SIZE,
- "The server terminated the connection.");
- return -1;
- }
- bp += bytes_read;
- cc += bytes_read;
- }
- p->bp = bp;
- p->cc = cc;
- return 0;
- }
|