12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379 |
- /* snifftest.c
- *
- * Copyright (C) 2006-2023 wolfSSL Inc.
- *
- * This file is part of wolfSSL.
- *
- * wolfSSL is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * wolfSSL is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA
- */
- #ifdef HAVE_CONFIG_H
- #include <config.h>
- #endif
- #include <wolfssl/wolfcrypt/settings.h>
- #include <wolfssl/wolfcrypt/types.h>
- #include <wolfssl/wolfcrypt/logging.h>
- #include <wolfssl/wolfcrypt/error-crypt.h>
- #include <wolfssl/version.h>
- #ifdef WOLFSSL_SNIFFER_STORE_DATA_CB
- #include <wolfssl/wolfcrypt/memory.h>
- #endif
- #ifdef THREADED_SNIFFTEST
- #include <pthread.h>
- #endif
- /* Build Options:
- * THREADED_SNIFFTEST: Enable threaded version of the sniffer test
- */
- /* For windows tests force sniffer build option on */
- #ifdef _WIN32
- #define WOLFSSL_SNIFFER
- #endif
- #ifndef WOLFSSL_SNIFFER
- #ifndef NO_MAIN_DRIVER
- /* blank build */
- #include <stdio.h>
- #include <stdlib.h>
- int main(void)
- {
- printf("do ./configure --enable-sniffer to enable build support\n");
- return EXIT_SUCCESS;
- }
- #endif /* !NO_MAIN_DRIVER */
- #else
- /* do a full build */
- #ifdef _MSC_VER
- /* builds on *nix too, for scanf device and port */
- #define _CRT_SECURE_NO_WARNINGS
- #endif
- #include <pcap/pcap.h> /* pcap stuff */
- #include <stdio.h> /* printf */
- #include <stdlib.h> /* EXIT_SUCCESS */
- #include <string.h> /* strcmp */
- #include <signal.h> /* signal */
- #include <ctype.h> /* isprint */
- #include <wolfssl/sniffer.h>
- #ifndef _WIN32
- #include <sys/socket.h> /* AF_INET */
- #include <arpa/inet.h>
- #include <netinet/in.h>
- #endif
- typedef unsigned char byte;
- enum {
- ETHER_IF_FRAME_LEN = 14, /* ethernet interface frame length */
- NULL_IF_FRAME_LEN = 4, /* no link interface frame length */
- };
- /* A TLS record can be 16k and change. The chain is broken up into 2K chunks.
- * This covers the TLS record, plus a chunk for TCP/IP headers. */
- #ifndef CHAIN_INPUT_CHUNK_SIZE
- #define CHAIN_INPUT_CHUNK_SIZE 2048
- #elif (CHAIN_INPUT_CHUNK_SIZE < 256)
- #undef CHAIN_INPUT_CHUNK_SIZE
- #define CHAIN_INPUT_CHUNK_SIZE 256
- #elif (CHAIN_INPUT_CHUNK_SIZE > 16384)
- #undef CHAIN_INPUT_CHUNK_SIZE
- #define CHAIN_INPUT_CHUNK_SIZE 16384
- #endif
- #define CHAIN_INPUT_COUNT ((16384 / CHAIN_INPUT_CHUNK_SIZE) + 1)
- #ifndef STORE_DATA_BLOCK_SZ
- #define STORE_DATA_BLOCK_SZ 1024
- #endif
- #if defined(HAVE_ECC) && !defined(NO_ECC_SECP) && (!defined(NO_ECC256) || defined(HAVE_ALL_CURVES))
- #define DEFAULT_SERVER_EPH_KEY_ECC "../../certs/statickeys/ecc-secp256r1.pem"
- #else
- #define DEFAULT_SERVER_EPH_KEY_ECC ""
- #endif
- #ifndef NO_DH
- #define DEFAULT_SERVER_EPH_KEY_DH "../../certs/statickeys/dh-ffdhe2048.pem"
- #else
- #define DEFAULT_SERVER_EPH_KEY_DH ""
- #endif
- #ifdef HAVE_CURVE25519
- #define DEFAULT_SERVER_EPH_KEY_X25519 "../../certs/statickeys/x25519.pem"
- #else
- #define DEFAULT_SERVER_EPH_KEY_X25519 ""
- #endif
- #ifndef DEFAULT_SERVER_EPH_KEY
- #define DEFAULT_SERVER_EPH_KEY \
- DEFAULT_SERVER_EPH_KEY_ECC "," \
- DEFAULT_SERVER_EPH_KEY_DH "," \
- DEFAULT_SERVER_EPH_KEY_X25519
- #endif
- #define DEFAULT_SERVER_KEY_RSA "../../certs/server-key.pem"
- #define DEFAULT_SERVER_KEY_ECC "../../certs/ecc-key.pem"
- #ifndef DEFAULT_SERVER_KEY
- #ifndef NO_RSA
- #define DEFAULT_SERVER_KEY DEFAULT_SERVER_KEY_RSA
- #elif defined(HAVE_ECC)
- #define DEFAULT_SERVER_KEY DEFAULT_SERVER_KEY_ECC
- #endif
- #endif
- #define DEFAULT_SERVER_IP "127.0.0.1"
- #define DEFAULT_SERVER_PORT (443)
- #ifdef WOLFSSL_SNIFFER_WATCH
- static const byte rsaHash[] = {
- 0x3d, 0x4a, 0x60, 0xfc, 0xbf, 0xe5, 0x4d, 0x3e,
- 0x85, 0x62, 0xf2, 0xfc, 0xdb, 0x0d, 0x51, 0xdd,
- 0xcd, 0xc2, 0x53, 0x81, 0x1a, 0x67, 0x31, 0xa0,
- 0x7f, 0xd2, 0x11, 0x74, 0xbf, 0xea, 0xc9, 0xc5
- };
- static const byte eccHash[] = {
- 0x9e, 0x45, 0xb6, 0xf8, 0xc6, 0x5d, 0x60, 0x90,
- 0x40, 0x8f, 0xd2, 0x0e, 0xb1, 0x59, 0xe7, 0xbd,
- 0xb0, 0x9b, 0x3c, 0x7a, 0x3a, 0xbe, 0x13, 0x52,
- 0x07, 0x4f, 0x1a, 0x64, 0x45, 0xe0, 0x13, 0x34
- };
- #endif
- static pcap_t* pcap = NULL;
- static pcap_if_t* alldevs = NULL;
- static struct bpf_program pcap_fp;
- static void FreeAll(void)
- {
- if (pcap) {
- pcap_freecode(&pcap_fp);
- pcap_close(pcap);
- }
- if (alldevs)
- pcap_freealldevs(alldevs);
- #ifndef _WIN32
- ssl_FreeSniffer();
- #endif
- }
- #ifdef WOLFSSL_SNIFFER_STATS
- static void DumpStats(void)
- {
- SSLStats sslStats;
- ssl_ReadStatistics(&sslStats);
- printf("SSL Stats (sslStandardConns):%lu\n",
- sslStats.sslStandardConns);
- printf("SSL Stats (sslClientAuthConns):%lu\n",
- sslStats.sslClientAuthConns);
- printf("SSL Stats (sslResumedConns):%lu\n",
- sslStats.sslResumedConns);
- printf("SSL Stats (sslEphemeralMisses):%lu\n",
- sslStats.sslEphemeralMisses);
- printf("SSL Stats (sslResumptionInserts):%lu\n",
- sslStats.sslResumptionInserts);
- printf("SSL Stats (sslResumeMisses):%lu\n",
- sslStats.sslResumeMisses);
- printf("SSL Stats (sslCiphersUnsupported):%lu\n",
- sslStats.sslCiphersUnsupported);
- printf("SSL Stats (sslKeysUnmatched):%lu\n",
- sslStats.sslKeysUnmatched);
- printf("SSL Stats (sslKeyFails):%lu\n",
- sslStats.sslKeyFails);
- printf("SSL Stats (sslDecodeFails):%lu\n",
- sslStats.sslDecodeFails);
- printf("SSL Stats (sslAlerts):%lu\n",
- sslStats.sslAlerts);
- printf("SSL Stats (sslDecryptedBytes):%lu\n",
- sslStats.sslDecryptedBytes);
- printf("SSL Stats (sslEncryptedBytes):%lu\n",
- sslStats.sslEncryptedBytes);
- printf("SSL Stats (sslEncryptedPackets):%lu\n",
- sslStats.sslEncryptedPackets);
- printf("SSL Stats (sslDecryptedPackets):%lu\n",
- sslStats.sslDecryptedPackets);
- printf("SSL Stats (sslKeyMatches):%lu\n",
- sslStats.sslKeyMatches);
- printf("SSL Stats (sslEncryptedConns):%lu\n",
- sslStats.sslEncryptedConns);
- }
- #endif /* WOLFSSL_SNIFFER_STATS */
- static void sig_handler(const int sig)
- {
- printf("SIGINT handled = %d.\n", sig);
- FreeAll();
- #ifdef WOLFSSL_SNIFFER_STATS
- DumpStats();
- #endif
- if (sig)
- exit(EXIT_SUCCESS);
- }
- static void err_sys(const char* msg)
- {
- fprintf(stderr, "%s\n", msg);
- if (msg)
- exit(EXIT_FAILURE);
- }
- #ifdef _WIN32
- #define SNPRINTF _snprintf
- #else
- #define SNPRINTF snprintf
- #endif
- static char* iptos(const struct in_addr* addr)
- {
- static char output[32];
- byte *p = (byte*)&addr->s_addr;
- snprintf(output, sizeof(output), "%d.%d.%d.%d", p[0], p[1], p[2], p[3]);
- return output;
- }
- static const char* ip6tos(const struct in6_addr* addr)
- {
- static char output[42];
- return inet_ntop(AF_INET6, addr, output, 42);
- }
- #if defined(WOLFSSL_SNIFFER_STORE_DATA_CB) || defined(WOLFSSL_SNIFFER_CHAIN_INPUT)
- static inline unsigned int min(unsigned int a, unsigned int b)
- {
- return a > b ? b : a;
- }
- #endif
- #ifdef WOLFSSL_SNIFFER_WATCH
- static int myWatchCb(void* vSniffer,
- const unsigned char* certHash, unsigned int certHashSz,
- const unsigned char* certChain, unsigned int certChainSz,
- void* ctx, char* error)
- {
- const char* certName = NULL;
- (void)certChain;
- (void)certChainSz;
- (void)ctx;
- if (certHashSz == sizeof(rsaHash) &&
- XMEMCMP(certHash, rsaHash, certHashSz) == 0) {
- certName = DEFAULT_SERVER_KEY_RSA;
- }
- if (certHashSz == sizeof(eccHash) &&
- XMEMCMP(certHash, eccHash, certHashSz) == 0) {
- certName = DEFAULT_SERVER_KEY_ECC;
- }
- if (certName == NULL) {
- /* don't return error if key is not loaded */
- printf("Warning: No matching key found for cert hash\n");
- return 0;
- }
- return ssl_SetWatchKey_file(vSniffer, certName, FILETYPE_PEM, NULL, error);
- }
- #endif /* WOLFSSL_SNIFFER_WATCH */
- #ifdef WOLFSSL_SNIFFER_STORE_DATA_CB
- static int myStoreDataCb(const unsigned char* decryptBuf,
- unsigned int decryptBufSz, unsigned int decryptBufOffset, void* ctx)
- {
- byte** data = (byte**)ctx;
- unsigned int qty;
- if (data == NULL)
- return -1;
- if (decryptBufSz < decryptBufOffset)
- return -1;
- qty = min(decryptBufSz - decryptBufOffset, STORE_DATA_BLOCK_SZ);
- if (*data == NULL) {
- byte* tmpData;
- tmpData = (byte*)XREALLOC(*data, decryptBufSz + 1,
- NULL, DYNAMIC_TYPE_TMP_BUFFER);
- if (tmpData == NULL) {
- XFREE(*data, NULL, DYNAMIC_TYPE_TMP_BUFFER);
- *data = NULL;
- return -1;
- }
- *data = tmpData;
- }
- XMEMCPY(*data + decryptBufOffset, decryptBuf + decryptBufOffset, qty);
- return qty;
- }
- #endif /* WOLFSSL_SNIFFER_STORE_DATA_CB */
- /* try and load as both static ephemeral and private key */
- /* only fail if no key is loaded */
- /* Allow comma separated list of files */
- static int load_key(const char* name, const char* server, int port,
- const char* keyFiles, const char* passwd, char* err)
- {
- int ret = -1;
- int loadCount = 0;
- char *keyFile, *ptr = NULL;
- keyFile = XSTRTOK((char*)keyFiles, ",", &ptr);
- while (keyFile != NULL) {
- #ifdef WOLFSSL_STATIC_EPHEMERAL
- #ifdef HAVE_SNI
- ret = ssl_SetNamedEphemeralKey(name, server, port, keyFile,
- FILETYPE_PEM, passwd, err);
- #else
- ret = ssl_SetEphemeralKey(server, port, keyFile,
- FILETYPE_PEM, passwd, err);
- #endif
- if (ret == 0)
- loadCount++;
- #endif
- #ifdef HAVE_SNI
- ret = ssl_SetNamedPrivateKey(name, server, port, keyFile,
- FILETYPE_PEM, passwd, err);
- #else
- ret = ssl_SetPrivateKey(server, port, keyFile,
- FILETYPE_PEM, passwd, err);
- #endif
- if (ret == 0)
- loadCount++;
- if (loadCount == 0) {
- printf("Failed loading private key %s: ret %d\n", keyFile, ret);
- printf("Please run directly from wolfSSL root dir\n");
- ret = -1;
- }
- else {
- ret = 0;
- }
- keyFile = XSTRTOK(NULL, ",", &ptr);
- }
- (void)name;
- return ret;
- }
- static void TrimNewLine(char* str)
- {
- word32 strSz = 0;
- if (str)
- strSz = (word32)XSTRLEN(str);
- if (strSz > 0 && (str[strSz-1] == '\n' || str[strSz-1] == '\r'))
- str[strSz-1] = '\0';
- }
- static void show_appinfo(void)
- {
- printf("snifftest %s\n", LIBWOLFSSL_VERSION_STRING);
- /* list enabled sniffer features */
- printf("sniffer features: "
- #ifdef WOLFSSL_SNIFFER_STATS
- "stats, "
- #endif
- #ifdef WOLFSSL_SNIFFER_WATCH
- "watch, "
- #endif
- #ifdef WOLFSSL_SNIFFER_STORE_DATA_CB
- "store_data_cb "
- #endif
- #ifdef WOLFSSL_SNIFFER_CHAIN_INPUT
- "chain_input "
- #endif
- #ifdef WOLFSSL_SNIFFER_KEY_CALLBACK
- "key_callback "
- #endif
- #ifdef DEBUG_SNIFFER
- "debug "
- #endif
- #ifdef WOLFSSL_TLS13
- "tls_v13 "
- #endif
- #ifndef WOLFSSL_NO_TLS12
- "tls_v12 "
- #endif
- #ifdef HAVE_SESSION_TICKET
- "session_ticket "
- #endif
- #ifdef WOLFSSL_STATIC_EPHEMERAL
- "static_ephemeral "
- #endif
- #ifdef WOLFSSL_ENCRYPTED_KEYS
- "encrypted_keys "
- #endif
- #ifdef HAVE_SNI
- "sni "
- #endif
- #ifdef HAVE_EXTENDED_MASTER
- "extended_master "
- #endif
- #ifdef HAVE_MAX_FRAGMENT
- "max fragment "
- #endif
- #ifdef WOLFSSL_ASYNC_CRYPT
- "async_crypt "
- #endif
- #ifndef NO_RSA
- "rsa "
- #endif
- #if !defined(NO_DH) && defined(WOLFSSL_DH_EXTRA)
- "dh "
- #endif
- #ifdef HAVE_ECC
- "ecc "
- #endif
- #ifdef HAVE_CURVE448
- "x448 "
- #endif
- #ifdef HAVE_CURVE22519
- "x22519 "
- #endif
- #ifdef WOLFSSL_STATIC_RSA
- "rsa_static "
- #endif
- #ifdef WOLFSSL_STATIC_DH
- "dh_static "
- #endif
- #ifdef WOLFSSL_SNIFFER_KEYLOGFILE
- "ssl_keylog_file "
- #endif /* WOLFSSL_SNIFFER_KEYLOGFILE */
- "\n\n"
- );
- }
- typedef struct SnifferPacket {
- byte* packet;
- int length;
- int lastRet;
- int packetNumber;
- #ifdef THREADED_SNIFFTEST
- struct SnifferPacket* next;
- struct SnifferPacket* prev;
- int placeholder;
- #endif
- } SnifferPacket;
- #ifdef WOLFSSL_ASYNC_CRYPT
- static THREAD_LS_T SnifferPacket asyncQueue[WOLF_ASYNC_MAX_PENDING];
- /* returns index to queue */
- static int SnifferAsyncQueueAdd(int lastRet, void* chain, int chainSz,
- int isChain, int packetNumber)
- {
- int ret, i, length;
- byte* packet;
- #ifdef WOLFSSL_SNIFFER_CHAIN_INPUT
- if (isChain) {
- struct iovec* vchain = (struct iovec*)chain;
- length = 0;
- for (i = 0; i < chainSz; i++)
- length += vchain[i].iov_len;
- packet = (byte*)vchain[0].iov_base;
- }
- else
- #endif
- {
- packet = (byte*)chain;
- length = chainSz;
- }
- /* find first free idx */
- ret = MEMORY_E;
- for (i=0; i<WOLF_ASYNC_MAX_PENDING; i++) {
- if (asyncQueue[i].packet == NULL) {
- if (ret == MEMORY_E) {
- ret = i;
- break;
- }
- }
- }
- if (ret != MEMORY_E) {
- asyncQueue[ret].packet = (byte*)XMALLOC(length, NULL,
- DYNAMIC_TYPE_TMP_BUFFER);
- if (asyncQueue[ret].packet == NULL) {
- return MEMORY_E;
- }
- XMEMCPY(asyncQueue[ret].packet, packet, length);
- asyncQueue[ret].length = length;
- asyncQueue[ret].lastRet = lastRet;
- asyncQueue[ret].packetNumber = packetNumber;
- }
- (void)isChain;
- return ret;
- }
- static int SnifferAsyncPollQueue(byte** data, char* err, SSLInfo* sslInfo,
- int* queueSz)
- {
- int ret = 0, i;
- WOLF_EVENT* events[WOLF_ASYNC_MAX_PENDING];
- int eventCount = 0;
- /* try to process existing items in queue */
- for (i=0; i<WOLF_ASYNC_MAX_PENDING; i++) {
- if (asyncQueue[i].packet != NULL) {
- (*queueSz)++;
- /* do poll for events on hardware */
- ret = ssl_PollSniffer(events, WOLF_ASYNC_MAX_PENDING,
- WOLF_POLL_FLAG_CHECK_HW, &eventCount);
- if (ret == 0) {
- /* attempt to reprocess pending packet */
- #ifdef DEBUG_SNIFFER
- printf("Packet Number: %d (retry)\n", asyncQueue[i].packetNumber);
- #endif
- ret = ssl_DecodePacketAsync(asyncQueue[i].packet,
- asyncQueue[i].length, 0, data, err, sslInfo, NULL);
- asyncQueue[i].lastRet = ret;
- if (ret != WC_PENDING_E) {
- if (ret < 0) {
- printf("ssl_Decode ret = %d, %s on packet number %d\n",
- ret, err, asyncQueue[i].packetNumber);
- }
- /* done, so free and break to process below */
- XFREE(asyncQueue[i].packet, NULL, DYNAMIC_TYPE_TMP_BUFFER);
- asyncQueue[i].packet = NULL;
- if (ret != 0) {
- /* decrypted some data or found error, so return */
- break;
- }
- }
- }
- }
- }
- if (ret == WC_PENDING_E) {
- ret = 0; /* nothing new */
- }
- return ret;
- }
- #endif /* WOLFSSL_ASYNC_CRYPT */
- #ifdef THREADED_SNIFFTEST
- typedef struct {
- volatile int lockCount;
- pthread_mutex_t mutex;
- pthread_cond_t cond;
- } wm_Sem;
- /* Posix style semaphore */
- static int wm_SemInit(wm_Sem *s)
- {
- s->lockCount = 0;
- pthread_mutex_init(&s->mutex, NULL);
- pthread_cond_init(&s->cond, NULL);
- return 0;
- }
- static int wm_SemFree(wm_Sem *s)
- {
- pthread_mutex_destroy(&s->mutex);
- pthread_cond_destroy(&s->cond);
- return 0;
- }
- static int wm_SemLock(wm_Sem *s)
- {
- pthread_mutex_lock(&s->mutex);
- while (s->lockCount > 0)
- pthread_cond_wait(&s->cond, &s->mutex);
- s->lockCount++;
- pthread_mutex_unlock(&s->mutex);
- return 0;
- }
- static int wm_SemUnlock(wm_Sem *s)
- {
- pthread_mutex_lock(&s->mutex);
- s->lockCount--;
- pthread_cond_signal(&s->cond);
- pthread_mutex_unlock(&s->mutex);
- return 0;
- }
- typedef struct SnifferWorker {
- SnifferPacket *head; /* head for doubly-linked list of sniffer packets */
- SnifferPacket *tail; /* tail for doubly-linked list of sniffer packets */
- wm_Sem sem;
- pthread_t tid;
- char *server;
- char *keyFilesSrc;
- char *passwd;
- int port;
- int hadBadPacket; /* track if sniffer worker saw bad packet */
- int unused;
- int id;
- int shutdown;
- } SnifferWorker;
- static int ssl_Init_SnifferWorker(SnifferWorker* worker, int port,
- const char* server, const char* keyFilesSrc, const char* passwd, int id)
- {
- wm_SemInit(&worker->sem);
- worker->server = (char*)server;
- worker->keyFilesSrc = (char*)keyFilesSrc;
- worker->passwd = (char*)passwd;
- worker->port = port;
- worker->unused = 0;
- worker->shutdown = 0;
- worker ->id = id;
- worker->head = (SnifferPacket*)XMALLOC(sizeof(SnifferPacket), NULL,
- DYNAMIC_TYPE_TMP_BUFFER);
- if (worker->head == NULL) {
- return MEMORY_E;
- }
- XMEMSET(worker->head, 0, sizeof(SnifferPacket));
- worker->tail = worker->head;
- worker->head->packet = NULL;
- worker->head->next = NULL;
- worker->head->prev = NULL;
- worker->head->placeholder = 1;
- return 0;
- }
- static void ssl_Free_SnifferWorker(SnifferWorker* worker)
- {
- wm_SemFree(&worker->sem);
- if (worker->head) {
- XFREE(worker->head, NULL, DYNAMIC_TYPE_TMP_BUFFER);
- worker->head = NULL;
- }
- }
- static int SnifferWorkerPacketAdd(SnifferWorker* worker, int lastRet,
- byte* packet, int length, int packetNumber)
- {
- SnifferPacket* newEntry;
- newEntry = (SnifferPacket*)XMALLOC(sizeof(SnifferPacket), NULL,
- DYNAMIC_TYPE_TMP_BUFFER);
- if (newEntry == NULL) {
- return MEMORY_E;
- }
- XMEMSET(newEntry, 0, sizeof(SnifferPacket));
- newEntry->packet = (byte*)XMALLOC(length, NULL, DYNAMIC_TYPE_TMP_BUFFER);
- if (newEntry->packet == NULL) {
- XFREE(newEntry, NULL, DYNAMIC_TYPE_TMP_BUFFER);
- return MEMORY_E;
- }
- /* Set newEntry fields to input values */
- XMEMCPY(newEntry->packet, packet, length);
- newEntry->length = length;
- newEntry->lastRet = lastRet;
- newEntry->packetNumber = packetNumber;
- newEntry->placeholder = 0;
- /* Create worker head if null */
- if (worker->head == NULL) {
- worker->head = (SnifferPacket*)XMALLOC(sizeof(SnifferPacket), NULL,
- DYNAMIC_TYPE_TMP_BUFFER);
- XMEMSET(worker->head, 0, sizeof(SnifferPacket));
- worker->tail = worker->head;
- worker->head->packet = NULL;
- worker->head->next = NULL;
- worker->head->prev = NULL;
- worker->head->placeholder = 1;
- }
- if (worker->head->placeholder) {
- /* First packet added to be to SnifferWorker linked list,
- * set head and tail to the new packet */
- XFREE(worker->head, NULL, DYNAMIC_TYPE_TMP_BUFFER);
- newEntry->next = NULL;
- newEntry->prev = NULL;
- worker->head = newEntry;
- worker->tail = newEntry;
- }
- else {
- /* Add packet to SnifferWorker linked list and move tail */
- newEntry->prev = worker->tail;
- newEntry->next = NULL;
- worker->tail->next = newEntry;
- worker->tail = newEntry;
- }
- return 0;
- }
- #endif /* THREADED_SNIFFTEST */
- static int DecodePacket(byte* packet, int length, int packetNumber, char err[])
- {
- int ret, j;
- int hadBadPacket = 0;
- int isChain = 0;
- int chainSz;
- void* chain;
- byte* data = NULL; /* pointer to decrypted data */
- SSLInfo sslInfo;
- #ifdef WOLFSSL_SNIFFER_CHAIN_INPUT
- struct iovec chains[CHAIN_INPUT_COUNT];
- unsigned int remainder;
- j = 0;
- chainSz = 0;
- isChain = 1;
- remainder = length;
- do {
- unsigned int chunkSz = min(remainder, CHAIN_INPUT_CHUNK_SIZE);
- chains[chainSz].iov_base = (void*)(packet + j);
- chains[chainSz].iov_len = chunkSz;
- j += chunkSz;
- remainder -= chunkSz;
- chainSz++;
- } while (j < (int)length);
- chain = (void*)chains;
- #else
- chain = (void*)packet;
- chainSz = length;
- #endif
- #if defined(DEBUG_SNIFFER)
- printf("Packet Number: %d\n", packetNumber);
- #endif
- /* decode packet */
- #ifdef WOLFSSL_ASYNC_CRYPT
- /* For async call the original API again with same data,
- * or call with different sessions for multiple concurrent
- * stream processing */
- ret = ssl_DecodePacketAsync(chain, chainSz, isChain, &data, err,
- &sslInfo, NULL);
- /* WC_PENDING_E: Hardware is processing or stream is blocked
- * (waiting on WC_PENDING_E) */
- if (ret == WC_PENDING_E) {
- /* add to queue, for later processing */
- #ifdef DEBUG_SNIFFER
- printf("Steam is pending, queue packet %d\n", packetNumber);
- #endif
- ret = SnifferAsyncQueueAdd(ret, chain, chainSz, isChain,
- packetNumber);
- if (ret >= 0) {
- ret = 0; /* mark event just added */
- }
- }
- #elif defined(WOLFSSL_SNIFFER_CHAIN_INPUT) && \
- defined(WOLFSSL_SNIFFER_STORE_DATA_CB)
- ret = ssl_DecodePacketWithChainSessionInfoStoreData(chain, chainSz,
- &data, &sslInfo, err);
- #elif defined(WOLFSSL_SNIFFER_CHAIN_INPUT)
- (void)sslInfo;
- ret = ssl_DecodePacketWithChain(chain, chainSz, &data, err);
- #elif defined(WOLFSSL_SNIFFER_STORE_DATA_CB)
- ret = ssl_DecodePacketWithSessionInfoStoreData(packet,
- length, &data, &sslInfo, err);
- #else
- ret = ssl_DecodePacketWithSessionInfo(packet, length, &data,
- &sslInfo, err);
- #endif
- if (ret < 0) {
- printf("ssl_Decode ret = %d, %s on packet number %d\n", ret, err,
- packetNumber);
- hadBadPacket = 1;
- }
- if (data != NULL && ret > 0) {
- /* Convert non-printable data to periods. */
- for (j = 0; j < ret; j++) {
- if (isprint(data[j]) || isspace(data[j])) continue;
- data[j] = '.';
- }
- data[ret] = 0;
- printf("SSL App Data(%d:%d):%s\n", packetNumber, ret, data);
- ssl_FreeZeroDecodeBuffer(&data, ret, err);
- }
- (void)isChain;
- (void)chain;
- (void)chainSz;
- return hadBadPacket;
- }
- #ifdef THREADED_SNIFFTEST
- static void* snifferWorker(void* arg)
- {
- SnifferWorker* worker = (SnifferWorker*)arg;
- char err[PCAP_ERRBUF_SIZE];
- ssl_InitSniffer_ex2(worker->id);
- ssl_Trace("./tracefile.txt", err);
- ssl_EnableRecovery(1, -1, err);
- #ifdef WOLFSSL_SNIFFER_WATCH
- ssl_SetWatchKeyCallback(myWatchCb, err);
- #endif
- #ifdef WOLFSSL_SNIFFER_STORE_DATA_CB
- ssl_SetStoreDataCallback(myStoreDataCb);
- #endif
- load_key(NULL, worker->server, worker->port, worker->keyFilesSrc,
- worker->passwd, err);
- /* continue processing the workers packets and keep expecting them
- * until the shutdown flag is set */
- while (!worker->shutdown) {
- while (worker->head) {
- int ret = 0;
- byte* packet;
- int length;
- int packetNumber;
- #ifdef WOLFSSL_ASYNC_CRYPT
- SSLInfo sslInfo;
- byte* data;
- int queueSz = 0;
- /* poll hardware and attempt to process items in queue. If
- * returns > 0 then data pointer has decrypted something */
- SnifferAsyncPollQueue(&data, err, &sslInfo, &queueSz);
- if (queueSz >= WOLF_ASYNC_MAX_PENDING) {
- /* queue full, poll again */
- continue;
- }
- #endif
- /* Shutdown worker if it was not utilized */
- if (worker->unused) {
- XFREE(worker->head, NULL, DYNAMIC_TYPE_TMP_BUFFER);
- worker->head = NULL;
- break;
- }
- /* get lock */
- wm_SemLock(&worker->sem);
- /* get packet for current worker head */
- packet = worker->head->packet;
- length = worker->head->length;
- packetNumber = worker->head->packetNumber;
- wm_SemUnlock(&worker->sem);
- if (packet == NULL) {
- continue;
- }
- /* Decode Packet, ret value will indicate whether a
- * bad packet was encountered */
- ret = DecodePacket(packet, length, packetNumber, err);
- if (ret) {
- worker->hadBadPacket = 1;
- }
- /* get lock */
- wm_SemLock(&worker->sem);
- XFREE(worker->head->packet, NULL, DYNAMIC_TYPE_TMP_BUFFER);
- worker->head->packet = NULL;
- if (worker->head->next) {
- /* Move head and free */
- worker->head = worker->head->next;
- XFREE(worker->head->prev, NULL, DYNAMIC_TYPE_TMP_BUFFER);
- worker->head->prev = NULL;
- }
- else {
- /* No other packets in list. Keep looping until more packets
- * arrive or worker is shutdown. */
- XFREE(worker->head, NULL, DYNAMIC_TYPE_TMP_BUFFER);
- worker->head = NULL;
- }
- wm_SemUnlock(&worker->sem);
- } /* while (worker->head) */
- if (worker->unused) {
- break;
- }
- } /* while (worker->head) */
- /* Thread cleanup */
- ssl_FreeSniffer();
- #if defined(HAVE_ECC) && defined(FP_ECC)
- wc_ecc_fp_free();
- #endif
- return NULL;
- }
- #endif /* THREADED_SNIFFTEST */
- int main(int argc, char** argv)
- {
- int ret = 0;
- int hadBadPacket = 0;
- int inum = 0;
- int saveFile = 0;
- int i = 0, defDev = 0;
- int packetNumber = 0;
- int frame = ETHER_IF_FRAME_LEN;
- char err[PCAP_ERRBUF_SIZE];
- char filter[32];
- const char *keyFilesSrc = NULL;
- #ifdef WOLFSSL_SNIFFER_KEYLOGFILE
- const char *sslKeyLogFile = NULL;
- #endif /* WOLFSSL_SNIFFER_KEYLOGFILE */
- char keyFilesBuf[MAX_FILENAME_SZ];
- char keyFilesUser[MAX_FILENAME_SZ];
- const char *server = DEFAULT_SERVER_IP;
- int port = DEFAULT_SERVER_PORT;
- const char *sniName = NULL;
- const char *passwd = NULL;
- pcap_if_t *d;
- pcap_addr_t *a;
- #ifdef THREADED_SNIFFTEST
- int workerThreadCount;
- #ifdef HAVE_SESSION_TICKET
- /* Multiple threads on resume not yet supported */
- workerThreadCount = 1;
- #else
- workerThreadCount = 5;
- #endif
- #endif
- show_appinfo();
- signal(SIGINT, sig_handler);
- #ifndef THREADED_SNIFFTEST
- #ifndef _WIN32
- ssl_InitSniffer(); /* dll load on Windows */
- #endif
- ssl_Trace("./tracefile.txt", err);
- ssl_EnableRecovery(1, -1, err);
- #ifdef WOLFSSL_SNIFFER_WATCH
- ssl_SetWatchKeyCallback(myWatchCb, err);
- #endif
- #ifdef WOLFSSL_SNIFFER_STORE_DATA_CB
- ssl_SetStoreDataCallback(myStoreDataCb);
- #endif
- #endif
- if (argc == 1) {
- char cmdLineArg[128];
- /* normal case, user chooses device and port */
- if (pcap_findalldevs(&alldevs, err) == -1)
- err_sys("Error in pcap_findalldevs");
- for (d = alldevs; d; d=d->next) {
- printf("%d. %s", ++i, d->name);
- if (strcmp(d->name, "lo0") == 0) {
- defDev = i;
- }
- if (d->description)
- printf(" (%s)\n", d->description);
- else
- printf(" (No description available)\n");
- }
- if (i == 0)
- err_sys("No interfaces found! Make sure pcap or WinPcap is"
- " installed correctly and you have sufficient permissions");
- printf("Enter the interface number (1-%d) [default: %d]: ", i, defDev);
- XMEMSET(cmdLineArg, 0, sizeof(cmdLineArg));
- if (XFGETS(cmdLineArg, sizeof(cmdLineArg), stdin))
- inum = XATOI(cmdLineArg);
- if (inum == 0)
- inum = defDev;
- else if (inum < 1 || inum > i)
- err_sys("Interface number out of range");
- /* Jump to the selected adapter */
- for (d = alldevs, i = 0; i < inum - 1; d = d->next, i++);
- pcap = pcap_create(d->name, err);
- if (pcap == NULL) printf("pcap_create failed %s\n", err);
- /* print out addresses for selected interface */
- for (a = d->addresses; a; a = a->next) {
- if (a->addr->sa_family == AF_INET) {
- server =
- iptos(&((struct sockaddr_in *)a->addr)->sin_addr);
- printf("server = %s\n", server);
- }
- else if (a->addr->sa_family == AF_INET6) {
- server =
- ip6tos(&((struct sockaddr_in6 *)a->addr)->sin6_addr);
- printf("server = %s\n", server);
- }
- }
- if (server == NULL)
- err_sys("Unable to get device IPv4 or IPv6 address");
- ret = pcap_set_snaplen(pcap, 65536);
- if (ret != 0) printf("pcap_set_snaplen failed %s\n", pcap_geterr(pcap));
- ret = pcap_set_timeout(pcap, 1000);
- if (ret != 0) printf("pcap_set_timeout failed %s\n", pcap_geterr(pcap));
- ret = pcap_set_buffer_size(pcap, 1000000);
- if (ret != 0)
- printf("pcap_set_buffer_size failed %s\n", pcap_geterr(pcap));
- ret = pcap_set_promisc(pcap, 1);
- if (ret != 0) printf("pcap_set_promisc failed %s\n", pcap_geterr(pcap));
- ret = pcap_activate(pcap);
- if (ret != 0) printf("pcap_activate failed %s\n", pcap_geterr(pcap));
- printf("Enter the port to scan [default: 11111]: ");
- XMEMSET(cmdLineArg, 0, sizeof(cmdLineArg));
- if (XFGETS(cmdLineArg, sizeof(cmdLineArg), stdin)) {
- port = XATOI(cmdLineArg);
- }
- if (port <= 0)
- port = 11111;
- SNPRINTF(filter, sizeof(filter), "tcp and port %d", port);
- ret = pcap_compile(pcap, &pcap_fp, filter, 0, 0);
- if (ret != 0) printf("pcap_compile failed %s\n", pcap_geterr(pcap));
- ret = pcap_setfilter(pcap, &pcap_fp);
- if (ret != 0) printf("pcap_setfilter failed %s\n", pcap_geterr(pcap));
- /* optionally enter the private key to use */
- #if defined(WOLFSSL_STATIC_EPHEMERAL) && defined(DEFAULT_SERVER_EPH_KEY)
- keyFilesSrc = DEFAULT_SERVER_EPH_KEY;
- #else
- keyFilesSrc = DEFAULT_SERVER_KEY;
- #endif
- printf("Enter the server key [default: %s]: ", keyFilesSrc);
- XMEMSET(keyFilesBuf, 0, sizeof(keyFilesBuf));
- XMEMSET(keyFilesUser, 0, sizeof(keyFilesUser));
- if (XFGETS(keyFilesUser, sizeof(keyFilesUser), stdin)) {
- TrimNewLine(keyFilesUser);
- if (XSTRLEN(keyFilesUser) > 0) {
- keyFilesSrc = keyFilesUser;
- }
- }
- XSTRNCPY(keyFilesBuf, keyFilesSrc, sizeof(keyFilesBuf));
- /* optionally enter a named key (SNI) */
- #if !defined(WOLFSSL_SNIFFER_WATCH) && defined(HAVE_SNI)
- printf("Enter alternate SNI [default: none]: ");
- XMEMSET(cmdLineArg, 0, sizeof(cmdLineArg));
- if (XFGETS(cmdLineArg, sizeof(cmdLineArg), stdin)) {
- TrimNewLine(cmdLineArg);
- if (XSTRLEN(cmdLineArg) > 0) {
- sniName = cmdLineArg;
- }
- }
- #endif /* !WOLFSSL_SNIFFER_WATCH && HAVE_SNI */
- /* get IPv4 or IPv6 addresses for selected interface */
- for (a = d->addresses; a; a = a->next) {
- server = NULL;
- if (a->addr->sa_family == AF_INET) {
- server =
- iptos(&((struct sockaddr_in *)a->addr)->sin_addr);
- }
- else if (a->addr->sa_family == AF_INET6) {
- server =
- ip6tos(&((struct sockaddr_in6 *)a->addr)->sin6_addr);
- }
- if (server) {
- XSTRNCPY(keyFilesBuf, keyFilesSrc, sizeof(keyFilesBuf));
- ret = load_key(sniName, server, port, keyFilesBuf, NULL, err);
- if (ret != 0) {
- exit(EXIT_FAILURE);
- }
- }
- }
- }
- else {
- char *pcapFile = NULL;
- for (i = 1; i < argc; i++) {
- if (strcmp(argv[i], "-pcap") == 0 && i + 1 < argc) {
- pcapFile = argv[++i];
- }
- else if (strcmp(argv[i], "-key") == 0 && i + 1 < argc) {
- keyFilesSrc = argv[++i];
- }
- else if (strcmp(argv[i], "-server") == 0 && i + 1 < argc) {
- server = argv[++i];
- }
- else if (strcmp(argv[i], "-port") == 0 && i + 1 < argc) {
- port = XATOI(argv[++i]);
- }
- else if (strcmp(argv[i], "-password") == 0 && i + 1 < argc) {
- passwd = argv[++i];
- }
- #if defined(WOLFSSL_SNIFFER_KEYLOGFILE)
- else if (strcmp(argv[i], "-keylogfile") == 0 && i + 1 < argc) {
- sslKeyLogFile = argv[++i];
- }
- #endif /* WOLFSSL_SNIFFER_KEYLOGFILE */
- #if defined(THREADED_SNIFFTEST)
- else if (strcmp(argv[i], "-threads") == 0 && i + 1 < argc) {
- workerThreadCount = XATOI(argv[++i]);
- }
- #endif /* THREADED_SNIFFTEST */
- else {
- fprintf(stderr, "Invalid option or missing argument: %s\n", argv[i]);
- fprintf(stderr, "Usage: %s -pcap pcap_arg -key key_arg"
- " [-password password_arg] [-server server_arg] [-port port_arg]"
- #if defined(WOLFSSL_SNIFFER_KEYLOGFILE)
- " [-keylogfile keylogfile_arg]"
- #endif /* WOLFSSL_SNIFFER_KEYLOGFILE */
- #if defined(THREADED_SNIFFTEST)
- " [-threads threads_arg]"
- #endif /* THREADED_SNIFFTEST */
- "\n", argv[0]);
- exit(EXIT_FAILURE);
- }
- }
- if (!pcapFile) {
- fprintf(stderr, "Error: -pcap option is required.\n");
- exit(EXIT_FAILURE);
- }
- #if defined(WOLFSSL_SNIFFER_KEYLOGFILE)
- /* If we offer keylog support, then user must provide EITHER a pubkey
- * OR a keylog file but NOT both */
- if ((!keyFilesSrc && !sslKeyLogFile) || (keyFilesSrc && sslKeyLogFile)) {
- fprintf(stderr, "Error: either -key OR -keylogfile option required but NOT both.\n");
- exit(EXIT_FAILURE);
- }
- #else
- if (!keyFilesSrc) {
- fprintf(stderr, "Error: -key option is required.\n");
- exit(EXIT_FAILURE);
- }
- #endif
- saveFile = 1;
- pcap = pcap_open_offline(pcapFile , err);
- if (pcap == NULL) {
- fprintf(stderr, "pcap_open_offline failed %s\n", err);
- err_sys(err);
- }
- else {
- #if defined(WOLFSSL_SNIFFER_KEYLOGFILE)
- if (sslKeyLogFile != NULL) {
- ret = ssl_LoadSecretsFromKeyLogFile(sslKeyLogFile, err);
- if (ret != 0) {
- fprintf(stderr, "ERROR=%d, unable to load secrets from keylog file\n",ret);
- err_sys(err);
- }
- ret = ssl_CreateKeyLogSnifferServer(server, port, err);
- if (ret != 0) {
- fprintf(stderr, "ERROR=%d, unable to create keylog sniffer server\n",ret);
- err_sys(err);
- }
- }
- else
- #endif /* WOLFSSL_SNIFFER_KEYLOGFILE */
- {
- ret = load_key(NULL, server, port, keyFilesSrc, passwd, err);
- if (ret != 0) {
- fprintf(stderr, "Failed to load key\n");
- err_sys(err);
- }
- }
- /* Only let through TCP/IP packets */
- ret = pcap_compile(pcap, &pcap_fp, "(ip6 or ip) and tcp", 0, 0);
- if (ret != 0) {
- fprintf(stderr, "pcap_compile failed %s\n", pcap_geterr(pcap));
- exit(EXIT_FAILURE);
- }
- ret = pcap_setfilter(pcap, &pcap_fp);
- if (ret != 0) {
- fprintf(stderr, "pcap_setfilter failed %s\n", pcap_geterr(pcap));
- exit(EXIT_FAILURE);
- }
- }
- }
- if (ret != 0)
- err_sys(err);
- if (pcap_datalink(pcap) == DLT_NULL)
- frame = NULL_IF_FRAME_LEN;
- #ifdef THREADED_SNIFFTEST
- SnifferWorker workers[workerThreadCount];
- int used[workerThreadCount];
- XMEMSET(used, 0, sizeof(used));
- XMEMSET(&workers, 0, sizeof(workers));
- for (i=0; i<workerThreadCount; i++) {
- ssl_Init_SnifferWorker(&workers[i], port, server, keyFilesSrc,
- passwd, i);
- pthread_create(&workers[i].tid, NULL, snifferWorker, &workers[i]);
- }
- #endif
- while (1) {
- struct pcap_pkthdr header;
- const unsigned char* packet = NULL;
- byte* data = NULL; /* pointer to decrypted data */
- #ifdef THREADED_SNIFFTEST
- SnifferStreamInfo info;
- uint8_t infoSum;
- uint8_t* infoPtr;
- int threadNum;
- #endif
- #if defined(WOLFSSL_ASYNC_CRYPT)
- SSLInfo sslInfo;
- int queueSz = 0;
- XMEMSET(&sslInfo, 0, sizeof(sslInfo));
- /* poll hardware and attempt to process items in queue. If returns > 0
- * then data pointer has decrypted something */
- SnifferAsyncPollQueue(&data, err, &sslInfo, &queueSz);
- if (queueSz >= WOLF_ASYNC_MAX_PENDING) {
- /* queue full, poll again */
- continue;
- }
- #endif
- ret = 0; /* reset status */
- if (data == NULL) {
- /* grab next pcap packet */
- packetNumber++;
- packet = pcap_next(pcap, &header);
- }
- if (packet) {
- if (header.caplen > 40) { /* min ip(20) + min tcp(20) */
- packet += frame;
- header.caplen -= frame;
- }
- else {
- /* packet doesn't contain minimum ip/tcp header */
- continue;
- }
- #ifdef THREADED_SNIFFTEST
- XMEMSET(&info, 0, sizeof(SnifferStreamInfo));
- ret = ssl_DecodePacket_GetStream(&info, packet, header.caplen, err);
- /* calculate SnifferStreamInfo checksum */
- infoSum = 0;
- infoPtr = (uint8_t*)&info;
- for (i=0; i<(int)sizeof(SnifferStreamInfo); i++) {
- infoSum += infoPtr[i];
- }
- /* determine thread to handle stream */
- threadNum = infoSum % workerThreadCount;
- used[threadNum] = 1;
- #ifdef DEBUG_SNIFFER
- printf("Sending packet %d to thread number %d\n", packetNumber,
- threadNum);
- #endif
- /* get lock on thread mutex */
- wm_SemLock(&workers[threadNum].sem);
- /* add the packet to the worker's linked list */
- if (SnifferWorkerPacketAdd(&workers[threadNum], ret, (byte*)packet,
- header.caplen, packetNumber)) {
- printf("Unable to add packet %d to worker", packetNumber);
- break;
- }
- wm_SemUnlock(&workers[threadNum].sem);
- #else
- /* Decode Packet, ret value will indicate whether a
- * bad packet was encountered */
- hadBadPacket = DecodePacket((byte*)packet, header.caplen,
- packetNumber,err);
- #endif
- }
- /* check if we are done reading file */
- if (packet == NULL && data == NULL && saveFile) {
- #ifdef WOLFSSL_ASYNC_CRYPT
- /* if items pending still then keep processing */
- if (queueSz > 0)
- continue;
- #endif
- break;
- }
- }
- #ifdef THREADED_SNIFFTEST
- for (i=0; i<workerThreadCount; i++) {
- workers[i].shutdown = 1;
- if (used[i] == 0)
- workers[i].unused = 1;
- pthread_join(workers[i].tid, NULL);
- }
- for (i=0; i<workerThreadCount; i++) {
- if (workers[i].hadBadPacket) {
- hadBadPacket = 1;
- }
- ssl_Free_SnifferWorker(&workers[i]);
- }
- #endif
- FreeAll();
- return hadBadPacket ? EXIT_FAILURE : EXIT_SUCCESS;
- }
- #endif /* full build */
|