12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977 |
- /* socks5.c -- generated by Trunnel v1.5.3.
- * https://gitweb.torproject.org/trunnel.git
- * You probably shouldn't edit this file.
- */
- #include <stdlib.h>
- #include "trunnel-impl.h"
- #include "socks5.h"
- #define TRUNNEL_SET_ERROR_CODE(obj) \
- do { \
- (obj)->trunnel_error_code_ = 1; \
- } while (0)
- #if defined(__COVERITY__) || defined(__clang_analyzer__)
- /* If we're running a static analysis tool, we don't want it to complain
- * that some of our remaining-bytes checks are dead-code. */
- int socks_deadcode_dummy__ = 0;
- #define OR_DEADCODE_DUMMY || socks_deadcode_dummy__
- #else
- #define OR_DEADCODE_DUMMY
- #endif
- #define CHECK_REMAINING(nbytes, label) \
- do { \
- if (remaining < (nbytes) OR_DEADCODE_DUMMY) { \
- goto label; \
- } \
- } while (0)
- domainname_t *
- domainname_new(void)
- {
- domainname_t *val = trunnel_calloc(1, sizeof(domainname_t));
- if (NULL == val)
- return NULL;
- return val;
- }
- /** Release all storage held inside 'obj', but do not free 'obj'.
- */
- static void
- domainname_clear(domainname_t *obj)
- {
- (void) obj;
- TRUNNEL_DYNARRAY_WIPE(&obj->name);
- TRUNNEL_DYNARRAY_CLEAR(&obj->name);
- }
- void
- domainname_free(domainname_t *obj)
- {
- if (obj == NULL)
- return;
- domainname_clear(obj);
- trunnel_memwipe(obj, sizeof(domainname_t));
- trunnel_free_(obj);
- }
- uint8_t
- domainname_get_len(const domainname_t *inp)
- {
- return inp->len;
- }
- int
- domainname_set_len(domainname_t *inp, uint8_t val)
- {
- inp->len = val;
- return 0;
- }
- size_t
- domainname_getlen_name(const domainname_t *inp)
- {
- return TRUNNEL_DYNARRAY_LEN(&inp->name);
- }
- char
- domainname_get_name(domainname_t *inp, size_t idx)
- {
- return TRUNNEL_DYNARRAY_GET(&inp->name, idx);
- }
- char
- domainname_getconst_name(const domainname_t *inp, size_t idx)
- {
- return domainname_get_name((domainname_t*)inp, idx);
- }
- int
- domainname_set_name(domainname_t *inp, size_t idx, char elt)
- {
- TRUNNEL_DYNARRAY_SET(&inp->name, idx, elt);
- return 0;
- }
- int
- domainname_add_name(domainname_t *inp, char elt)
- {
- #if SIZE_MAX >= UINT8_MAX
- if (inp->name.n_ == UINT8_MAX)
- goto trunnel_alloc_failed;
- #endif
- TRUNNEL_DYNARRAY_ADD(char, &inp->name, elt, {});
- return 0;
- trunnel_alloc_failed:
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- char *
- domainname_getarray_name(domainname_t *inp)
- {
- return inp->name.elts_;
- }
- const char *
- domainname_getconstarray_name(const domainname_t *inp)
- {
- return (const char *)domainname_getarray_name((domainname_t*)inp);
- }
- int
- domainname_setlen_name(domainname_t *inp, size_t newlen)
- {
- #if UINT8_MAX < SIZE_MAX
- if (newlen > UINT8_MAX)
- goto trunnel_alloc_failed;
- #endif
- return trunnel_string_setlen(&inp->name, newlen,
- &inp->trunnel_error_code_);
- trunnel_alloc_failed:
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- const char *
- domainname_getstr_name(domainname_t *inp)
- {
- return trunnel_string_getstr(&inp->name);
- }
- int
- domainname_setstr0_name(domainname_t *inp, const char *val, size_t len)
- {
- #if UINT8_MAX < SIZE_MAX
- if (len > UINT8_MAX) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- #endif
- return trunnel_string_setstr0(&inp->name, val, len, &inp->trunnel_error_code_);
- }
- int
- domainname_setstr_name(domainname_t *inp, const char *val)
- {
- return domainname_setstr0_name(inp, val, strlen(val));
- }
- const char *
- domainname_check(const domainname_t *obj)
- {
- if (obj == NULL)
- return "Object was NULL";
- if (obj->trunnel_error_code_)
- return "A set function failed on this object";
- if (TRUNNEL_DYNARRAY_LEN(&obj->name) != obj->len)
- return "Length mismatch for name";
- return NULL;
- }
- ssize_t
- domainname_encoded_len(const domainname_t *obj)
- {
- ssize_t result = 0;
- if (NULL != domainname_check(obj))
- return -1;
- /* Length of u8 len */
- result += 1;
- /* Length of char name[len] */
- result += TRUNNEL_DYNARRAY_LEN(&obj->name);
- return result;
- }
- int
- domainname_clear_errors(domainname_t *obj)
- {
- int r = obj->trunnel_error_code_;
- obj->trunnel_error_code_ = 0;
- return r;
- }
- ssize_t
- domainname_encode(uint8_t *output, const size_t avail, const domainname_t *obj)
- {
- ssize_t result = 0;
- size_t written = 0;
- uint8_t *ptr = output;
- const char *msg;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- const ssize_t encoded_len = domainname_encoded_len(obj);
- #endif
- if (NULL != (msg = domainname_check(obj)))
- goto check_failed;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- trunnel_assert(encoded_len >= 0);
- #endif
- /* Encode u8 len */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->len));
- written += 1; ptr += 1;
- /* Encode char name[len] */
- {
- size_t elt_len = TRUNNEL_DYNARRAY_LEN(&obj->name);
- trunnel_assert(obj->len == elt_len);
- trunnel_assert(written <= avail);
- if (avail - written < elt_len)
- goto truncated;
- if (elt_len)
- memcpy(ptr, obj->name.elts_, elt_len);
- written += elt_len; ptr += elt_len;
- }
- trunnel_assert(ptr == output + written);
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- {
- trunnel_assert(encoded_len >= 0);
- trunnel_assert((size_t)encoded_len == written);
- }
- #endif
- return written;
- truncated:
- result = -2;
- goto fail;
- check_failed:
- (void)msg;
- result = -1;
- goto fail;
- fail:
- trunnel_assert(result < 0);
- return result;
- }
- /** As domainname_parse(), but do not allocate the output object.
- */
- static ssize_t
- domainname_parse_into(domainname_t *obj, const uint8_t *input, const size_t len_in)
- {
- const uint8_t *ptr = input;
- size_t remaining = len_in;
- ssize_t result = 0;
- (void)result;
- /* Parse u8 len */
- CHECK_REMAINING(1, truncated);
- obj->len = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- /* Parse char name[len] */
- CHECK_REMAINING(obj->len, truncated);
- if (domainname_setstr0_name(obj, (const char*)ptr, obj->len))
- goto fail;
- ptr += obj->len; remaining -= obj->len;
- trunnel_assert(ptr + remaining == input + len_in);
- return len_in - remaining;
- truncated:
- return -2;
- fail:
- result = -1;
- return result;
- }
- ssize_t
- domainname_parse(domainname_t **output, const uint8_t *input, const size_t len_in)
- {
- ssize_t result;
- *output = domainname_new();
- if (NULL == *output)
- return -1;
- result = domainname_parse_into(*output, input, len_in);
- if (result < 0) {
- domainname_free(*output);
- *output = NULL;
- }
- return result;
- }
- socks4_client_request_t *
- socks4_client_request_new(void)
- {
- socks4_client_request_t *val = trunnel_calloc(1, sizeof(socks4_client_request_t));
- if (NULL == val)
- return NULL;
- val->version = 4;
- val->command = CMD_BIND;
- return val;
- }
- /** Release all storage held inside 'obj', but do not free 'obj'.
- */
- static void
- socks4_client_request_clear(socks4_client_request_t *obj)
- {
- (void) obj;
- trunnel_wipestr(obj->username);
- trunnel_free(obj->username);
- trunnel_wipestr(obj->socks4a_addr_hostname);
- trunnel_free(obj->socks4a_addr_hostname);
- }
- void
- socks4_client_request_free(socks4_client_request_t *obj)
- {
- if (obj == NULL)
- return;
- socks4_client_request_clear(obj);
- trunnel_memwipe(obj, sizeof(socks4_client_request_t));
- trunnel_free_(obj);
- }
- uint8_t
- socks4_client_request_get_version(const socks4_client_request_t *inp)
- {
- return inp->version;
- }
- int
- socks4_client_request_set_version(socks4_client_request_t *inp, uint8_t val)
- {
- if (! ((val == 4))) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- inp->version = val;
- return 0;
- }
- uint8_t
- socks4_client_request_get_command(const socks4_client_request_t *inp)
- {
- return inp->command;
- }
- int
- socks4_client_request_set_command(socks4_client_request_t *inp, uint8_t val)
- {
- if (! ((val == CMD_BIND || val == CMD_CONNECT || val == CMD_RESOLVE || val == CMD_RESOLVE_PTR))) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- inp->command = val;
- return 0;
- }
- uint16_t
- socks4_client_request_get_port(const socks4_client_request_t *inp)
- {
- return inp->port;
- }
- int
- socks4_client_request_set_port(socks4_client_request_t *inp, uint16_t val)
- {
- inp->port = val;
- return 0;
- }
- uint32_t
- socks4_client_request_get_addr(const socks4_client_request_t *inp)
- {
- return inp->addr;
- }
- int
- socks4_client_request_set_addr(socks4_client_request_t *inp, uint32_t val)
- {
- inp->addr = val;
- return 0;
- }
- const char *
- socks4_client_request_get_username(const socks4_client_request_t *inp)
- {
- return inp->username;
- }
- int
- socks4_client_request_set_username(socks4_client_request_t *inp, const char *val)
- {
- trunnel_free(inp->username);
- if (NULL == (inp->username = trunnel_strdup(val))) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- return 0;
- }
- const char *
- socks4_client_request_get_socks4a_addr_hostname(const socks4_client_request_t *inp)
- {
- return inp->socks4a_addr_hostname;
- }
- int
- socks4_client_request_set_socks4a_addr_hostname(socks4_client_request_t *inp, const char *val)
- {
- trunnel_free(inp->socks4a_addr_hostname);
- if (NULL == (inp->socks4a_addr_hostname = trunnel_strdup(val))) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- return 0;
- }
- const char *
- socks4_client_request_check(const socks4_client_request_t *obj)
- {
- if (obj == NULL)
- return "Object was NULL";
- if (obj->trunnel_error_code_)
- return "A set function failed on this object";
- if (! (obj->version == 4))
- return "Integer out of bounds";
- if (! (obj->command == CMD_BIND || obj->command == CMD_CONNECT || obj->command == CMD_RESOLVE || obj->command == CMD_RESOLVE_PTR))
- return "Integer out of bounds";
- if (NULL == obj->username)
- return "Missing username";
- switch (obj->addr) {
- case 1:
- case 2:
- case 3:
- case 4:
- case 5:
- case 6:
- case 7:
- case 8:
- case 9:
- case 10:
- case 11:
- case 12:
- case 13:
- case 14:
- case 15:
- case 16:
- case 17:
- case 18:
- case 19:
- case 20:
- case 21:
- case 22:
- case 23:
- case 24:
- case 25:
- case 26:
- case 27:
- case 28:
- case 29:
- case 30:
- case 31:
- case 32:
- case 33:
- case 34:
- case 35:
- case 36:
- case 37:
- case 38:
- case 39:
- case 40:
- case 41:
- case 42:
- case 43:
- case 44:
- case 45:
- case 46:
- case 47:
- case 48:
- case 49:
- case 50:
- case 51:
- case 52:
- case 53:
- case 54:
- case 55:
- case 56:
- case 57:
- case 58:
- case 59:
- case 60:
- case 61:
- case 62:
- case 63:
- case 64:
- case 65:
- case 66:
- case 67:
- case 68:
- case 69:
- case 70:
- case 71:
- case 72:
- case 73:
- case 74:
- case 75:
- case 76:
- case 77:
- case 78:
- case 79:
- case 80:
- case 81:
- case 82:
- case 83:
- case 84:
- case 85:
- case 86:
- case 87:
- case 88:
- case 89:
- case 90:
- case 91:
- case 92:
- case 93:
- case 94:
- case 95:
- case 96:
- case 97:
- case 98:
- case 99:
- case 100:
- case 101:
- case 102:
- case 103:
- case 104:
- case 105:
- case 106:
- case 107:
- case 108:
- case 109:
- case 110:
- case 111:
- case 112:
- case 113:
- case 114:
- case 115:
- case 116:
- case 117:
- case 118:
- case 119:
- case 120:
- case 121:
- case 122:
- case 123:
- case 124:
- case 125:
- case 126:
- case 127:
- case 128:
- case 129:
- case 130:
- case 131:
- case 132:
- case 133:
- case 134:
- case 135:
- case 136:
- case 137:
- case 138:
- case 139:
- case 140:
- case 141:
- case 142:
- case 143:
- case 144:
- case 145:
- case 146:
- case 147:
- case 148:
- case 149:
- case 150:
- case 151:
- case 152:
- case 153:
- case 154:
- case 155:
- case 156:
- case 157:
- case 158:
- case 159:
- case 160:
- case 161:
- case 162:
- case 163:
- case 164:
- case 165:
- case 166:
- case 167:
- case 168:
- case 169:
- case 170:
- case 171:
- case 172:
- case 173:
- case 174:
- case 175:
- case 176:
- case 177:
- case 178:
- case 179:
- case 180:
- case 181:
- case 182:
- case 183:
- case 184:
- case 185:
- case 186:
- case 187:
- case 188:
- case 189:
- case 190:
- case 191:
- case 192:
- case 193:
- case 194:
- case 195:
- case 196:
- case 197:
- case 198:
- case 199:
- case 200:
- case 201:
- case 202:
- case 203:
- case 204:
- case 205:
- case 206:
- case 207:
- case 208:
- case 209:
- case 210:
- case 211:
- case 212:
- case 213:
- case 214:
- case 215:
- case 216:
- case 217:
- case 218:
- case 219:
- case 220:
- case 221:
- case 222:
- case 223:
- case 224:
- case 225:
- case 226:
- case 227:
- case 228:
- case 229:
- case 230:
- case 231:
- case 232:
- case 233:
- case 234:
- case 235:
- case 236:
- case 237:
- case 238:
- case 239:
- case 240:
- case 241:
- case 242:
- case 243:
- case 244:
- case 245:
- case 246:
- case 247:
- case 248:
- case 249:
- case 250:
- case 251:
- case 252:
- case 253:
- case 254:
- case 255:
- if (NULL == obj->socks4a_addr_hostname)
- return "Missing socks4a_addr_hostname";
- break;
- default:
- break;
- }
- return NULL;
- }
- ssize_t
- socks4_client_request_encoded_len(const socks4_client_request_t *obj)
- {
- ssize_t result = 0;
- if (NULL != socks4_client_request_check(obj))
- return -1;
- /* Length of u8 version IN [4] */
- result += 1;
- /* Length of u8 command IN [CMD_BIND, CMD_CONNECT, CMD_RESOLVE, CMD_RESOLVE_PTR] */
- result += 1;
- /* Length of u16 port */
- result += 2;
- /* Length of u32 addr */
- result += 4;
- /* Length of nulterm username */
- result += strlen(obj->username) + 1;
- switch (obj->addr) {
- case 1:
- case 2:
- case 3:
- case 4:
- case 5:
- case 6:
- case 7:
- case 8:
- case 9:
- case 10:
- case 11:
- case 12:
- case 13:
- case 14:
- case 15:
- case 16:
- case 17:
- case 18:
- case 19:
- case 20:
- case 21:
- case 22:
- case 23:
- case 24:
- case 25:
- case 26:
- case 27:
- case 28:
- case 29:
- case 30:
- case 31:
- case 32:
- case 33:
- case 34:
- case 35:
- case 36:
- case 37:
- case 38:
- case 39:
- case 40:
- case 41:
- case 42:
- case 43:
- case 44:
- case 45:
- case 46:
- case 47:
- case 48:
- case 49:
- case 50:
- case 51:
- case 52:
- case 53:
- case 54:
- case 55:
- case 56:
- case 57:
- case 58:
- case 59:
- case 60:
- case 61:
- case 62:
- case 63:
- case 64:
- case 65:
- case 66:
- case 67:
- case 68:
- case 69:
- case 70:
- case 71:
- case 72:
- case 73:
- case 74:
- case 75:
- case 76:
- case 77:
- case 78:
- case 79:
- case 80:
- case 81:
- case 82:
- case 83:
- case 84:
- case 85:
- case 86:
- case 87:
- case 88:
- case 89:
- case 90:
- case 91:
- case 92:
- case 93:
- case 94:
- case 95:
- case 96:
- case 97:
- case 98:
- case 99:
- case 100:
- case 101:
- case 102:
- case 103:
- case 104:
- case 105:
- case 106:
- case 107:
- case 108:
- case 109:
- case 110:
- case 111:
- case 112:
- case 113:
- case 114:
- case 115:
- case 116:
- case 117:
- case 118:
- case 119:
- case 120:
- case 121:
- case 122:
- case 123:
- case 124:
- case 125:
- case 126:
- case 127:
- case 128:
- case 129:
- case 130:
- case 131:
- case 132:
- case 133:
- case 134:
- case 135:
- case 136:
- case 137:
- case 138:
- case 139:
- case 140:
- case 141:
- case 142:
- case 143:
- case 144:
- case 145:
- case 146:
- case 147:
- case 148:
- case 149:
- case 150:
- case 151:
- case 152:
- case 153:
- case 154:
- case 155:
- case 156:
- case 157:
- case 158:
- case 159:
- case 160:
- case 161:
- case 162:
- case 163:
- case 164:
- case 165:
- case 166:
- case 167:
- case 168:
- case 169:
- case 170:
- case 171:
- case 172:
- case 173:
- case 174:
- case 175:
- case 176:
- case 177:
- case 178:
- case 179:
- case 180:
- case 181:
- case 182:
- case 183:
- case 184:
- case 185:
- case 186:
- case 187:
- case 188:
- case 189:
- case 190:
- case 191:
- case 192:
- case 193:
- case 194:
- case 195:
- case 196:
- case 197:
- case 198:
- case 199:
- case 200:
- case 201:
- case 202:
- case 203:
- case 204:
- case 205:
- case 206:
- case 207:
- case 208:
- case 209:
- case 210:
- case 211:
- case 212:
- case 213:
- case 214:
- case 215:
- case 216:
- case 217:
- case 218:
- case 219:
- case 220:
- case 221:
- case 222:
- case 223:
- case 224:
- case 225:
- case 226:
- case 227:
- case 228:
- case 229:
- case 230:
- case 231:
- case 232:
- case 233:
- case 234:
- case 235:
- case 236:
- case 237:
- case 238:
- case 239:
- case 240:
- case 241:
- case 242:
- case 243:
- case 244:
- case 245:
- case 246:
- case 247:
- case 248:
- case 249:
- case 250:
- case 251:
- case 252:
- case 253:
- case 254:
- case 255:
- /* Length of nulterm socks4a_addr_hostname */
- result += strlen(obj->socks4a_addr_hostname) + 1;
- break;
- default:
- break;
- }
- return result;
- }
- int
- socks4_client_request_clear_errors(socks4_client_request_t *obj)
- {
- int r = obj->trunnel_error_code_;
- obj->trunnel_error_code_ = 0;
- return r;
- }
- ssize_t
- socks4_client_request_encode(uint8_t *output, const size_t avail, const socks4_client_request_t *obj)
- {
- ssize_t result = 0;
- size_t written = 0;
- uint8_t *ptr = output;
- const char *msg;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- const ssize_t encoded_len = socks4_client_request_encoded_len(obj);
- #endif
- if (NULL != (msg = socks4_client_request_check(obj)))
- goto check_failed;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- trunnel_assert(encoded_len >= 0);
- #endif
- /* Encode u8 version IN [4] */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->version));
- written += 1; ptr += 1;
- /* Encode u8 command IN [CMD_BIND, CMD_CONNECT, CMD_RESOLVE, CMD_RESOLVE_PTR] */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->command));
- written += 1; ptr += 1;
- /* Encode u16 port */
- trunnel_assert(written <= avail);
- if (avail - written < 2)
- goto truncated;
- trunnel_set_uint16(ptr, trunnel_htons(obj->port));
- written += 2; ptr += 2;
- /* Encode u32 addr */
- trunnel_assert(written <= avail);
- if (avail - written < 4)
- goto truncated;
- trunnel_set_uint32(ptr, trunnel_htonl(obj->addr));
- written += 4; ptr += 4;
- /* Encode nulterm username */
- {
- size_t len = strlen(obj->username);
- trunnel_assert(written <= avail);
- if (avail - written < len + 1)
- goto truncated;
- memcpy(ptr, obj->username, len + 1);
- ptr += len + 1; written += len + 1;
- }
- /* Encode union socks4a_addr[addr] */
- trunnel_assert(written <= avail);
- switch (obj->addr) {
- case 1:
- case 2:
- case 3:
- case 4:
- case 5:
- case 6:
- case 7:
- case 8:
- case 9:
- case 10:
- case 11:
- case 12:
- case 13:
- case 14:
- case 15:
- case 16:
- case 17:
- case 18:
- case 19:
- case 20:
- case 21:
- case 22:
- case 23:
- case 24:
- case 25:
- case 26:
- case 27:
- case 28:
- case 29:
- case 30:
- case 31:
- case 32:
- case 33:
- case 34:
- case 35:
- case 36:
- case 37:
- case 38:
- case 39:
- case 40:
- case 41:
- case 42:
- case 43:
- case 44:
- case 45:
- case 46:
- case 47:
- case 48:
- case 49:
- case 50:
- case 51:
- case 52:
- case 53:
- case 54:
- case 55:
- case 56:
- case 57:
- case 58:
- case 59:
- case 60:
- case 61:
- case 62:
- case 63:
- case 64:
- case 65:
- case 66:
- case 67:
- case 68:
- case 69:
- case 70:
- case 71:
- case 72:
- case 73:
- case 74:
- case 75:
- case 76:
- case 77:
- case 78:
- case 79:
- case 80:
- case 81:
- case 82:
- case 83:
- case 84:
- case 85:
- case 86:
- case 87:
- case 88:
- case 89:
- case 90:
- case 91:
- case 92:
- case 93:
- case 94:
- case 95:
- case 96:
- case 97:
- case 98:
- case 99:
- case 100:
- case 101:
- case 102:
- case 103:
- case 104:
- case 105:
- case 106:
- case 107:
- case 108:
- case 109:
- case 110:
- case 111:
- case 112:
- case 113:
- case 114:
- case 115:
- case 116:
- case 117:
- case 118:
- case 119:
- case 120:
- case 121:
- case 122:
- case 123:
- case 124:
- case 125:
- case 126:
- case 127:
- case 128:
- case 129:
- case 130:
- case 131:
- case 132:
- case 133:
- case 134:
- case 135:
- case 136:
- case 137:
- case 138:
- case 139:
- case 140:
- case 141:
- case 142:
- case 143:
- case 144:
- case 145:
- case 146:
- case 147:
- case 148:
- case 149:
- case 150:
- case 151:
- case 152:
- case 153:
- case 154:
- case 155:
- case 156:
- case 157:
- case 158:
- case 159:
- case 160:
- case 161:
- case 162:
- case 163:
- case 164:
- case 165:
- case 166:
- case 167:
- case 168:
- case 169:
- case 170:
- case 171:
- case 172:
- case 173:
- case 174:
- case 175:
- case 176:
- case 177:
- case 178:
- case 179:
- case 180:
- case 181:
- case 182:
- case 183:
- case 184:
- case 185:
- case 186:
- case 187:
- case 188:
- case 189:
- case 190:
- case 191:
- case 192:
- case 193:
- case 194:
- case 195:
- case 196:
- case 197:
- case 198:
- case 199:
- case 200:
- case 201:
- case 202:
- case 203:
- case 204:
- case 205:
- case 206:
- case 207:
- case 208:
- case 209:
- case 210:
- case 211:
- case 212:
- case 213:
- case 214:
- case 215:
- case 216:
- case 217:
- case 218:
- case 219:
- case 220:
- case 221:
- case 222:
- case 223:
- case 224:
- case 225:
- case 226:
- case 227:
- case 228:
- case 229:
- case 230:
- case 231:
- case 232:
- case 233:
- case 234:
- case 235:
- case 236:
- case 237:
- case 238:
- case 239:
- case 240:
- case 241:
- case 242:
- case 243:
- case 244:
- case 245:
- case 246:
- case 247:
- case 248:
- case 249:
- case 250:
- case 251:
- case 252:
- case 253:
- case 254:
- case 255:
- /* Encode nulterm socks4a_addr_hostname */
- {
- size_t len = strlen(obj->socks4a_addr_hostname);
- trunnel_assert(written <= avail);
- if (avail - written < len + 1)
- goto truncated;
- memcpy(ptr, obj->socks4a_addr_hostname, len + 1);
- ptr += len + 1; written += len + 1;
- }
- break;
- default:
- break;
- }
- trunnel_assert(ptr == output + written);
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- {
- trunnel_assert(encoded_len >= 0);
- trunnel_assert((size_t)encoded_len == written);
- }
- #endif
- return written;
- truncated:
- result = -2;
- goto fail;
- check_failed:
- (void)msg;
- result = -1;
- goto fail;
- fail:
- trunnel_assert(result < 0);
- return result;
- }
- /** As socks4_client_request_parse(), but do not allocate the output
- * object.
- */
- static ssize_t
- socks4_client_request_parse_into(socks4_client_request_t *obj, const uint8_t *input, const size_t len_in)
- {
- const uint8_t *ptr = input;
- size_t remaining = len_in;
- ssize_t result = 0;
- (void)result;
- /* Parse u8 version IN [4] */
- CHECK_REMAINING(1, truncated);
- obj->version = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- if (! (obj->version == 4))
- goto fail;
- /* Parse u8 command IN [CMD_BIND, CMD_CONNECT, CMD_RESOLVE, CMD_RESOLVE_PTR] */
- CHECK_REMAINING(1, truncated);
- obj->command = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- if (! (obj->command == CMD_BIND || obj->command == CMD_CONNECT || obj->command == CMD_RESOLVE || obj->command == CMD_RESOLVE_PTR))
- goto fail;
- /* Parse u16 port */
- CHECK_REMAINING(2, truncated);
- obj->port = trunnel_ntohs(trunnel_get_uint16(ptr));
- remaining -= 2; ptr += 2;
- /* Parse u32 addr */
- CHECK_REMAINING(4, truncated);
- obj->addr = trunnel_ntohl(trunnel_get_uint32(ptr));
- remaining -= 4; ptr += 4;
- /* Parse nulterm username */
- {
- uint8_t *eos = (uint8_t*)memchr(ptr, 0, remaining);
- size_t memlen;
- if (eos == NULL)
- goto truncated;
- trunnel_assert(eos >= ptr);
- trunnel_assert((size_t)(eos - ptr) < SIZE_MAX - 1);
- memlen = ((size_t)(eos - ptr)) + 1;
- if (!(obj->username = trunnel_malloc(memlen)))
- goto fail;
- memcpy(obj->username, ptr, memlen);
- remaining -= memlen; ptr += memlen;
- }
- /* Parse union socks4a_addr[addr] */
- switch (obj->addr) {
- case 1:
- case 2:
- case 3:
- case 4:
- case 5:
- case 6:
- case 7:
- case 8:
- case 9:
- case 10:
- case 11:
- case 12:
- case 13:
- case 14:
- case 15:
- case 16:
- case 17:
- case 18:
- case 19:
- case 20:
- case 21:
- case 22:
- case 23:
- case 24:
- case 25:
- case 26:
- case 27:
- case 28:
- case 29:
- case 30:
- case 31:
- case 32:
- case 33:
- case 34:
- case 35:
- case 36:
- case 37:
- case 38:
- case 39:
- case 40:
- case 41:
- case 42:
- case 43:
- case 44:
- case 45:
- case 46:
- case 47:
- case 48:
- case 49:
- case 50:
- case 51:
- case 52:
- case 53:
- case 54:
- case 55:
- case 56:
- case 57:
- case 58:
- case 59:
- case 60:
- case 61:
- case 62:
- case 63:
- case 64:
- case 65:
- case 66:
- case 67:
- case 68:
- case 69:
- case 70:
- case 71:
- case 72:
- case 73:
- case 74:
- case 75:
- case 76:
- case 77:
- case 78:
- case 79:
- case 80:
- case 81:
- case 82:
- case 83:
- case 84:
- case 85:
- case 86:
- case 87:
- case 88:
- case 89:
- case 90:
- case 91:
- case 92:
- case 93:
- case 94:
- case 95:
- case 96:
- case 97:
- case 98:
- case 99:
- case 100:
- case 101:
- case 102:
- case 103:
- case 104:
- case 105:
- case 106:
- case 107:
- case 108:
- case 109:
- case 110:
- case 111:
- case 112:
- case 113:
- case 114:
- case 115:
- case 116:
- case 117:
- case 118:
- case 119:
- case 120:
- case 121:
- case 122:
- case 123:
- case 124:
- case 125:
- case 126:
- case 127:
- case 128:
- case 129:
- case 130:
- case 131:
- case 132:
- case 133:
- case 134:
- case 135:
- case 136:
- case 137:
- case 138:
- case 139:
- case 140:
- case 141:
- case 142:
- case 143:
- case 144:
- case 145:
- case 146:
- case 147:
- case 148:
- case 149:
- case 150:
- case 151:
- case 152:
- case 153:
- case 154:
- case 155:
- case 156:
- case 157:
- case 158:
- case 159:
- case 160:
- case 161:
- case 162:
- case 163:
- case 164:
- case 165:
- case 166:
- case 167:
- case 168:
- case 169:
- case 170:
- case 171:
- case 172:
- case 173:
- case 174:
- case 175:
- case 176:
- case 177:
- case 178:
- case 179:
- case 180:
- case 181:
- case 182:
- case 183:
- case 184:
- case 185:
- case 186:
- case 187:
- case 188:
- case 189:
- case 190:
- case 191:
- case 192:
- case 193:
- case 194:
- case 195:
- case 196:
- case 197:
- case 198:
- case 199:
- case 200:
- case 201:
- case 202:
- case 203:
- case 204:
- case 205:
- case 206:
- case 207:
- case 208:
- case 209:
- case 210:
- case 211:
- case 212:
- case 213:
- case 214:
- case 215:
- case 216:
- case 217:
- case 218:
- case 219:
- case 220:
- case 221:
- case 222:
- case 223:
- case 224:
- case 225:
- case 226:
- case 227:
- case 228:
- case 229:
- case 230:
- case 231:
- case 232:
- case 233:
- case 234:
- case 235:
- case 236:
- case 237:
- case 238:
- case 239:
- case 240:
- case 241:
- case 242:
- case 243:
- case 244:
- case 245:
- case 246:
- case 247:
- case 248:
- case 249:
- case 250:
- case 251:
- case 252:
- case 253:
- case 254:
- case 255:
- /* Parse nulterm socks4a_addr_hostname */
- {
- uint8_t *eos = (uint8_t*)memchr(ptr, 0, remaining);
- size_t memlen;
- if (eos == NULL)
- goto truncated;
- trunnel_assert(eos >= ptr);
- trunnel_assert((size_t)(eos - ptr) < SIZE_MAX - 1);
- memlen = ((size_t)(eos - ptr)) + 1;
- if (!(obj->socks4a_addr_hostname = trunnel_malloc(memlen)))
- goto fail;
- memcpy(obj->socks4a_addr_hostname, ptr, memlen);
- remaining -= memlen; ptr += memlen;
- }
- break;
- default:
- break;
- }
- trunnel_assert(ptr + remaining == input + len_in);
- return len_in - remaining;
- truncated:
- return -2;
- fail:
- result = -1;
- return result;
- }
- ssize_t
- socks4_client_request_parse(socks4_client_request_t **output, const uint8_t *input, const size_t len_in)
- {
- ssize_t result;
- *output = socks4_client_request_new();
- if (NULL == *output)
- return -1;
- result = socks4_client_request_parse_into(*output, input, len_in);
- if (result < 0) {
- socks4_client_request_free(*output);
- *output = NULL;
- }
- return result;
- }
- socks4_server_reply_t *
- socks4_server_reply_new(void)
- {
- socks4_server_reply_t *val = trunnel_calloc(1, sizeof(socks4_server_reply_t));
- if (NULL == val)
- return NULL;
- return val;
- }
- /** Release all storage held inside 'obj', but do not free 'obj'.
- */
- static void
- socks4_server_reply_clear(socks4_server_reply_t *obj)
- {
- (void) obj;
- }
- void
- socks4_server_reply_free(socks4_server_reply_t *obj)
- {
- if (obj == NULL)
- return;
- socks4_server_reply_clear(obj);
- trunnel_memwipe(obj, sizeof(socks4_server_reply_t));
- trunnel_free_(obj);
- }
- uint8_t
- socks4_server_reply_get_version(const socks4_server_reply_t *inp)
- {
- return inp->version;
- }
- int
- socks4_server_reply_set_version(socks4_server_reply_t *inp, uint8_t val)
- {
- if (! ((val == 0 || val == 4))) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- inp->version = val;
- return 0;
- }
- uint8_t
- socks4_server_reply_get_status(const socks4_server_reply_t *inp)
- {
- return inp->status;
- }
- int
- socks4_server_reply_set_status(socks4_server_reply_t *inp, uint8_t val)
- {
- inp->status = val;
- return 0;
- }
- uint16_t
- socks4_server_reply_get_port(const socks4_server_reply_t *inp)
- {
- return inp->port;
- }
- int
- socks4_server_reply_set_port(socks4_server_reply_t *inp, uint16_t val)
- {
- inp->port = val;
- return 0;
- }
- uint32_t
- socks4_server_reply_get_addr(const socks4_server_reply_t *inp)
- {
- return inp->addr;
- }
- int
- socks4_server_reply_set_addr(socks4_server_reply_t *inp, uint32_t val)
- {
- inp->addr = val;
- return 0;
- }
- const char *
- socks4_server_reply_check(const socks4_server_reply_t *obj)
- {
- if (obj == NULL)
- return "Object was NULL";
- if (obj->trunnel_error_code_)
- return "A set function failed on this object";
- if (! (obj->version == 0 || obj->version == 4))
- return "Integer out of bounds";
- return NULL;
- }
- ssize_t
- socks4_server_reply_encoded_len(const socks4_server_reply_t *obj)
- {
- ssize_t result = 0;
- if (NULL != socks4_server_reply_check(obj))
- return -1;
- /* Length of u8 version IN [0, 4] */
- result += 1;
- /* Length of u8 status */
- result += 1;
- /* Length of u16 port */
- result += 2;
- /* Length of u32 addr */
- result += 4;
- return result;
- }
- int
- socks4_server_reply_clear_errors(socks4_server_reply_t *obj)
- {
- int r = obj->trunnel_error_code_;
- obj->trunnel_error_code_ = 0;
- return r;
- }
- ssize_t
- socks4_server_reply_encode(uint8_t *output, const size_t avail, const socks4_server_reply_t *obj)
- {
- ssize_t result = 0;
- size_t written = 0;
- uint8_t *ptr = output;
- const char *msg;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- const ssize_t encoded_len = socks4_server_reply_encoded_len(obj);
- #endif
- if (NULL != (msg = socks4_server_reply_check(obj)))
- goto check_failed;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- trunnel_assert(encoded_len >= 0);
- #endif
- /* Encode u8 version IN [0, 4] */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->version));
- written += 1; ptr += 1;
- /* Encode u8 status */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->status));
- written += 1; ptr += 1;
- /* Encode u16 port */
- trunnel_assert(written <= avail);
- if (avail - written < 2)
- goto truncated;
- trunnel_set_uint16(ptr, trunnel_htons(obj->port));
- written += 2; ptr += 2;
- /* Encode u32 addr */
- trunnel_assert(written <= avail);
- if (avail - written < 4)
- goto truncated;
- trunnel_set_uint32(ptr, trunnel_htonl(obj->addr));
- written += 4; ptr += 4;
- trunnel_assert(ptr == output + written);
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- {
- trunnel_assert(encoded_len >= 0);
- trunnel_assert((size_t)encoded_len == written);
- }
- #endif
- return written;
- truncated:
- result = -2;
- goto fail;
- check_failed:
- (void)msg;
- result = -1;
- goto fail;
- fail:
- trunnel_assert(result < 0);
- return result;
- }
- /** As socks4_server_reply_parse(), but do not allocate the output
- * object.
- */
- static ssize_t
- socks4_server_reply_parse_into(socks4_server_reply_t *obj, const uint8_t *input, const size_t len_in)
- {
- const uint8_t *ptr = input;
- size_t remaining = len_in;
- ssize_t result = 0;
- (void)result;
- /* Parse u8 version IN [0, 4] */
- CHECK_REMAINING(1, truncated);
- obj->version = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- if (! (obj->version == 0 || obj->version == 4))
- goto fail;
- /* Parse u8 status */
- CHECK_REMAINING(1, truncated);
- obj->status = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- /* Parse u16 port */
- CHECK_REMAINING(2, truncated);
- obj->port = trunnel_ntohs(trunnel_get_uint16(ptr));
- remaining -= 2; ptr += 2;
- /* Parse u32 addr */
- CHECK_REMAINING(4, truncated);
- obj->addr = trunnel_ntohl(trunnel_get_uint32(ptr));
- remaining -= 4; ptr += 4;
- trunnel_assert(ptr + remaining == input + len_in);
- return len_in - remaining;
- truncated:
- return -2;
- fail:
- result = -1;
- return result;
- }
- ssize_t
- socks4_server_reply_parse(socks4_server_reply_t **output, const uint8_t *input, const size_t len_in)
- {
- ssize_t result;
- *output = socks4_server_reply_new();
- if (NULL == *output)
- return -1;
- result = socks4_server_reply_parse_into(*output, input, len_in);
- if (result < 0) {
- socks4_server_reply_free(*output);
- *output = NULL;
- }
- return result;
- }
- socks5_client_userpass_auth_t *
- socks5_client_userpass_auth_new(void)
- {
- socks5_client_userpass_auth_t *val = trunnel_calloc(1, sizeof(socks5_client_userpass_auth_t));
- if (NULL == val)
- return NULL;
- val->version = 1;
- return val;
- }
- /** Release all storage held inside 'obj', but do not free 'obj'.
- */
- static void
- socks5_client_userpass_auth_clear(socks5_client_userpass_auth_t *obj)
- {
- (void) obj;
- TRUNNEL_DYNARRAY_WIPE(&obj->username);
- TRUNNEL_DYNARRAY_CLEAR(&obj->username);
- TRUNNEL_DYNARRAY_WIPE(&obj->passwd);
- TRUNNEL_DYNARRAY_CLEAR(&obj->passwd);
- }
- void
- socks5_client_userpass_auth_free(socks5_client_userpass_auth_t *obj)
- {
- if (obj == NULL)
- return;
- socks5_client_userpass_auth_clear(obj);
- trunnel_memwipe(obj, sizeof(socks5_client_userpass_auth_t));
- trunnel_free_(obj);
- }
- uint8_t
- socks5_client_userpass_auth_get_version(const socks5_client_userpass_auth_t *inp)
- {
- return inp->version;
- }
- int
- socks5_client_userpass_auth_set_version(socks5_client_userpass_auth_t *inp, uint8_t val)
- {
- if (! ((val == 1))) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- inp->version = val;
- return 0;
- }
- uint8_t
- socks5_client_userpass_auth_get_username_len(const socks5_client_userpass_auth_t *inp)
- {
- return inp->username_len;
- }
- int
- socks5_client_userpass_auth_set_username_len(socks5_client_userpass_auth_t *inp, uint8_t val)
- {
- inp->username_len = val;
- return 0;
- }
- size_t
- socks5_client_userpass_auth_getlen_username(const socks5_client_userpass_auth_t *inp)
- {
- return TRUNNEL_DYNARRAY_LEN(&inp->username);
- }
- char
- socks5_client_userpass_auth_get_username(socks5_client_userpass_auth_t *inp, size_t idx)
- {
- return TRUNNEL_DYNARRAY_GET(&inp->username, idx);
- }
- char
- socks5_client_userpass_auth_getconst_username(const socks5_client_userpass_auth_t *inp, size_t idx)
- {
- return socks5_client_userpass_auth_get_username((socks5_client_userpass_auth_t*)inp, idx);
- }
- int
- socks5_client_userpass_auth_set_username(socks5_client_userpass_auth_t *inp, size_t idx, char elt)
- {
- TRUNNEL_DYNARRAY_SET(&inp->username, idx, elt);
- return 0;
- }
- int
- socks5_client_userpass_auth_add_username(socks5_client_userpass_auth_t *inp, char elt)
- {
- #if SIZE_MAX >= UINT8_MAX
- if (inp->username.n_ == UINT8_MAX)
- goto trunnel_alloc_failed;
- #endif
- TRUNNEL_DYNARRAY_ADD(char, &inp->username, elt, {});
- return 0;
- trunnel_alloc_failed:
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- char *
- socks5_client_userpass_auth_getarray_username(socks5_client_userpass_auth_t *inp)
- {
- return inp->username.elts_;
- }
- const char *
- socks5_client_userpass_auth_getconstarray_username(const socks5_client_userpass_auth_t *inp)
- {
- return (const char *)socks5_client_userpass_auth_getarray_username((socks5_client_userpass_auth_t*)inp);
- }
- int
- socks5_client_userpass_auth_setlen_username(socks5_client_userpass_auth_t *inp, size_t newlen)
- {
- #if UINT8_MAX < SIZE_MAX
- if (newlen > UINT8_MAX)
- goto trunnel_alloc_failed;
- #endif
- return trunnel_string_setlen(&inp->username, newlen,
- &inp->trunnel_error_code_);
- trunnel_alloc_failed:
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- const char *
- socks5_client_userpass_auth_getstr_username(socks5_client_userpass_auth_t *inp)
- {
- return trunnel_string_getstr(&inp->username);
- }
- int
- socks5_client_userpass_auth_setstr0_username(socks5_client_userpass_auth_t *inp, const char *val, size_t len)
- {
- #if UINT8_MAX < SIZE_MAX
- if (len > UINT8_MAX) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- #endif
- return trunnel_string_setstr0(&inp->username, val, len, &inp->trunnel_error_code_);
- }
- int
- socks5_client_userpass_auth_setstr_username(socks5_client_userpass_auth_t *inp, const char *val)
- {
- return socks5_client_userpass_auth_setstr0_username(inp, val, strlen(val));
- }
- uint8_t
- socks5_client_userpass_auth_get_passwd_len(const socks5_client_userpass_auth_t *inp)
- {
- return inp->passwd_len;
- }
- int
- socks5_client_userpass_auth_set_passwd_len(socks5_client_userpass_auth_t *inp, uint8_t val)
- {
- inp->passwd_len = val;
- return 0;
- }
- size_t
- socks5_client_userpass_auth_getlen_passwd(const socks5_client_userpass_auth_t *inp)
- {
- return TRUNNEL_DYNARRAY_LEN(&inp->passwd);
- }
- char
- socks5_client_userpass_auth_get_passwd(socks5_client_userpass_auth_t *inp, size_t idx)
- {
- return TRUNNEL_DYNARRAY_GET(&inp->passwd, idx);
- }
- char
- socks5_client_userpass_auth_getconst_passwd(const socks5_client_userpass_auth_t *inp, size_t idx)
- {
- return socks5_client_userpass_auth_get_passwd((socks5_client_userpass_auth_t*)inp, idx);
- }
- int
- socks5_client_userpass_auth_set_passwd(socks5_client_userpass_auth_t *inp, size_t idx, char elt)
- {
- TRUNNEL_DYNARRAY_SET(&inp->passwd, idx, elt);
- return 0;
- }
- int
- socks5_client_userpass_auth_add_passwd(socks5_client_userpass_auth_t *inp, char elt)
- {
- #if SIZE_MAX >= UINT8_MAX
- if (inp->passwd.n_ == UINT8_MAX)
- goto trunnel_alloc_failed;
- #endif
- TRUNNEL_DYNARRAY_ADD(char, &inp->passwd, elt, {});
- return 0;
- trunnel_alloc_failed:
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- char *
- socks5_client_userpass_auth_getarray_passwd(socks5_client_userpass_auth_t *inp)
- {
- return inp->passwd.elts_;
- }
- const char *
- socks5_client_userpass_auth_getconstarray_passwd(const socks5_client_userpass_auth_t *inp)
- {
- return (const char *)socks5_client_userpass_auth_getarray_passwd((socks5_client_userpass_auth_t*)inp);
- }
- int
- socks5_client_userpass_auth_setlen_passwd(socks5_client_userpass_auth_t *inp, size_t newlen)
- {
- #if UINT8_MAX < SIZE_MAX
- if (newlen > UINT8_MAX)
- goto trunnel_alloc_failed;
- #endif
- return trunnel_string_setlen(&inp->passwd, newlen,
- &inp->trunnel_error_code_);
- trunnel_alloc_failed:
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- const char *
- socks5_client_userpass_auth_getstr_passwd(socks5_client_userpass_auth_t *inp)
- {
- return trunnel_string_getstr(&inp->passwd);
- }
- int
- socks5_client_userpass_auth_setstr0_passwd(socks5_client_userpass_auth_t *inp, const char *val, size_t len)
- {
- #if UINT8_MAX < SIZE_MAX
- if (len > UINT8_MAX) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- #endif
- return trunnel_string_setstr0(&inp->passwd, val, len, &inp->trunnel_error_code_);
- }
- int
- socks5_client_userpass_auth_setstr_passwd(socks5_client_userpass_auth_t *inp, const char *val)
- {
- return socks5_client_userpass_auth_setstr0_passwd(inp, val, strlen(val));
- }
- const char *
- socks5_client_userpass_auth_check(const socks5_client_userpass_auth_t *obj)
- {
- if (obj == NULL)
- return "Object was NULL";
- if (obj->trunnel_error_code_)
- return "A set function failed on this object";
- if (! (obj->version == 1))
- return "Integer out of bounds";
- if (TRUNNEL_DYNARRAY_LEN(&obj->username) != obj->username_len)
- return "Length mismatch for username";
- if (TRUNNEL_DYNARRAY_LEN(&obj->passwd) != obj->passwd_len)
- return "Length mismatch for passwd";
- return NULL;
- }
- ssize_t
- socks5_client_userpass_auth_encoded_len(const socks5_client_userpass_auth_t *obj)
- {
- ssize_t result = 0;
- if (NULL != socks5_client_userpass_auth_check(obj))
- return -1;
- /* Length of u8 version IN [1] */
- result += 1;
- /* Length of u8 username_len */
- result += 1;
- /* Length of char username[username_len] */
- result += TRUNNEL_DYNARRAY_LEN(&obj->username);
- /* Length of u8 passwd_len */
- result += 1;
- /* Length of char passwd[passwd_len] */
- result += TRUNNEL_DYNARRAY_LEN(&obj->passwd);
- return result;
- }
- int
- socks5_client_userpass_auth_clear_errors(socks5_client_userpass_auth_t *obj)
- {
- int r = obj->trunnel_error_code_;
- obj->trunnel_error_code_ = 0;
- return r;
- }
- ssize_t
- socks5_client_userpass_auth_encode(uint8_t *output, const size_t avail, const socks5_client_userpass_auth_t *obj)
- {
- ssize_t result = 0;
- size_t written = 0;
- uint8_t *ptr = output;
- const char *msg;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- const ssize_t encoded_len = socks5_client_userpass_auth_encoded_len(obj);
- #endif
- if (NULL != (msg = socks5_client_userpass_auth_check(obj)))
- goto check_failed;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- trunnel_assert(encoded_len >= 0);
- #endif
- /* Encode u8 version IN [1] */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->version));
- written += 1; ptr += 1;
- /* Encode u8 username_len */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->username_len));
- written += 1; ptr += 1;
- /* Encode char username[username_len] */
- {
- size_t elt_len = TRUNNEL_DYNARRAY_LEN(&obj->username);
- trunnel_assert(obj->username_len == elt_len);
- trunnel_assert(written <= avail);
- if (avail - written < elt_len)
- goto truncated;
- if (elt_len)
- memcpy(ptr, obj->username.elts_, elt_len);
- written += elt_len; ptr += elt_len;
- }
- /* Encode u8 passwd_len */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->passwd_len));
- written += 1; ptr += 1;
- /* Encode char passwd[passwd_len] */
- {
- size_t elt_len = TRUNNEL_DYNARRAY_LEN(&obj->passwd);
- trunnel_assert(obj->passwd_len == elt_len);
- trunnel_assert(written <= avail);
- if (avail - written < elt_len)
- goto truncated;
- if (elt_len)
- memcpy(ptr, obj->passwd.elts_, elt_len);
- written += elt_len; ptr += elt_len;
- }
- trunnel_assert(ptr == output + written);
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- {
- trunnel_assert(encoded_len >= 0);
- trunnel_assert((size_t)encoded_len == written);
- }
- #endif
- return written;
- truncated:
- result = -2;
- goto fail;
- check_failed:
- (void)msg;
- result = -1;
- goto fail;
- fail:
- trunnel_assert(result < 0);
- return result;
- }
- /** As socks5_client_userpass_auth_parse(), but do not allocate the
- * output object.
- */
- static ssize_t
- socks5_client_userpass_auth_parse_into(socks5_client_userpass_auth_t *obj, const uint8_t *input, const size_t len_in)
- {
- const uint8_t *ptr = input;
- size_t remaining = len_in;
- ssize_t result = 0;
- (void)result;
- /* Parse u8 version IN [1] */
- CHECK_REMAINING(1, truncated);
- obj->version = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- if (! (obj->version == 1))
- goto fail;
- /* Parse u8 username_len */
- CHECK_REMAINING(1, truncated);
- obj->username_len = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- /* Parse char username[username_len] */
- CHECK_REMAINING(obj->username_len, truncated);
- if (socks5_client_userpass_auth_setstr0_username(obj, (const char*)ptr, obj->username_len))
- goto fail;
- ptr += obj->username_len; remaining -= obj->username_len;
- /* Parse u8 passwd_len */
- CHECK_REMAINING(1, truncated);
- obj->passwd_len = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- /* Parse char passwd[passwd_len] */
- CHECK_REMAINING(obj->passwd_len, truncated);
- if (socks5_client_userpass_auth_setstr0_passwd(obj, (const char*)ptr, obj->passwd_len))
- goto fail;
- ptr += obj->passwd_len; remaining -= obj->passwd_len;
- trunnel_assert(ptr + remaining == input + len_in);
- return len_in - remaining;
- truncated:
- return -2;
- fail:
- result = -1;
- return result;
- }
- ssize_t
- socks5_client_userpass_auth_parse(socks5_client_userpass_auth_t **output, const uint8_t *input, const size_t len_in)
- {
- ssize_t result;
- *output = socks5_client_userpass_auth_new();
- if (NULL == *output)
- return -1;
- result = socks5_client_userpass_auth_parse_into(*output, input, len_in);
- if (result < 0) {
- socks5_client_userpass_auth_free(*output);
- *output = NULL;
- }
- return result;
- }
- socks5_client_version_t *
- socks5_client_version_new(void)
- {
- socks5_client_version_t *val = trunnel_calloc(1, sizeof(socks5_client_version_t));
- if (NULL == val)
- return NULL;
- val->version = 5;
- return val;
- }
- /** Release all storage held inside 'obj', but do not free 'obj'.
- */
- static void
- socks5_client_version_clear(socks5_client_version_t *obj)
- {
- (void) obj;
- TRUNNEL_DYNARRAY_WIPE(&obj->methods);
- TRUNNEL_DYNARRAY_CLEAR(&obj->methods);
- }
- void
- socks5_client_version_free(socks5_client_version_t *obj)
- {
- if (obj == NULL)
- return;
- socks5_client_version_clear(obj);
- trunnel_memwipe(obj, sizeof(socks5_client_version_t));
- trunnel_free_(obj);
- }
- uint8_t
- socks5_client_version_get_version(const socks5_client_version_t *inp)
- {
- return inp->version;
- }
- int
- socks5_client_version_set_version(socks5_client_version_t *inp, uint8_t val)
- {
- if (! ((val == 5))) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- inp->version = val;
- return 0;
- }
- uint8_t
- socks5_client_version_get_n_methods(const socks5_client_version_t *inp)
- {
- return inp->n_methods;
- }
- int
- socks5_client_version_set_n_methods(socks5_client_version_t *inp, uint8_t val)
- {
- inp->n_methods = val;
- return 0;
- }
- size_t
- socks5_client_version_getlen_methods(const socks5_client_version_t *inp)
- {
- return TRUNNEL_DYNARRAY_LEN(&inp->methods);
- }
- uint8_t
- socks5_client_version_get_methods(socks5_client_version_t *inp, size_t idx)
- {
- return TRUNNEL_DYNARRAY_GET(&inp->methods, idx);
- }
- uint8_t
- socks5_client_version_getconst_methods(const socks5_client_version_t *inp, size_t idx)
- {
- return socks5_client_version_get_methods((socks5_client_version_t*)inp, idx);
- }
- int
- socks5_client_version_set_methods(socks5_client_version_t *inp, size_t idx, uint8_t elt)
- {
- TRUNNEL_DYNARRAY_SET(&inp->methods, idx, elt);
- return 0;
- }
- int
- socks5_client_version_add_methods(socks5_client_version_t *inp, uint8_t elt)
- {
- #if SIZE_MAX >= UINT8_MAX
- if (inp->methods.n_ == UINT8_MAX)
- goto trunnel_alloc_failed;
- #endif
- TRUNNEL_DYNARRAY_ADD(uint8_t, &inp->methods, elt, {});
- return 0;
- trunnel_alloc_failed:
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- uint8_t *
- socks5_client_version_getarray_methods(socks5_client_version_t *inp)
- {
- return inp->methods.elts_;
- }
- const uint8_t *
- socks5_client_version_getconstarray_methods(const socks5_client_version_t *inp)
- {
- return (const uint8_t *)socks5_client_version_getarray_methods((socks5_client_version_t*)inp);
- }
- int
- socks5_client_version_setlen_methods(socks5_client_version_t *inp, size_t newlen)
- {
- uint8_t *newptr;
- #if UINT8_MAX < SIZE_MAX
- if (newlen > UINT8_MAX)
- goto trunnel_alloc_failed;
- #endif
- newptr = trunnel_dynarray_setlen(&inp->methods.allocated_,
- &inp->methods.n_, inp->methods.elts_, newlen,
- sizeof(inp->methods.elts_[0]), (trunnel_free_fn_t) NULL,
- &inp->trunnel_error_code_);
- if (newlen != 0 && newptr == NULL)
- goto trunnel_alloc_failed;
- inp->methods.elts_ = newptr;
- return 0;
- trunnel_alloc_failed:
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- const char *
- socks5_client_version_check(const socks5_client_version_t *obj)
- {
- if (obj == NULL)
- return "Object was NULL";
- if (obj->trunnel_error_code_)
- return "A set function failed on this object";
- if (! (obj->version == 5))
- return "Integer out of bounds";
- if (TRUNNEL_DYNARRAY_LEN(&obj->methods) != obj->n_methods)
- return "Length mismatch for methods";
- return NULL;
- }
- ssize_t
- socks5_client_version_encoded_len(const socks5_client_version_t *obj)
- {
- ssize_t result = 0;
- if (NULL != socks5_client_version_check(obj))
- return -1;
- /* Length of u8 version IN [5] */
- result += 1;
- /* Length of u8 n_methods */
- result += 1;
- /* Length of u8 methods[n_methods] */
- result += TRUNNEL_DYNARRAY_LEN(&obj->methods);
- return result;
- }
- int
- socks5_client_version_clear_errors(socks5_client_version_t *obj)
- {
- int r = obj->trunnel_error_code_;
- obj->trunnel_error_code_ = 0;
- return r;
- }
- ssize_t
- socks5_client_version_encode(uint8_t *output, const size_t avail, const socks5_client_version_t *obj)
- {
- ssize_t result = 0;
- size_t written = 0;
- uint8_t *ptr = output;
- const char *msg;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- const ssize_t encoded_len = socks5_client_version_encoded_len(obj);
- #endif
- if (NULL != (msg = socks5_client_version_check(obj)))
- goto check_failed;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- trunnel_assert(encoded_len >= 0);
- #endif
- /* Encode u8 version IN [5] */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->version));
- written += 1; ptr += 1;
- /* Encode u8 n_methods */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->n_methods));
- written += 1; ptr += 1;
- /* Encode u8 methods[n_methods] */
- {
- size_t elt_len = TRUNNEL_DYNARRAY_LEN(&obj->methods);
- trunnel_assert(obj->n_methods == elt_len);
- trunnel_assert(written <= avail);
- if (avail - written < elt_len)
- goto truncated;
- if (elt_len)
- memcpy(ptr, obj->methods.elts_, elt_len);
- written += elt_len; ptr += elt_len;
- }
- trunnel_assert(ptr == output + written);
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- {
- trunnel_assert(encoded_len >= 0);
- trunnel_assert((size_t)encoded_len == written);
- }
- #endif
- return written;
- truncated:
- result = -2;
- goto fail;
- check_failed:
- (void)msg;
- result = -1;
- goto fail;
- fail:
- trunnel_assert(result < 0);
- return result;
- }
- /** As socks5_client_version_parse(), but do not allocate the output
- * object.
- */
- static ssize_t
- socks5_client_version_parse_into(socks5_client_version_t *obj, const uint8_t *input, const size_t len_in)
- {
- const uint8_t *ptr = input;
- size_t remaining = len_in;
- ssize_t result = 0;
- (void)result;
- /* Parse u8 version IN [5] */
- CHECK_REMAINING(1, truncated);
- obj->version = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- if (! (obj->version == 5))
- goto fail;
- /* Parse u8 n_methods */
- CHECK_REMAINING(1, truncated);
- obj->n_methods = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- /* Parse u8 methods[n_methods] */
- CHECK_REMAINING(obj->n_methods, truncated);
- TRUNNEL_DYNARRAY_EXPAND(uint8_t, &obj->methods, obj->n_methods, {});
- obj->methods.n_ = obj->n_methods;
- if (obj->n_methods)
- memcpy(obj->methods.elts_, ptr, obj->n_methods);
- ptr += obj->n_methods; remaining -= obj->n_methods;
- trunnel_assert(ptr + remaining == input + len_in);
- return len_in - remaining;
- truncated:
- return -2;
- trunnel_alloc_failed:
- return -1;
- fail:
- result = -1;
- return result;
- }
- ssize_t
- socks5_client_version_parse(socks5_client_version_t **output, const uint8_t *input, const size_t len_in)
- {
- ssize_t result;
- *output = socks5_client_version_new();
- if (NULL == *output)
- return -1;
- result = socks5_client_version_parse_into(*output, input, len_in);
- if (result < 0) {
- socks5_client_version_free(*output);
- *output = NULL;
- }
- return result;
- }
- socks5_server_method_t *
- socks5_server_method_new(void)
- {
- socks5_server_method_t *val = trunnel_calloc(1, sizeof(socks5_server_method_t));
- if (NULL == val)
- return NULL;
- val->version = 5;
- return val;
- }
- /** Release all storage held inside 'obj', but do not free 'obj'.
- */
- static void
- socks5_server_method_clear(socks5_server_method_t *obj)
- {
- (void) obj;
- }
- void
- socks5_server_method_free(socks5_server_method_t *obj)
- {
- if (obj == NULL)
- return;
- socks5_server_method_clear(obj);
- trunnel_memwipe(obj, sizeof(socks5_server_method_t));
- trunnel_free_(obj);
- }
- uint8_t
- socks5_server_method_get_version(const socks5_server_method_t *inp)
- {
- return inp->version;
- }
- int
- socks5_server_method_set_version(socks5_server_method_t *inp, uint8_t val)
- {
- if (! ((val == 5))) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- inp->version = val;
- return 0;
- }
- uint8_t
- socks5_server_method_get_method(const socks5_server_method_t *inp)
- {
- return inp->method;
- }
- int
- socks5_server_method_set_method(socks5_server_method_t *inp, uint8_t val)
- {
- inp->method = val;
- return 0;
- }
- const char *
- socks5_server_method_check(const socks5_server_method_t *obj)
- {
- if (obj == NULL)
- return "Object was NULL";
- if (obj->trunnel_error_code_)
- return "A set function failed on this object";
- if (! (obj->version == 5))
- return "Integer out of bounds";
- return NULL;
- }
- ssize_t
- socks5_server_method_encoded_len(const socks5_server_method_t *obj)
- {
- ssize_t result = 0;
- if (NULL != socks5_server_method_check(obj))
- return -1;
- /* Length of u8 version IN [5] */
- result += 1;
- /* Length of u8 method */
- result += 1;
- return result;
- }
- int
- socks5_server_method_clear_errors(socks5_server_method_t *obj)
- {
- int r = obj->trunnel_error_code_;
- obj->trunnel_error_code_ = 0;
- return r;
- }
- ssize_t
- socks5_server_method_encode(uint8_t *output, const size_t avail, const socks5_server_method_t *obj)
- {
- ssize_t result = 0;
- size_t written = 0;
- uint8_t *ptr = output;
- const char *msg;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- const ssize_t encoded_len = socks5_server_method_encoded_len(obj);
- #endif
- if (NULL != (msg = socks5_server_method_check(obj)))
- goto check_failed;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- trunnel_assert(encoded_len >= 0);
- #endif
- /* Encode u8 version IN [5] */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->version));
- written += 1; ptr += 1;
- /* Encode u8 method */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->method));
- written += 1; ptr += 1;
- trunnel_assert(ptr == output + written);
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- {
- trunnel_assert(encoded_len >= 0);
- trunnel_assert((size_t)encoded_len == written);
- }
- #endif
- return written;
- truncated:
- result = -2;
- goto fail;
- check_failed:
- (void)msg;
- result = -1;
- goto fail;
- fail:
- trunnel_assert(result < 0);
- return result;
- }
- /** As socks5_server_method_parse(), but do not allocate the output
- * object.
- */
- static ssize_t
- socks5_server_method_parse_into(socks5_server_method_t *obj, const uint8_t *input, const size_t len_in)
- {
- const uint8_t *ptr = input;
- size_t remaining = len_in;
- ssize_t result = 0;
- (void)result;
- /* Parse u8 version IN [5] */
- CHECK_REMAINING(1, truncated);
- obj->version = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- if (! (obj->version == 5))
- goto fail;
- /* Parse u8 method */
- CHECK_REMAINING(1, truncated);
- obj->method = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- trunnel_assert(ptr + remaining == input + len_in);
- return len_in - remaining;
- truncated:
- return -2;
- fail:
- result = -1;
- return result;
- }
- ssize_t
- socks5_server_method_parse(socks5_server_method_t **output, const uint8_t *input, const size_t len_in)
- {
- ssize_t result;
- *output = socks5_server_method_new();
- if (NULL == *output)
- return -1;
- result = socks5_server_method_parse_into(*output, input, len_in);
- if (result < 0) {
- socks5_server_method_free(*output);
- *output = NULL;
- }
- return result;
- }
- socks5_server_userpass_auth_t *
- socks5_server_userpass_auth_new(void)
- {
- socks5_server_userpass_auth_t *val = trunnel_calloc(1, sizeof(socks5_server_userpass_auth_t));
- if (NULL == val)
- return NULL;
- val->version = 1;
- return val;
- }
- /** Release all storage held inside 'obj', but do not free 'obj'.
- */
- static void
- socks5_server_userpass_auth_clear(socks5_server_userpass_auth_t *obj)
- {
- (void) obj;
- }
- void
- socks5_server_userpass_auth_free(socks5_server_userpass_auth_t *obj)
- {
- if (obj == NULL)
- return;
- socks5_server_userpass_auth_clear(obj);
- trunnel_memwipe(obj, sizeof(socks5_server_userpass_auth_t));
- trunnel_free_(obj);
- }
- uint8_t
- socks5_server_userpass_auth_get_version(const socks5_server_userpass_auth_t *inp)
- {
- return inp->version;
- }
- int
- socks5_server_userpass_auth_set_version(socks5_server_userpass_auth_t *inp, uint8_t val)
- {
- if (! ((val == 1))) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- inp->version = val;
- return 0;
- }
- uint8_t
- socks5_server_userpass_auth_get_status(const socks5_server_userpass_auth_t *inp)
- {
- return inp->status;
- }
- int
- socks5_server_userpass_auth_set_status(socks5_server_userpass_auth_t *inp, uint8_t val)
- {
- inp->status = val;
- return 0;
- }
- const char *
- socks5_server_userpass_auth_check(const socks5_server_userpass_auth_t *obj)
- {
- if (obj == NULL)
- return "Object was NULL";
- if (obj->trunnel_error_code_)
- return "A set function failed on this object";
- if (! (obj->version == 1))
- return "Integer out of bounds";
- return NULL;
- }
- ssize_t
- socks5_server_userpass_auth_encoded_len(const socks5_server_userpass_auth_t *obj)
- {
- ssize_t result = 0;
- if (NULL != socks5_server_userpass_auth_check(obj))
- return -1;
- /* Length of u8 version IN [1] */
- result += 1;
- /* Length of u8 status */
- result += 1;
- return result;
- }
- int
- socks5_server_userpass_auth_clear_errors(socks5_server_userpass_auth_t *obj)
- {
- int r = obj->trunnel_error_code_;
- obj->trunnel_error_code_ = 0;
- return r;
- }
- ssize_t
- socks5_server_userpass_auth_encode(uint8_t *output, const size_t avail, const socks5_server_userpass_auth_t *obj)
- {
- ssize_t result = 0;
- size_t written = 0;
- uint8_t *ptr = output;
- const char *msg;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- const ssize_t encoded_len = socks5_server_userpass_auth_encoded_len(obj);
- #endif
- if (NULL != (msg = socks5_server_userpass_auth_check(obj)))
- goto check_failed;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- trunnel_assert(encoded_len >= 0);
- #endif
- /* Encode u8 version IN [1] */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->version));
- written += 1; ptr += 1;
- /* Encode u8 status */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->status));
- written += 1; ptr += 1;
- trunnel_assert(ptr == output + written);
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- {
- trunnel_assert(encoded_len >= 0);
- trunnel_assert((size_t)encoded_len == written);
- }
- #endif
- return written;
- truncated:
- result = -2;
- goto fail;
- check_failed:
- (void)msg;
- result = -1;
- goto fail;
- fail:
- trunnel_assert(result < 0);
- return result;
- }
- /** As socks5_server_userpass_auth_parse(), but do not allocate the
- * output object.
- */
- static ssize_t
- socks5_server_userpass_auth_parse_into(socks5_server_userpass_auth_t *obj, const uint8_t *input, const size_t len_in)
- {
- const uint8_t *ptr = input;
- size_t remaining = len_in;
- ssize_t result = 0;
- (void)result;
- /* Parse u8 version IN [1] */
- CHECK_REMAINING(1, truncated);
- obj->version = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- if (! (obj->version == 1))
- goto fail;
- /* Parse u8 status */
- CHECK_REMAINING(1, truncated);
- obj->status = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- trunnel_assert(ptr + remaining == input + len_in);
- return len_in - remaining;
- truncated:
- return -2;
- fail:
- result = -1;
- return result;
- }
- ssize_t
- socks5_server_userpass_auth_parse(socks5_server_userpass_auth_t **output, const uint8_t *input, const size_t len_in)
- {
- ssize_t result;
- *output = socks5_server_userpass_auth_new();
- if (NULL == *output)
- return -1;
- result = socks5_server_userpass_auth_parse_into(*output, input, len_in);
- if (result < 0) {
- socks5_server_userpass_auth_free(*output);
- *output = NULL;
- }
- return result;
- }
- socks5_client_request_t *
- socks5_client_request_new(void)
- {
- socks5_client_request_t *val = trunnel_calloc(1, sizeof(socks5_client_request_t));
- if (NULL == val)
- return NULL;
- val->version = 5;
- val->command = CMD_BIND;
- return val;
- }
- /** Release all storage held inside 'obj', but do not free 'obj'.
- */
- static void
- socks5_client_request_clear(socks5_client_request_t *obj)
- {
- (void) obj;
- domainname_free(obj->dest_addr_domainname);
- obj->dest_addr_domainname = NULL;
- }
- void
- socks5_client_request_free(socks5_client_request_t *obj)
- {
- if (obj == NULL)
- return;
- socks5_client_request_clear(obj);
- trunnel_memwipe(obj, sizeof(socks5_client_request_t));
- trunnel_free_(obj);
- }
- uint8_t
- socks5_client_request_get_version(const socks5_client_request_t *inp)
- {
- return inp->version;
- }
- int
- socks5_client_request_set_version(socks5_client_request_t *inp, uint8_t val)
- {
- if (! ((val == 5))) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- inp->version = val;
- return 0;
- }
- uint8_t
- socks5_client_request_get_command(const socks5_client_request_t *inp)
- {
- return inp->command;
- }
- int
- socks5_client_request_set_command(socks5_client_request_t *inp, uint8_t val)
- {
- if (! ((val == CMD_BIND || val == CMD_CONNECT || val == CMD_RESOLVE || val == CMD_RESOLVE_PTR || val == CMD_UDP_ASSOCIATE))) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- inp->command = val;
- return 0;
- }
- uint8_t
- socks5_client_request_get_reserved(const socks5_client_request_t *inp)
- {
- return inp->reserved;
- }
- int
- socks5_client_request_set_reserved(socks5_client_request_t *inp, uint8_t val)
- {
- if (! ((val == 0))) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- inp->reserved = val;
- return 0;
- }
- uint8_t
- socks5_client_request_get_atype(const socks5_client_request_t *inp)
- {
- return inp->atype;
- }
- int
- socks5_client_request_set_atype(socks5_client_request_t *inp, uint8_t val)
- {
- inp->atype = val;
- return 0;
- }
- uint32_t
- socks5_client_request_get_dest_addr_ipv4(const socks5_client_request_t *inp)
- {
- return inp->dest_addr_ipv4;
- }
- int
- socks5_client_request_set_dest_addr_ipv4(socks5_client_request_t *inp, uint32_t val)
- {
- inp->dest_addr_ipv4 = val;
- return 0;
- }
- size_t
- socks5_client_request_getlen_dest_addr_ipv6(const socks5_client_request_t *inp)
- {
- (void)inp; return 16;
- }
- uint8_t
- socks5_client_request_get_dest_addr_ipv6(socks5_client_request_t *inp, size_t idx)
- {
- trunnel_assert(idx < 16);
- return inp->dest_addr_ipv6[idx];
- }
- uint8_t
- socks5_client_request_getconst_dest_addr_ipv6(const socks5_client_request_t *inp, size_t idx)
- {
- return socks5_client_request_get_dest_addr_ipv6((socks5_client_request_t*)inp, idx);
- }
- int
- socks5_client_request_set_dest_addr_ipv6(socks5_client_request_t *inp, size_t idx, uint8_t elt)
- {
- trunnel_assert(idx < 16);
- inp->dest_addr_ipv6[idx] = elt;
- return 0;
- }
- uint8_t *
- socks5_client_request_getarray_dest_addr_ipv6(socks5_client_request_t *inp)
- {
- return inp->dest_addr_ipv6;
- }
- const uint8_t *
- socks5_client_request_getconstarray_dest_addr_ipv6(const socks5_client_request_t *inp)
- {
- return (const uint8_t *)socks5_client_request_getarray_dest_addr_ipv6((socks5_client_request_t*)inp);
- }
- struct domainname_st *
- socks5_client_request_get_dest_addr_domainname(socks5_client_request_t *inp)
- {
- return inp->dest_addr_domainname;
- }
- const struct domainname_st *
- socks5_client_request_getconst_dest_addr_domainname(const socks5_client_request_t *inp)
- {
- return socks5_client_request_get_dest_addr_domainname((socks5_client_request_t*) inp);
- }
- int
- socks5_client_request_set_dest_addr_domainname(socks5_client_request_t *inp, struct domainname_st *val)
- {
- if (inp->dest_addr_domainname && inp->dest_addr_domainname != val)
- domainname_free(inp->dest_addr_domainname);
- return socks5_client_request_set0_dest_addr_domainname(inp, val);
- }
- int
- socks5_client_request_set0_dest_addr_domainname(socks5_client_request_t *inp, struct domainname_st *val)
- {
- inp->dest_addr_domainname = val;
- return 0;
- }
- uint16_t
- socks5_client_request_get_dest_port(const socks5_client_request_t *inp)
- {
- return inp->dest_port;
- }
- int
- socks5_client_request_set_dest_port(socks5_client_request_t *inp, uint16_t val)
- {
- inp->dest_port = val;
- return 0;
- }
- const char *
- socks5_client_request_check(const socks5_client_request_t *obj)
- {
- if (obj == NULL)
- return "Object was NULL";
- if (obj->trunnel_error_code_)
- return "A set function failed on this object";
- if (! (obj->version == 5))
- return "Integer out of bounds";
- if (! (obj->command == CMD_BIND || obj->command == CMD_CONNECT || obj->command == CMD_RESOLVE || obj->command == CMD_RESOLVE_PTR || obj->command == CMD_UDP_ASSOCIATE))
- return "Integer out of bounds";
- if (! (obj->reserved == 0))
- return "Integer out of bounds";
- switch (obj->atype) {
- case ATYPE_IPV4:
- break;
- case ATYPE_IPV6:
- break;
- case ATYPE_DOMAINNAME:
- {
- const char *msg;
- if (NULL != (msg = domainname_check(obj->dest_addr_domainname)))
- return msg;
- }
- break;
- default:
- return "Bad tag for union";
- break;
- }
- return NULL;
- }
- ssize_t
- socks5_client_request_encoded_len(const socks5_client_request_t *obj)
- {
- ssize_t result = 0;
- if (NULL != socks5_client_request_check(obj))
- return -1;
- /* Length of u8 version IN [5] */
- result += 1;
- /* Length of u8 command IN [CMD_BIND, CMD_CONNECT, CMD_RESOLVE, CMD_RESOLVE_PTR, CMD_UDP_ASSOCIATE] */
- result += 1;
- /* Length of u8 reserved IN [0] */
- result += 1;
- /* Length of u8 atype */
- result += 1;
- switch (obj->atype) {
- case ATYPE_IPV4:
- /* Length of u32 dest_addr_ipv4 */
- result += 4;
- break;
- case ATYPE_IPV6:
- /* Length of u8 dest_addr_ipv6[16] */
- result += 16;
- break;
- case ATYPE_DOMAINNAME:
- /* Length of struct domainname dest_addr_domainname */
- result += domainname_encoded_len(obj->dest_addr_domainname);
- break;
- default:
- trunnel_assert(0);
- break;
- }
- /* Length of u16 dest_port */
- result += 2;
- return result;
- }
- int
- socks5_client_request_clear_errors(socks5_client_request_t *obj)
- {
- int r = obj->trunnel_error_code_;
- obj->trunnel_error_code_ = 0;
- return r;
- }
- ssize_t
- socks5_client_request_encode(uint8_t *output, const size_t avail, const socks5_client_request_t *obj)
- {
- ssize_t result = 0;
- size_t written = 0;
- uint8_t *ptr = output;
- const char *msg;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- const ssize_t encoded_len = socks5_client_request_encoded_len(obj);
- #endif
- if (NULL != (msg = socks5_client_request_check(obj)))
- goto check_failed;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- trunnel_assert(encoded_len >= 0);
- #endif
- /* Encode u8 version IN [5] */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->version));
- written += 1; ptr += 1;
- /* Encode u8 command IN [CMD_BIND, CMD_CONNECT, CMD_RESOLVE, CMD_RESOLVE_PTR, CMD_UDP_ASSOCIATE] */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->command));
- written += 1; ptr += 1;
- /* Encode u8 reserved IN [0] */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->reserved));
- written += 1; ptr += 1;
- /* Encode u8 atype */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->atype));
- written += 1; ptr += 1;
- /* Encode union dest_addr[atype] */
- trunnel_assert(written <= avail);
- switch (obj->atype) {
- case ATYPE_IPV4:
- /* Encode u32 dest_addr_ipv4 */
- trunnel_assert(written <= avail);
- if (avail - written < 4)
- goto truncated;
- trunnel_set_uint32(ptr, trunnel_htonl(obj->dest_addr_ipv4));
- written += 4; ptr += 4;
- break;
- case ATYPE_IPV6:
- /* Encode u8 dest_addr_ipv6[16] */
- trunnel_assert(written <= avail);
- if (avail - written < 16)
- goto truncated;
- memcpy(ptr, obj->dest_addr_ipv6, 16);
- written += 16; ptr += 16;
- break;
- case ATYPE_DOMAINNAME:
- /* Encode struct domainname dest_addr_domainname */
- trunnel_assert(written <= avail);
- result = domainname_encode(ptr, avail - written, obj->dest_addr_domainname);
- if (result < 0)
- goto fail; /* XXXXXXX !*/
- written += result; ptr += result;
- break;
- default:
- trunnel_assert(0);
- break;
- }
- /* Encode u16 dest_port */
- trunnel_assert(written <= avail);
- if (avail - written < 2)
- goto truncated;
- trunnel_set_uint16(ptr, trunnel_htons(obj->dest_port));
- written += 2; ptr += 2;
- trunnel_assert(ptr == output + written);
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- {
- trunnel_assert(encoded_len >= 0);
- trunnel_assert((size_t)encoded_len == written);
- }
- #endif
- return written;
- truncated:
- result = -2;
- goto fail;
- check_failed:
- (void)msg;
- result = -1;
- goto fail;
- fail:
- trunnel_assert(result < 0);
- return result;
- }
- /** As socks5_client_request_parse(), but do not allocate the output
- * object.
- */
- static ssize_t
- socks5_client_request_parse_into(socks5_client_request_t *obj, const uint8_t *input, const size_t len_in)
- {
- const uint8_t *ptr = input;
- size_t remaining = len_in;
- ssize_t result = 0;
- (void)result;
- /* Parse u8 version IN [5] */
- CHECK_REMAINING(1, truncated);
- obj->version = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- if (! (obj->version == 5))
- goto fail;
- /* Parse u8 command IN [CMD_BIND, CMD_CONNECT, CMD_RESOLVE, CMD_RESOLVE_PTR, CMD_UDP_ASSOCIATE] */
- CHECK_REMAINING(1, truncated);
- obj->command = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- if (! (obj->command == CMD_BIND || obj->command == CMD_CONNECT || obj->command == CMD_RESOLVE || obj->command == CMD_RESOLVE_PTR || obj->command == CMD_UDP_ASSOCIATE))
- goto fail;
- /* Parse u8 reserved IN [0] */
- CHECK_REMAINING(1, truncated);
- obj->reserved = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- if (! (obj->reserved == 0))
- goto fail;
- /* Parse u8 atype */
- CHECK_REMAINING(1, truncated);
- obj->atype = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- /* Parse union dest_addr[atype] */
- switch (obj->atype) {
- case ATYPE_IPV4:
- /* Parse u32 dest_addr_ipv4 */
- CHECK_REMAINING(4, truncated);
- obj->dest_addr_ipv4 = trunnel_ntohl(trunnel_get_uint32(ptr));
- remaining -= 4; ptr += 4;
- break;
- case ATYPE_IPV6:
- /* Parse u8 dest_addr_ipv6[16] */
- CHECK_REMAINING(16, truncated);
- memcpy(obj->dest_addr_ipv6, ptr, 16);
- remaining -= 16; ptr += 16;
- break;
- case ATYPE_DOMAINNAME:
- /* Parse struct domainname dest_addr_domainname */
- result = domainname_parse(&obj->dest_addr_domainname, ptr, remaining);
- if (result < 0)
- goto relay_fail;
- trunnel_assert((size_t)result <= remaining);
- remaining -= result; ptr += result;
- break;
- default:
- goto fail;
- break;
- }
- /* Parse u16 dest_port */
- CHECK_REMAINING(2, truncated);
- obj->dest_port = trunnel_ntohs(trunnel_get_uint16(ptr));
- remaining -= 2; ptr += 2;
- trunnel_assert(ptr + remaining == input + len_in);
- return len_in - remaining;
- truncated:
- return -2;
- relay_fail:
- trunnel_assert(result < 0);
- return result;
- fail:
- result = -1;
- return result;
- }
- ssize_t
- socks5_client_request_parse(socks5_client_request_t **output, const uint8_t *input, const size_t len_in)
- {
- ssize_t result;
- *output = socks5_client_request_new();
- if (NULL == *output)
- return -1;
- result = socks5_client_request_parse_into(*output, input, len_in);
- if (result < 0) {
- socks5_client_request_free(*output);
- *output = NULL;
- }
- return result;
- }
- socks5_server_reply_t *
- socks5_server_reply_new(void)
- {
- socks5_server_reply_t *val = trunnel_calloc(1, sizeof(socks5_server_reply_t));
- if (NULL == val)
- return NULL;
- val->version = 5;
- return val;
- }
- /** Release all storage held inside 'obj', but do not free 'obj'.
- */
- static void
- socks5_server_reply_clear(socks5_server_reply_t *obj)
- {
- (void) obj;
- domainname_free(obj->bind_addr_domainname);
- obj->bind_addr_domainname = NULL;
- }
- void
- socks5_server_reply_free(socks5_server_reply_t *obj)
- {
- if (obj == NULL)
- return;
- socks5_server_reply_clear(obj);
- trunnel_memwipe(obj, sizeof(socks5_server_reply_t));
- trunnel_free_(obj);
- }
- uint8_t
- socks5_server_reply_get_version(const socks5_server_reply_t *inp)
- {
- return inp->version;
- }
- int
- socks5_server_reply_set_version(socks5_server_reply_t *inp, uint8_t val)
- {
- if (! ((val == 5))) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- inp->version = val;
- return 0;
- }
- uint8_t
- socks5_server_reply_get_reply(const socks5_server_reply_t *inp)
- {
- return inp->reply;
- }
- int
- socks5_server_reply_set_reply(socks5_server_reply_t *inp, uint8_t val)
- {
- inp->reply = val;
- return 0;
- }
- uint8_t
- socks5_server_reply_get_reserved(const socks5_server_reply_t *inp)
- {
- return inp->reserved;
- }
- int
- socks5_server_reply_set_reserved(socks5_server_reply_t *inp, uint8_t val)
- {
- if (! ((val == 0))) {
- TRUNNEL_SET_ERROR_CODE(inp);
- return -1;
- }
- inp->reserved = val;
- return 0;
- }
- uint8_t
- socks5_server_reply_get_atype(const socks5_server_reply_t *inp)
- {
- return inp->atype;
- }
- int
- socks5_server_reply_set_atype(socks5_server_reply_t *inp, uint8_t val)
- {
- inp->atype = val;
- return 0;
- }
- uint32_t
- socks5_server_reply_get_bind_addr_ipv4(const socks5_server_reply_t *inp)
- {
- return inp->bind_addr_ipv4;
- }
- int
- socks5_server_reply_set_bind_addr_ipv4(socks5_server_reply_t *inp, uint32_t val)
- {
- inp->bind_addr_ipv4 = val;
- return 0;
- }
- size_t
- socks5_server_reply_getlen_bind_addr_ipv6(const socks5_server_reply_t *inp)
- {
- (void)inp; return 16;
- }
- uint8_t
- socks5_server_reply_get_bind_addr_ipv6(socks5_server_reply_t *inp, size_t idx)
- {
- trunnel_assert(idx < 16);
- return inp->bind_addr_ipv6[idx];
- }
- uint8_t
- socks5_server_reply_getconst_bind_addr_ipv6(const socks5_server_reply_t *inp, size_t idx)
- {
- return socks5_server_reply_get_bind_addr_ipv6((socks5_server_reply_t*)inp, idx);
- }
- int
- socks5_server_reply_set_bind_addr_ipv6(socks5_server_reply_t *inp, size_t idx, uint8_t elt)
- {
- trunnel_assert(idx < 16);
- inp->bind_addr_ipv6[idx] = elt;
- return 0;
- }
- uint8_t *
- socks5_server_reply_getarray_bind_addr_ipv6(socks5_server_reply_t *inp)
- {
- return inp->bind_addr_ipv6;
- }
- const uint8_t *
- socks5_server_reply_getconstarray_bind_addr_ipv6(const socks5_server_reply_t *inp)
- {
- return (const uint8_t *)socks5_server_reply_getarray_bind_addr_ipv6((socks5_server_reply_t*)inp);
- }
- struct domainname_st *
- socks5_server_reply_get_bind_addr_domainname(socks5_server_reply_t *inp)
- {
- return inp->bind_addr_domainname;
- }
- const struct domainname_st *
- socks5_server_reply_getconst_bind_addr_domainname(const socks5_server_reply_t *inp)
- {
- return socks5_server_reply_get_bind_addr_domainname((socks5_server_reply_t*) inp);
- }
- int
- socks5_server_reply_set_bind_addr_domainname(socks5_server_reply_t *inp, struct domainname_st *val)
- {
- if (inp->bind_addr_domainname && inp->bind_addr_domainname != val)
- domainname_free(inp->bind_addr_domainname);
- return socks5_server_reply_set0_bind_addr_domainname(inp, val);
- }
- int
- socks5_server_reply_set0_bind_addr_domainname(socks5_server_reply_t *inp, struct domainname_st *val)
- {
- inp->bind_addr_domainname = val;
- return 0;
- }
- uint16_t
- socks5_server_reply_get_bind_port(const socks5_server_reply_t *inp)
- {
- return inp->bind_port;
- }
- int
- socks5_server_reply_set_bind_port(socks5_server_reply_t *inp, uint16_t val)
- {
- inp->bind_port = val;
- return 0;
- }
- const char *
- socks5_server_reply_check(const socks5_server_reply_t *obj)
- {
- if (obj == NULL)
- return "Object was NULL";
- if (obj->trunnel_error_code_)
- return "A set function failed on this object";
- if (! (obj->version == 5))
- return "Integer out of bounds";
- if (! (obj->reserved == 0))
- return "Integer out of bounds";
- switch (obj->atype) {
- case ATYPE_IPV4:
- break;
- case ATYPE_IPV6:
- break;
- case ATYPE_DOMAINNAME:
- {
- const char *msg;
- if (NULL != (msg = domainname_check(obj->bind_addr_domainname)))
- return msg;
- }
- break;
- default:
- return "Bad tag for union";
- break;
- }
- return NULL;
- }
- ssize_t
- socks5_server_reply_encoded_len(const socks5_server_reply_t *obj)
- {
- ssize_t result = 0;
- if (NULL != socks5_server_reply_check(obj))
- return -1;
- /* Length of u8 version IN [5] */
- result += 1;
- /* Length of u8 reply */
- result += 1;
- /* Length of u8 reserved IN [0] */
- result += 1;
- /* Length of u8 atype */
- result += 1;
- switch (obj->atype) {
- case ATYPE_IPV4:
- /* Length of u32 bind_addr_ipv4 */
- result += 4;
- break;
- case ATYPE_IPV6:
- /* Length of u8 bind_addr_ipv6[16] */
- result += 16;
- break;
- case ATYPE_DOMAINNAME:
- /* Length of struct domainname bind_addr_domainname */
- result += domainname_encoded_len(obj->bind_addr_domainname);
- break;
- default:
- trunnel_assert(0);
- break;
- }
- /* Length of u16 bind_port */
- result += 2;
- return result;
- }
- int
- socks5_server_reply_clear_errors(socks5_server_reply_t *obj)
- {
- int r = obj->trunnel_error_code_;
- obj->trunnel_error_code_ = 0;
- return r;
- }
- ssize_t
- socks5_server_reply_encode(uint8_t *output, const size_t avail, const socks5_server_reply_t *obj)
- {
- ssize_t result = 0;
- size_t written = 0;
- uint8_t *ptr = output;
- const char *msg;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- const ssize_t encoded_len = socks5_server_reply_encoded_len(obj);
- #endif
- if (NULL != (msg = socks5_server_reply_check(obj)))
- goto check_failed;
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- trunnel_assert(encoded_len >= 0);
- #endif
- /* Encode u8 version IN [5] */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->version));
- written += 1; ptr += 1;
- /* Encode u8 reply */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->reply));
- written += 1; ptr += 1;
- /* Encode u8 reserved IN [0] */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->reserved));
- written += 1; ptr += 1;
- /* Encode u8 atype */
- trunnel_assert(written <= avail);
- if (avail - written < 1)
- goto truncated;
- trunnel_set_uint8(ptr, (obj->atype));
- written += 1; ptr += 1;
- /* Encode union bind_addr[atype] */
- trunnel_assert(written <= avail);
- switch (obj->atype) {
- case ATYPE_IPV4:
- /* Encode u32 bind_addr_ipv4 */
- trunnel_assert(written <= avail);
- if (avail - written < 4)
- goto truncated;
- trunnel_set_uint32(ptr, trunnel_htonl(obj->bind_addr_ipv4));
- written += 4; ptr += 4;
- break;
- case ATYPE_IPV6:
- /* Encode u8 bind_addr_ipv6[16] */
- trunnel_assert(written <= avail);
- if (avail - written < 16)
- goto truncated;
- memcpy(ptr, obj->bind_addr_ipv6, 16);
- written += 16; ptr += 16;
- break;
- case ATYPE_DOMAINNAME:
- /* Encode struct domainname bind_addr_domainname */
- trunnel_assert(written <= avail);
- result = domainname_encode(ptr, avail - written, obj->bind_addr_domainname);
- if (result < 0)
- goto fail; /* XXXXXXX !*/
- written += result; ptr += result;
- break;
- default:
- trunnel_assert(0);
- break;
- }
- /* Encode u16 bind_port */
- trunnel_assert(written <= avail);
- if (avail - written < 2)
- goto truncated;
- trunnel_set_uint16(ptr, trunnel_htons(obj->bind_port));
- written += 2; ptr += 2;
- trunnel_assert(ptr == output + written);
- #ifdef TRUNNEL_CHECK_ENCODED_LEN
- {
- trunnel_assert(encoded_len >= 0);
- trunnel_assert((size_t)encoded_len == written);
- }
- #endif
- return written;
- truncated:
- result = -2;
- goto fail;
- check_failed:
- (void)msg;
- result = -1;
- goto fail;
- fail:
- trunnel_assert(result < 0);
- return result;
- }
- /** As socks5_server_reply_parse(), but do not allocate the output
- * object.
- */
- static ssize_t
- socks5_server_reply_parse_into(socks5_server_reply_t *obj, const uint8_t *input, const size_t len_in)
- {
- const uint8_t *ptr = input;
- size_t remaining = len_in;
- ssize_t result = 0;
- (void)result;
- /* Parse u8 version IN [5] */
- CHECK_REMAINING(1, truncated);
- obj->version = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- if (! (obj->version == 5))
- goto fail;
- /* Parse u8 reply */
- CHECK_REMAINING(1, truncated);
- obj->reply = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- /* Parse u8 reserved IN [0] */
- CHECK_REMAINING(1, truncated);
- obj->reserved = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- if (! (obj->reserved == 0))
- goto fail;
- /* Parse u8 atype */
- CHECK_REMAINING(1, truncated);
- obj->atype = (trunnel_get_uint8(ptr));
- remaining -= 1; ptr += 1;
- /* Parse union bind_addr[atype] */
- switch (obj->atype) {
- case ATYPE_IPV4:
- /* Parse u32 bind_addr_ipv4 */
- CHECK_REMAINING(4, truncated);
- obj->bind_addr_ipv4 = trunnel_ntohl(trunnel_get_uint32(ptr));
- remaining -= 4; ptr += 4;
- break;
- case ATYPE_IPV6:
- /* Parse u8 bind_addr_ipv6[16] */
- CHECK_REMAINING(16, truncated);
- memcpy(obj->bind_addr_ipv6, ptr, 16);
- remaining -= 16; ptr += 16;
- break;
- case ATYPE_DOMAINNAME:
- /* Parse struct domainname bind_addr_domainname */
- result = domainname_parse(&obj->bind_addr_domainname, ptr, remaining);
- if (result < 0)
- goto relay_fail;
- trunnel_assert((size_t)result <= remaining);
- remaining -= result; ptr += result;
- break;
- default:
- goto fail;
- break;
- }
- /* Parse u16 bind_port */
- CHECK_REMAINING(2, truncated);
- obj->bind_port = trunnel_ntohs(trunnel_get_uint16(ptr));
- remaining -= 2; ptr += 2;
- trunnel_assert(ptr + remaining == input + len_in);
- return len_in - remaining;
- truncated:
- return -2;
- relay_fail:
- trunnel_assert(result < 0);
- return result;
- fail:
- result = -1;
- return result;
- }
- ssize_t
- socks5_server_reply_parse(socks5_server_reply_t **output, const uint8_t *input, const size_t len_in)
- {
- ssize_t result;
- *output = socks5_server_reply_new();
- if (NULL == *output)
- return -1;
- result = socks5_server_reply_parse_into(*output, input, len_in);
- if (result < 0) {
- socks5_server_reply_free(*output);
- *output = NULL;
- }
- return result;
- }
|