12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717 |
- /* Copyright 2004-2005 Roger Dingledine, Nick Mathewson. */
- /* See LICENSE for licensing information */
- /* $Id$ */
- const char control_c_id[] =
- "$Id$";
- /**
- * \file control.c
- * \brief Implementation for Tor's control-socket interface.
- **/
- #include "or.h"
- #define STATE_IS_OPEN(s) ((s) == CONTROL_CONN_STATE_OPEN_V0 || \
- (s) == CONTROL_CONN_STATE_OPEN_V1)
- #define STATE_IS_V0(s) ((s) == CONTROL_CONN_STATE_NEEDAUTH_V0 || \
- (s) == CONTROL_CONN_STATE_OPEN_V0)
- /*
- * See control-spec.txt and control-spec-v0.txt for full details on
- * protocol(s).
- *
- */
- /* Recognized message type codes. */
- #define CONTROL0_CMD_ERROR 0x0000
- #define CONTROL0_CMD_DONE 0x0001
- #define CONTROL0_CMD_SETCONF 0x0002
- #define CONTROL0_CMD_GETCONF 0x0003
- #define CONTROL0_CMD_CONFVALUE 0x0004
- #define CONTROL0_CMD_SETEVENTS 0x0005
- #define CONTROL0_CMD_EVENT 0x0006
- #define CONTROL0_CMD_AUTHENTICATE 0x0007
- #define CONTROL0_CMD_SAVECONF 0x0008
- #define CONTROL0_CMD_SIGNAL 0x0009
- #define CONTROL0_CMD_MAPADDRESS 0x000A
- #define CONTROL0_CMD_GETINFO 0x000B
- #define CONTROL0_CMD_INFOVALUE 0x000C
- #define CONTROL0_CMD_EXTENDCIRCUIT 0x000D
- #define CONTROL0_CMD_ATTACHSTREAM 0x000E
- #define CONTROL0_CMD_POSTDESCRIPTOR 0x000F
- #define CONTROL0_CMD_FRAGMENTHEADER 0x0010
- #define CONTROL0_CMD_FRAGMENT 0x0011
- #define CONTROL0_CMD_REDIRECTSTREAM 0x0012
- #define CONTROL0_CMD_CLOSESTREAM 0x0013
- #define CONTROL0_CMD_CLOSECIRCUIT 0x0014
- #define _CONTROL0_CMD_MAX_RECOGNIZED 0x0014
- /* Recognized error codes. */
- #define ERR_UNSPECIFIED 0x0000
- #define ERR_INTERNAL 0x0001
- #define ERR_UNRECOGNIZED_TYPE 0x0002
- #define ERR_SYNTAX 0x0003
- #define ERR_UNRECOGNIZED_CONFIG_KEY 0x0004
- #define ERR_INVALID_CONFIG_VALUE 0x0005
- #define ERR_UNRECOGNIZED_EVENT_CODE 0x0006
- #define ERR_UNAUTHORIZED 0x0007
- #define ERR_REJECTED_AUTHENTICATION 0x0008
- #define ERR_RESOURCE_EXHAUSETED 0x0009
- #define ERR_NO_STREAM 0x000A
- #define ERR_NO_CIRC 0x000B
- #define ERR_NO_ROUTER 0x000C
- /* Recognized asynchronous event types. */
- #define _EVENT_MIN 0x0001
- #define EVENT_CIRCUIT_STATUS 0x0001
- #define EVENT_STREAM_STATUS 0x0002
- #define EVENT_OR_CONN_STATUS 0x0003
- #define EVENT_BANDWIDTH_USED 0x0004
- #define EVENT_LOG_OBSOLETE 0x0005
- #define EVENT_NEW_DESC 0x0006
- #define EVENT_DEBUG_MSG 0x0007
- #define EVENT_INFO_MSG 0x0008
- #define EVENT_NOTICE_MSG 0x0009
- #define EVENT_WARN_MSG 0x000A
- #define EVENT_ERR_MSG 0x000B
- #define LAST_V0_EVENT 0x000B
- #define EVENT_ADDRMAP 0x000C
- #define EVENT_AUTHDIR_NEWDESCS 0x000D
- #define _EVENT_MAX 0x000D
- /** Array mapping from message type codes to human-readable message
- * type names. Used for compatibility with version 0 of the control
- * protocol. */
- static const char * CONTROL0_COMMANDS[_CONTROL0_CMD_MAX_RECOGNIZED+1] = {
- "error",
- "done",
- "setconf",
- "getconf",
- "confvalue",
- "setevents",
- "events",
- "authenticate",
- "saveconf",
- "signal",
- "mapaddress",
- "getinfo",
- "infovalue",
- "extendcircuit",
- "attachstream",
- "postdescriptor",
- "fragmentheader",
- "fragment",
- };
- /** Bitfield: The bit 1<<e is set if <b>any</b> open control
- * connection is interested in events of type <b>e</b>. We use this
- * so that we can decide to skip generating event messages that nobody
- * has interest in without having to walk over the global connection
- * list to find out.
- **/
- static uint32_t global_event_mask0 = 0;
- static uint32_t global_event_mask1 = 0;
- /** True iff we have disabled log messages from being sent to the controller */
- static int disable_log_messages = 0;
- /** Macro: true if any control connection is interested in events of type
- * <b>e</b>. */
- #define EVENT_IS_INTERESTING0(e) (global_event_mask0 & (1<<(e)))
- #define EVENT_IS_INTERESTING1(e) (global_event_mask1 & (1<<(e)))
- #define EVENT_IS_INTERESTING(e) \
- ((global_event_mask0|global_event_mask1) & (1<<(e)))
- /** If we're using cookie-type authentication, how long should our cookies be?
- */
- #define AUTHENTICATION_COOKIE_LEN 32
- /** If true, we've set authentication_cookie to a secret code and
- * stored it to disk. */
- static int authentication_cookie_is_set = 0;
- static char authentication_cookie[AUTHENTICATION_COOKIE_LEN];
- static void connection_printf_to_buf(connection_t *conn,
- const char *format, ...)
- CHECK_PRINTF(2,3);
- /*static*/ size_t write_escaped_data(const char *data, size_t len,
- int translate_newlines, char **out);
- /*static*/ size_t read_escaped_data(const char *data, size_t len,
- int translate_newlines, char **out);
- static void send_control0_message(connection_t *conn, uint16_t type,
- uint32_t len, const char *body);
- static void send_control_done(connection_t *conn);
- static void send_control_done2(connection_t *conn, const char *msg,
- size_t len);
- static void send_control0_error(connection_t *conn, uint16_t error,
- const char *message);
- static void send_control0_event(uint16_t event, uint32_t len,
- const char *body);
- static void send_control1_event(uint16_t event, const char *format, ...)
- CHECK_PRINTF(2,3);
- static int handle_control_setconf(connection_t *conn, uint32_t len,
- char *body);
- static int handle_control_resetconf(connection_t *conn, uint32_t len,
- char *body);
- static int handle_control_getconf(connection_t *conn, uint32_t len,
- const char *body);
- static int handle_control_setevents(connection_t *conn, uint32_t len,
- const char *body);
- static int handle_control_authenticate(connection_t *conn, uint32_t len,
- const char *body);
- static int handle_control_saveconf(connection_t *conn, uint32_t len,
- const char *body);
- static int handle_control_signal(connection_t *conn, uint32_t len,
- const char *body);
- static int handle_control_mapaddress(connection_t *conn, uint32_t len,
- const char *body);
- static int handle_control_getinfo(connection_t *conn, uint32_t len,
- const char *body);
- static int handle_control_extendcircuit(connection_t *conn, uint32_t len,
- const char *body);
- static int handle_control_attachstream(connection_t *conn, uint32_t len,
- const char *body);
- static int handle_control_postdescriptor(connection_t *conn, uint32_t len,
- const char *body);
- static int handle_control_redirectstream(connection_t *conn, uint32_t len,
- const char *body);
- static int handle_control_closestream(connection_t *conn, uint32_t len,
- const char *body);
- static int handle_control_closecircuit(connection_t *conn, uint32_t len,
- const char *body);
- static int write_stream_target_to_buf(connection_t *conn, char *buf,
- size_t len);
- /** Given a possibly invalid message type code <b>cmd</b>, return a
- * human-readable string equivalent. */
- static INLINE const char *
- control_cmd_to_string(uint16_t cmd)
- {
- return (cmd<=_CONTROL0_CMD_MAX_RECOGNIZED) ?
- CONTROL0_COMMANDS[cmd] : "Unknown";
- }
- /** Given a control event code for a message event, return the corresponding
- * log severity. */
- static INLINE int
- event_to_log_severity(int event)
- {
- switch (event) {
- case EVENT_DEBUG_MSG: return LOG_DEBUG;
- case EVENT_INFO_MSG: return LOG_INFO;
- case EVENT_NOTICE_MSG: return LOG_NOTICE;
- case EVENT_WARN_MSG: return LOG_WARN;
- case EVENT_ERR_MSG: return LOG_ERR;
- default: return -1;
- }
- }
- /** Given a log severity, return the corresponding control event code. */
- static INLINE int
- log_severity_to_event(int severity)
- {
- switch (severity) {
- case LOG_DEBUG: return EVENT_DEBUG_MSG;
- case LOG_INFO: return EVENT_INFO_MSG;
- case LOG_NOTICE: return EVENT_NOTICE_MSG;
- case LOG_WARN: return EVENT_WARN_MSG;
- case LOG_ERR: return EVENT_ERR_MSG;
- default: return -1;
- }
- }
- /** Set <b>global_event_maskX</b> (where X is 0 or 1) to the bitwise OR
- * of each live control connection's event_mask field. */
- void
- control_update_global_event_mask(void)
- {
- connection_t **conns;
- int n_conns, i;
- global_event_mask0 = 0;
- global_event_mask1 = 0;
- get_connection_array(&conns, &n_conns);
- for (i = 0; i < n_conns; ++i) {
- if (conns[i]->type == CONN_TYPE_CONTROL &&
- STATE_IS_OPEN(conns[i]->state)) {
- if (STATE_IS_V0(conns[i]->state))
- global_event_mask0 |= conns[i]->event_mask;
- else
- global_event_mask1 |= conns[i]->event_mask;
- }
- }
- control_adjust_event_log_severity();
- }
- /** Adjust the log severities that result in control_event_logmsg being called
- * to match the severity of log messages that any controllers are interested
- * in. */
- void
- control_adjust_event_log_severity(void)
- {
- int i;
- int min_log_event=EVENT_ERR_MSG, max_log_event=EVENT_DEBUG_MSG;
- for (i = EVENT_DEBUG_MSG; i <= EVENT_ERR_MSG; ++i) {
- if (EVENT_IS_INTERESTING(i)) {
- min_log_event = i;
- break;
- }
- }
- for (i = EVENT_ERR_MSG; i >= EVENT_DEBUG_MSG; --i) {
- if (EVENT_IS_INTERESTING(i)) {
- max_log_event = i;
- break;
- }
- }
- if (EVENT_IS_INTERESTING(EVENT_LOG_OBSOLETE)) {
- if (min_log_event > EVENT_NOTICE_MSG)
- min_log_event = EVENT_NOTICE_MSG;
- if (max_log_event < EVENT_ERR_MSG)
- max_log_event = EVENT_ERR_MSG;
- }
- change_callback_log_severity(event_to_log_severity(min_log_event),
- event_to_log_severity(max_log_event),
- control_event_logmsg);
- }
- /** Append a NUL-terminated string <b>s</b> to the end of
- * <b>conn</b>-\>outbuf
- */
- static INLINE void
- connection_write_str_to_buf(const char *s, connection_t *conn)
- {
- size_t len = strlen(s);
- connection_write_to_buf(s, len, conn);
- }
- /** Given a <b>len</b>-character string in <b>data</b>, made of lines
- * terminated by CRLF, allocate a new string in *<b>out</b>, and copy
- * the contents of <b>data</b> into *<b>out</b>, adding a period
- * before any period that that appears at the start of a line, and
- * adding a period-CRLF line at the end. If <b>translate_newlines</b>
- * is true, replace all LF characters sequences with CRLF. Return the
- * number of bytes in *<b>out</b>.
- */
- /* static */ size_t
- write_escaped_data(const char *data, size_t len, int translate_newlines,
- char **out)
- {
- size_t sz_out = len+8;
- char *outp;
- const char *end;
- int i;
- int start_of_line;
- for (i=0; i<(int)len; ++i) {
- if (data[i]== '\n')
- sz_out += 2; /* Maybe add a CR; maybe add a dot. */
- }
- *out = outp = tor_malloc(sz_out+1);
- end = data+len;
- start_of_line = 1;
- while (data < end) {
- if (*data == '\n') {
- if (translate_newlines)
- *outp++ = '\r';
- start_of_line = 1;
- } else if (*data == '.') {
- if (start_of_line) {
- start_of_line = 0;
- *outp++ = '.';
- }
- } else {
- start_of_line = 0;
- }
- *outp++ = *data++;
- }
- if (outp < *out+2 || memcmp(outp-2, "\r\n", 2)) {
- *outp++ = '\r';
- *outp++ = '\n';
- }
- *outp++ = '.';
- *outp++ = '\r';
- *outp++ = '\n';
- *outp = '\0'; /* NUL-terminate just in case. */
- tor_assert((outp - *out) <= (int)sz_out);
- return outp - *out;
- }
- /** Given a <b>len</b>-character string in <b>data</b>, made of lines
- * terminated by CRLF, allocate a new string in *<b>out</b>, and copy
- * the contents of <b>data</b> into *<b>out</b>, removing any period
- * that appears at the start of a line. If <b>translate_newlines</b>
- * is true, replace all CRLF sequences with LF. Return the number of
- * bytes in *<b>out</b>. */
- /*static*/ size_t
- read_escaped_data(const char *data, size_t len, int translate_newlines,
- char **out)
- {
- char *outp;
- const char *next;
- const char *end;
- *out = outp = tor_malloc(len+1);
- end = data+len;
- while (data < end) {
- if (*data == '.')
- ++data;
- if (translate_newlines)
- next = tor_memmem(data, end-data, "\r\n", 2);
- else
- next = tor_memmem(data, end-data, "\r\n.", 3);
- if (next) {
- memcpy(outp, data, next-data);
- outp += (next-data);
- data = next+2;
- } else {
- memcpy(outp, data, end-data);
- outp += (end-data);
- *outp = '\0';
- return outp - *out;
- }
- if (translate_newlines) {
- *outp++ = '\n';
- } else {
- *outp++ = '\r';
- *outp++ = '\n';
- }
- }
- *outp = '\0';
- return outp - *out;
- }
- /** Given a pointer to a string starting at <b>start</b> containing
- * <b>in_len_max</b> characters, decode a string beginning with a single
- * quote, containing any number of non-quote characters or characters escaped
- * with a backslash, and ending with a final quote. Place the resulting
- * string (unquoted, unescaped) into a newly allocated string in *<b>out</b>;
- * store its length in <b>out_len</b>. On success, return a pointer to the
- * character immediately following the escaped string. On failure, return
- * NULL. */
- static const char *
- get_escaped_string(const char *start, size_t in_len_max,
- char **out, size_t *out_len)
- {
- const char *cp, *end;
- char *outp;
- size_t len=0;
- if (*start != '\"')
- return NULL;
- cp = start+1;
- end = start+in_len_max;
- /* Calculate length. */
- while (1) {
- if (cp >= end)
- return NULL;
- else if (*cp == '\\') {
- if (++cp == end)
- return NULL; /* Can't escape EOS. */
- ++cp;
- ++len;
- } else if (*cp == '\"') {
- break;
- } else {
- ++cp;
- ++len;
- }
- }
- end = cp;
- outp = *out = tor_malloc(len+1);
- *out_len = len;
- cp = start+1;
- while (cp < end) {
- if (*cp == '\\')
- ++cp;
- *outp++ = *cp++;
- }
- *outp = '\0';
- tor_assert((outp - *out) == (int)*out_len);
- return end+1;
- }
- /** Acts like sprintf, but writes its formatted string to the end of
- * <b>conn</b>-\>outbuf. The message may be truncated if it is too long,
- * but it will always end with a CRLF sequence.
- *
- * Currently the length of the message is limited to 1024 (including the
- * ending \n\r\0. */
- static void
- connection_printf_to_buf(connection_t *conn, const char *format, ...)
- {
- #define CONNECTION_PRINTF_TO_BUF_BUFFERSIZE 1024
- va_list ap;
- char buf[CONNECTION_PRINTF_TO_BUF_BUFFERSIZE];
- int r;
- size_t len;
- va_start(ap,format);
- r = tor_vsnprintf(buf, sizeof(buf), format, ap);
- va_end(ap);
- len = strlen(buf);
- if (memcmp("\r\n\0", buf+len-2, 3)) {
- buf[CONNECTION_PRINTF_TO_BUF_BUFFERSIZE-1] = '\0';
- buf[CONNECTION_PRINTF_TO_BUF_BUFFERSIZE-2] = '\n';
- buf[CONNECTION_PRINTF_TO_BUF_BUFFERSIZE-3] = '\r';
- }
- connection_write_to_buf(buf, len, conn);
- }
- /** Send a message of type <b>type</b> containing <b>len</b> bytes
- * from <b>body</b> along the control connection <b>conn</b> */
- static void
- send_control0_message(connection_t *conn, uint16_t type, uint32_t len,
- const char *body)
- {
- char buf[10];
- tor_assert(conn);
- tor_assert(STATE_IS_V0(conn->state));
- tor_assert(len || !body);
- tor_assert(type <= _CONTROL0_CMD_MAX_RECOGNIZED);
- if (len < 65536) {
- set_uint16(buf, htons(len));
- set_uint16(buf+2, htons(type));
- connection_write_to_buf(buf, 4, conn);
- if (len)
- connection_write_to_buf(body, len, conn);
- } else {
- set_uint16(buf, htons(65535));
- set_uint16(buf+2, htons(CONTROL0_CMD_FRAGMENTHEADER));
- set_uint16(buf+4, htons(type));
- set_uint32(buf+6, htonl(len));
- connection_write_to_buf(buf, 10, conn);
- connection_write_to_buf(body, 65535-6, conn);
- len -= (65535-6);
- body += (65535-6);
- while (len) {
- size_t chunklen = (len<65535)?len:65535;
- set_uint16(buf, htons((uint16_t)chunklen));
- set_uint16(buf+2, htons(CONTROL0_CMD_FRAGMENT));
- connection_write_to_buf(buf, 4, conn);
- connection_write_to_buf(body, chunklen, conn);
- len -= chunklen;
- body += chunklen;
- }
- }
- }
- /** Send a "DONE" message down the control connection <b>conn</b> */
- static void
- send_control_done(connection_t *conn)
- {
- if (STATE_IS_V0(conn->state)) {
- send_control0_message(conn, CONTROL0_CMD_DONE, 0, NULL);
- } else {
- connection_write_str_to_buf("250 OK\r\n", conn);
- }
- }
- /** Send a "DONE" message down the v0 control message <b>conn</b>, with body
- * as provided in the <b>len</b> bytes at <b>msg</b>.
- */
- static void
- send_control_done2(connection_t *conn, const char *msg, size_t len)
- {
- if (len==0)
- len = strlen(msg);
- send_control0_message(conn, CONTROL0_CMD_DONE, len, msg);
- }
- /** Send an error message with error code <b>error</b> and body
- * <b>message</b> down the connection <b>conn</b> */
- static void
- send_control0_error(connection_t *conn, uint16_t error, const char *message)
- {
- char buf[256];
- size_t len;
- set_uint16(buf, htons(error));
- len = strlen(message);
- tor_assert(len < (256-2));
- memcpy(buf+2, message, len);
- send_control0_message(conn, CONTROL0_CMD_ERROR, (uint16_t)(len+2), buf);
- }
- /** Send an 'event' message of event type <b>event</b>, containing
- * <b>len</b> bytes in <b>body</b> to every control connection that
- * is interested in it. */
- static void
- send_control0_event(uint16_t event, uint32_t len, const char *body)
- {
- connection_t **conns;
- int n_conns, i;
- size_t buflen;
- char *buf;
- tor_assert(event >= _EVENT_MIN && event <= LAST_V0_EVENT);
- buflen = len + 2;
- buf = tor_malloc_zero(buflen);
- set_uint16(buf, htons(event));
- memcpy(buf+2, body, len);
- get_connection_array(&conns, &n_conns);
- for (i = 0; i < n_conns; ++i) {
- if (conns[i]->type == CONN_TYPE_CONTROL &&
- !conns[i]->marked_for_close &&
- conns[i]->state == CONTROL_CONN_STATE_OPEN_V0 &&
- conns[i]->event_mask & (1<<event)) {
- send_control0_message(conns[i], CONTROL0_CMD_EVENT, buflen, buf);
- if (event == EVENT_ERR_MSG)
- _connection_controller_force_write(conns[i]);
- }
- }
- tor_free(buf);
- }
- /* Send an event to all v1 controllers that are listening for code
- * <b>event</b>. The event's body is given by <b>msg</b>. */
- static void
- send_control1_event_string(uint16_t event, const char *msg)
- {
- connection_t **conns;
- int n_conns, i;
- tor_assert(event >= _EVENT_MIN && event <= _EVENT_MAX);
- get_connection_array(&conns, &n_conns);
- for (i = 0; i < n_conns; ++i) {
- if (conns[i]->type == CONN_TYPE_CONTROL &&
- !conns[i]->marked_for_close &&
- conns[i]->state == CONTROL_CONN_STATE_OPEN_V1 &&
- conns[i]->event_mask & (1<<event)) {
- connection_write_to_buf(msg, strlen(msg), conns[i]);
- if (event == EVENT_ERR_MSG)
- _connection_controller_force_write(conns[i]);
- }
- }
- }
- /* Send an event to all v1 controllers that are listening for code
- * <b>event</b>. The event's body is created by the printf-style format in
- * <b>format</b>, and other arguments as provided.
- *
- * Currently the length of the message is limited to 1024 (including the
- * ending \n\r\0. */
- static void
- send_control1_event(uint16_t event, const char *format, ...)
- {
- #define SEND_CONTROL1_EVENT_BUFFERSIZE 1024
- int r;
- char buf[SEND_CONTROL1_EVENT_BUFFERSIZE]; /* XXXX Length */
- va_list ap;
- size_t len;
- va_start(ap, format);
- r = tor_vsnprintf(buf, sizeof(buf), format, ap);
- va_end(ap);
- len = strlen(buf);
- if (memcmp("\r\n\0", buf+len-2, 3)) {
- /* if it is not properly terminated, do it now */
- buf[SEND_CONTROL1_EVENT_BUFFERSIZE-1] = '\0';
- buf[SEND_CONTROL1_EVENT_BUFFERSIZE-2] = '\n';
- buf[SEND_CONTROL1_EVENT_BUFFERSIZE-3] = '\r';
- }
- send_control1_event_string(event, buf);
- }
- /** Given a text circuit <b>id</b>, return the corresponding circuit. */
- static circuit_t *
- get_circ(const char *id)
- {
- unsigned long n_id;
- int ok;
- n_id = tor_parse_ulong(id, 10, 0, ULONG_MAX, &ok, NULL);
- if (!ok)
- return NULL;
- return circuit_get_by_global_id(n_id);
- }
- /** Given a text stream <b>id</b>, return the corresponding AP connection. */
- static connection_t *
- get_stream(const char *id)
- {
- unsigned long n_id;
- int ok;
- connection_t *conn;
- n_id = tor_parse_ulong(id, 10, 0, ULONG_MAX, &ok, NULL);
- if (!ok)
- return NULL;
- conn = connection_get_by_global_id(n_id);
- if (!conn || conn->type != CONN_TYPE_AP)
- return NULL;
- return conn;
- }
- /** Helper for setconf and resetconf. Acts like setconf, except
- * it passes <b>use_defaults</b> on to options_trial_assign().
- */
- static int
- control_setconf_helper(connection_t *conn, uint32_t len, char *body,
- int use_defaults, int clear_first)
- {
- int r;
- config_line_t *lines=NULL;
- char *start = body;
- int v0 = STATE_IS_V0(conn->state);
- if (!v0) {
- char *config = tor_malloc(len+1);
- char *outp = config;
- while (*body) {
- char *eq = body;
- while (!TOR_ISSPACE(*eq) && *eq != '=')
- ++eq;
- memcpy(outp, body, eq-body);
- outp += (eq-body);
- *outp++ = ' ';
- body = eq+1;
- if (*eq == '=') {
- if (*body != '\"') {
- while (!TOR_ISSPACE(*body))
- *outp++ = *body++;
- } else {
- char *val;
- size_t val_len;
- body = (char*)get_escaped_string(body, (len - (body-start)),
- &val, &val_len);
- if (!body) {
- connection_write_str_to_buf("551 Couldn't parse string\r\n", conn);
- tor_free(config);
- return 0;
- }
- memcpy(outp, val, val_len);
- outp += val_len;
- tor_free(val);
- }
- }
- while (TOR_ISSPACE(*body))
- ++body;
- *outp++ = '\n';
- }
- *outp = '\0';
- if (config_get_lines(config, &lines) < 0) {
- warn(LD_CONTROL,"Controller gave us config lines we can't parse.");
- connection_write_str_to_buf("551 Couldn't parse configuration\r\n",
- conn);
- tor_free(config);
- return 0;
- }
- tor_free(config);
- } else {
- if (config_get_lines(body, &lines) < 0) {
- warn(LD_CONTROL,"Controller gave us config lines we can't parse.");
- send_control0_error(conn, ERR_SYNTAX, "Couldn't parse configuration");
- return 0;
- }
- }
- if ((r=options_trial_assign(lines, use_defaults, clear_first)) < 0) {
- int v0_err;
- const char *msg;
- warn(LD_CONTROL,"Controller gave us config lines that didn't validate.");
- switch (r) {
- case -1:
- v0_err = ERR_UNRECOGNIZED_CONFIG_KEY;
- msg = "552 Unrecognized option";
- break;
- case -2:
- v0_err = ERR_INVALID_CONFIG_VALUE;
- msg = "513 Unacceptable option value";
- break;
- case -3:
- v0_err = ERR_INVALID_CONFIG_VALUE;
- msg = "553 Transition not allowed";
- break;
- case -4:
- default:
- v0_err = ERR_INVALID_CONFIG_VALUE;
- msg = "553 Unable to set option";
- break;
- }
- if (v0) {
- send_control0_error(conn, v0_err, msg);
- } else {
- connection_printf_to_buf(conn, "%s\r\n", msg);
- }
- config_free_lines(lines);
- return 0;
- }
- config_free_lines(lines);
- send_control_done(conn);
- return 0;
- }
- /** Called when we receive a SETCONF message: parse the body and try
- * to update our configuration. Reply with a DONE or ERROR message. */
- static int
- handle_control_setconf(connection_t *conn, uint32_t len, char *body)
- {
- return control_setconf_helper(conn, len, body, 0, 1);
- }
- /** Called when we receive a RESETCONF message: parse the body and try
- * to update our configuration. Reply with a DONE or ERROR message. */
- static int
- handle_control_resetconf(connection_t *conn, uint32_t len, char *body)
- {
- int v0 = STATE_IS_V0(conn->state);
- tor_assert(!v0);
- return control_setconf_helper(conn, len, body, 1, 1);
- }
- /** Called when we receive a GETCONF message. Parse the request, and
- * reply with a CONFVALUE or an ERROR message */
- static int
- handle_control_getconf(connection_t *conn, uint32_t body_len, const char *body)
- {
- smartlist_t *questions = NULL;
- smartlist_t *answers = NULL;
- smartlist_t *unrecognized = NULL;
- char *msg = NULL;
- size_t msg_len;
- or_options_t *options = get_options();
- int v0 = STATE_IS_V0(conn->state);
- questions = smartlist_create();
- if (v0) {
- smartlist_split_string(questions, body, "\n",
- SPLIT_SKIP_SPACE|SPLIT_IGNORE_BLANK, 0);
- } else {
- smartlist_split_string(questions, body, " ",
- SPLIT_SKIP_SPACE|SPLIT_IGNORE_BLANK, 0);
- }
- answers = smartlist_create();
- unrecognized = smartlist_create();
- SMARTLIST_FOREACH(questions, char *, q,
- {
- if (!option_is_recognized(q)) {
- if (v0) {
- send_control0_error(conn, ERR_UNRECOGNIZED_CONFIG_KEY, q);
- goto done;
- } else {
- smartlist_add(unrecognized, q);
- }
- } else {
- config_line_t *answer = option_get_assignment(options,q);
- if (!v0 && !answer) {
- const char *name = option_get_canonical_name(q);
- size_t alen = strlen(name)+8;
- char *astr = tor_malloc(alen);
- tor_snprintf(astr, alen, "250-%s\r\n", name);
- smartlist_add(answers, astr);
- }
- while (answer) {
- config_line_t *next;
- size_t alen = strlen(answer->key)+strlen(answer->value)+8;
- char *astr = tor_malloc(alen);
- if (v0)
- tor_snprintf(astr, alen, "%s %s\n", answer->key, answer->value);
- else
- tor_snprintf(astr, alen, "250-%s=%s\r\n",
- answer->key, answer->value);
- smartlist_add(answers, astr);
- next = answer->next;
- tor_free(answer->key);
- tor_free(answer->value);
- tor_free(answer);
- answer = next;
- }
- }
- });
- if (v0) {
- msg = smartlist_join_strings(answers, "", 0, &msg_len);
- send_control0_message(conn, CONTROL0_CMD_CONFVALUE,
- (uint16_t)msg_len, msg_len?msg:NULL);
- } else {
- int i,len;
- if ((len = smartlist_len(unrecognized))) {
- for (i=0; i < len-1; ++i)
- connection_printf_to_buf(conn,
- "552-Unrecognized configuration key \"%s\"\r\n",
- (char*)smartlist_get(unrecognized, i));
- connection_printf_to_buf(conn,
- "552 Unrecognized configuration key \"%s\"\r\n",
- (char*)smartlist_get(unrecognized, len-1));
- } else if ((len = smartlist_len(answers))) {
- char *tmp = smartlist_get(answers, len-1);
- tor_assert(strlen(tmp)>4);
- tmp[3] = ' ';
- msg = smartlist_join_strings(answers, "", 0, &msg_len);
- connection_write_to_buf(msg, msg_len, conn);
- } else {
- connection_write_str_to_buf("250 OK\r\n", conn);
- }
- }
- done:
- if (answers) {
- SMARTLIST_FOREACH(answers, char *, cp, tor_free(cp));
- smartlist_free(answers);
- }
- if (questions) {
- SMARTLIST_FOREACH(questions, char *, cp, tor_free(cp));
- smartlist_free(questions);
- }
- smartlist_free(unrecognized);
- tor_free(msg);
- return 0;
- }
- /** Called when we get a SETEVENTS message: update conn->event_mask,
- * and reply with DONE or ERROR. */
- static int
- handle_control_setevents(connection_t *conn, uint32_t len, const char *body)
- {
- uint16_t event_code;
- uint32_t event_mask = 0;
- unsigned int extended = 0;
- if (STATE_IS_V0(conn->state)) {
- if (len % 2) {
- send_control0_error(conn, ERR_SYNTAX,
- "Odd number of bytes in setevents message");
- return 0;
- }
- for (; len; len -= 2, body += 2) {
- event_code = ntohs(get_uint16(body));
- if (event_code < _EVENT_MIN || event_code > LAST_V0_EVENT) {
- send_control0_error(conn, ERR_UNRECOGNIZED_EVENT_CODE,
- "Unrecognized event code");
- return 0;
- }
- event_mask |= (1 << event_code);
- }
- } else {
- smartlist_t *events = smartlist_create();
- smartlist_split_string(events, body, " ",
- SPLIT_SKIP_SPACE|SPLIT_IGNORE_BLANK, 0);
- SMARTLIST_FOREACH(events, const char *, ev,
- {
- if (!strcasecmp(ev, "EXTENDED")) {
- extended = 1;
- continue;
- } else if (!strcasecmp(ev, "CIRC"))
- event_code = EVENT_CIRCUIT_STATUS;
- else if (!strcasecmp(ev, "STREAM"))
- event_code = EVENT_STREAM_STATUS;
- else if (!strcasecmp(ev, "ORCONN"))
- event_code = EVENT_OR_CONN_STATUS;
- else if (!strcasecmp(ev, "BW"))
- event_code = EVENT_BANDWIDTH_USED;
- else if (!strcasecmp(ev, "DEBUG"))
- event_code = EVENT_DEBUG_MSG;
- else if (!strcasecmp(ev, "INFO"))
- event_code = EVENT_INFO_MSG;
- else if (!strcasecmp(ev, "NOTICE"))
- event_code = EVENT_NOTICE_MSG;
- else if (!strcasecmp(ev, "WARN"))
- event_code = EVENT_WARN_MSG;
- else if (!strcasecmp(ev, "ERR"))
- event_code = EVENT_ERR_MSG;
- else if (!strcasecmp(ev, "NEWDESC"))
- event_code = EVENT_NEW_DESC;
- else if (!strcasecmp(ev, "ADDRMAP"))
- event_code = EVENT_ADDRMAP;
- else if (!strcasecmp(ev, "AUTHDIR_NEWDESCS"))
- event_code = EVENT_AUTHDIR_NEWDESCS;
- else {
- connection_printf_to_buf(conn, "552 Unrecognized event \"%s\"\r\n",
- ev);
- SMARTLIST_FOREACH(events, char *, e, tor_free(e));
- smartlist_free(events);
- return 0;
- }
- event_mask |= (1 << event_code);
- });
- SMARTLIST_FOREACH(events, char *, e, tor_free(e));
- smartlist_free(events);
- }
- conn->event_mask = event_mask;
- conn->control_events_are_extended = extended;
- control_update_global_event_mask();
- send_control_done(conn);
- return 0;
- }
- /** Decode the hashed, base64'd password stored in <b>hashed</b>. If
- * <b>buf</b> is provided, store the hashed password in the first
- * S2K_SPECIFIER_LEN+DIGEST_LEN bytes of <b>buf</b>. Return 0 on
- * success, -1 on failure.
- */
- int
- decode_hashed_password(char *buf, const char *hashed)
- {
- char decoded[64];
- if (!strcmpstart(hashed, "16:")) {
- if (base16_decode(decoded, sizeof(decoded), hashed+3, strlen(hashed+3))<0
- || strlen(hashed+3) != (S2K_SPECIFIER_LEN+DIGEST_LEN)*2) {
- return -1;
- }
- } else {
- if (base64_decode(decoded, sizeof(decoded), hashed, strlen(hashed))
- != S2K_SPECIFIER_LEN+DIGEST_LEN) {
- return -1;
- }
- }
- if (buf)
- memcpy(buf, decoded, S2K_SPECIFIER_LEN+DIGEST_LEN);
- return 0;
- }
- /** Called when we get an AUTHENTICATE message. Check whether the
- * authentication is valid, and if so, update the connection's state to
- * OPEN. Reply with DONE or ERROR.
- */
- static int
- handle_control_authenticate(connection_t *conn, uint32_t len, const char *body)
- {
- int used_quoted_string = 0;
- or_options_t *options = get_options();
- char *password;
- size_t password_len;
- if (STATE_IS_V0(conn->state)) {
- password = (char*)body;
- password_len = len;
- } else {
- if (TOR_ISXDIGIT(body[0])) {
- int i = 0;
- while (TOR_ISXDIGIT(body[i]))
- ++i;
- password = tor_malloc(i/2 + 1);
- if (base16_decode(password, i/2+1, body, i)<0) {
- connection_write_str_to_buf(
- "551 Invalid hexadecimal encoding. Maybe you tried a plain text "
- "password? If so, the standard requires you put it in double "
- "quotes.\r\n", conn);
- tor_free(password);
- return 0;
- }
- password_len = i/2;
- } else if (TOR_ISSPACE(body[0])) {
- password = tor_strdup("");
- password_len = 0;
- } else {
- if (!get_escaped_string(body, len, &password, &password_len)) {
- connection_write_str_to_buf("551 Invalid quoted string. You need "
- "to put the password in double quotes.\r\n", conn);
- return 0;
- }
- used_quoted_string = 1;
- }
- }
- if (options->CookieAuthentication) {
- if (len == AUTHENTICATION_COOKIE_LEN &&
- !memcmp(authentication_cookie, password, password_len)) {
- goto ok;
- }
- } else if (options->HashedControlPassword) {
- char expected[S2K_SPECIFIER_LEN+DIGEST_LEN];
- char received[DIGEST_LEN];
- if (decode_hashed_password(expected, options->HashedControlPassword)<0) {
- warn(LD_CONTROL,"Couldn't decode HashedControlPassword: invalid base16");
- goto err;
- }
- secret_to_key(received,DIGEST_LEN,password,password_len,expected);
- if (!memcmp(expected+S2K_SPECIFIER_LEN, received, DIGEST_LEN))
- goto ok;
- goto err;
- } else {
- /* if Tor doesn't demand any stronger authentication, then
- * the controller can get in with anything. */
- goto ok;
- }
- err:
- if (STATE_IS_V0(conn->state))
- send_control0_error(conn,ERR_REJECTED_AUTHENTICATION,
- "Authentication failed");
- else {
- tor_free(password);
- if (used_quoted_string)
- connection_write_str_to_buf("515 Authentication failed\r\n", conn);
- else
- connection_write_str_to_buf(
- "515 Authentication failed. Maybe you tried a plain text password? "
- "If so, the standard requires you put it in double quotes.\r\n",conn);
- }
- return 0;
- ok:
- info(LD_CONTROL, "Authenticated control connection (%d)", conn->s);
- send_control_done(conn);
- if (STATE_IS_V0(conn->state))
- conn->state = CONTROL_CONN_STATE_OPEN_V0;
- else {
- conn->state = CONTROL_CONN_STATE_OPEN_V1;
- tor_free(password);
- }
- return 0;
- }
- /** Called when we get a SAVECONF command. Try to flush the current options to
- * disk, and report success or failure. */
- static int
- handle_control_saveconf(connection_t *conn, uint32_t len,
- const char *body)
- {
- if (options_save_current()<0) {
- if (STATE_IS_V0(conn->state))
- send_control0_error(conn, ERR_INTERNAL,
- "Unable to write configuration to disk.");
- else
- connection_write_str_to_buf("551 Unable to write configuration to disk.",
- conn);
- } else {
- send_control_done(conn);
- }
- return 0;
- }
- /** Called when we get a SIGNAL command. React to the provided signal, and
- * report success or failure. (If the signal results in a shutdown, success
- * may not be reported.) */
- static int
- handle_control_signal(connection_t *conn, uint32_t len,
- const char *body)
- {
- int sig;
- if (STATE_IS_V0(conn->state)) {
- if (len != 1) {
- send_control0_error(conn, ERR_SYNTAX,
- "Body of SIGNAL command too long or too short.");
- return 0;
- } else {
- sig = (uint8_t)body[0];
- }
- } else {
- int n = 0;
- char *s;
- while (body[n] && ! TOR_ISSPACE(body[n]))
- ++n;
- s = tor_strndup(body, n);
- if (!strcasecmp(s, "RELOAD") || !strcasecmp(s, "HUP"))
- sig = SIGHUP;
- else if (!strcasecmp(s, "SHUTDOWN") || !strcasecmp(s, "INT"))
- sig = SIGINT;
- else if (!strcasecmp(s, "DUMP") || !strcasecmp(s, "USR1"))
- sig = SIGUSR1;
- else if (!strcasecmp(s, "DEBUG") || !strcasecmp(s, "USR2"))
- sig = SIGUSR2;
- else if (!strcasecmp(s, "HALT") || !strcasecmp(s, "TERM"))
- sig = SIGTERM;
- else {
- connection_printf_to_buf(conn, "552 Unrecognized signal code \"%s\"\r\n",
- body);
- sig = -1;
- }
- tor_free(s);
- if (sig<0)
- return 0;
- }
- if (control_signal_act(sig) < 0) {
- if (STATE_IS_V0(conn->state))
- send_control0_error(conn, ERR_SYNTAX, "Unrecognized signal number.");
- else
- connection_write_str_to_buf("551 Internal error acting on signal\r\n",
- conn);
- } else {
- send_control_done(conn);
- }
- return 0;
- }
- /** Called when we get a MAPADDRESS command; try to bind all listed addresses,
- * and report success or failrue. */
- static int
- handle_control_mapaddress(connection_t *conn, uint32_t len, const char *body)
- {
- smartlist_t *elts;
- smartlist_t *lines;
- smartlist_t *reply;
- char *r;
- size_t sz;
- int v0 = STATE_IS_V0(conn->state);
- lines = smartlist_create();
- elts = smartlist_create();
- reply = smartlist_create();
- if (v0)
- smartlist_split_string(lines, body, "\n",
- SPLIT_SKIP_SPACE|SPLIT_IGNORE_BLANK, 0);
- else
- smartlist_split_string(lines, body, " ",
- SPLIT_SKIP_SPACE|SPLIT_IGNORE_BLANK, 0);
- /* XXXX Make errors conformant. */
- SMARTLIST_FOREACH(lines, char *, line,
- {
- tor_strlower(line);
- if (v0)
- smartlist_split_string(elts, line, " ", 0, 2);
- else
- smartlist_split_string(elts, line, "=", 0, 2);
- if (smartlist_len(elts) == 2) {
- const char *from = smartlist_get(elts,0);
- const char *to = smartlist_get(elts,1);
- if (!is_plausible_address(from)) {
- warn(LD_CONTROL,"Skipping invalid argument '%s' in MapAddress msg",
- from);
- } else if (!is_plausible_address(to)) {
- warn(LD_CONTROL,"Skipping invalid argument '%s' in MapAddress msg",to);
- } else if (!strcmp(from, ".") || !strcmp(from, "0.0.0.0")) {
- const char *address = addressmap_register_virtual_address(
- !strcmp(from,".") ? RESOLVED_TYPE_HOSTNAME : RESOLVED_TYPE_IPV4,
- tor_strdup(to));
- if (!address) {
- warn(LD_CONTROL,
- "Unable to allocate address for '%s' in MapAddress msg",
- safe_str(line));
- } else {
- size_t anslen = strlen(address)+strlen(to)+8;
- char *ans = tor_malloc(anslen);
- if (v0)
- tor_snprintf(ans, anslen, "%s %s", address, to);
- else
- tor_snprintf(ans, anslen, "250-%s=%s", address, to);
- smartlist_add(reply, ans);
- }
- } else {
- addressmap_register(from, tor_strdup(to), 1);
- if (v0)
- smartlist_add(reply, tor_strdup(line));
- else {
- size_t anslen = strlen(line)+8;
- char *ans = tor_malloc(anslen);
- tor_snprintf(ans, anslen, "250-%s", line);
- smartlist_add(reply, ans);
- }
- }
- } else {
- warn(LD_CONTROL, "Skipping MapAddress line with wrong number of items.");
- }
- SMARTLIST_FOREACH(elts, char *, cp, tor_free(cp));
- smartlist_clear(elts);
- });
- SMARTLIST_FOREACH(lines, char *, cp, tor_free(cp));
- smartlist_free(lines);
- smartlist_free(elts);
- if (v0) {
- r = smartlist_join_strings(reply, "\n", 1, &sz);
- send_control_done2(conn,r,sz);
- } else {
- if (smartlist_len(reply))
- ((char*)smartlist_get(reply,smartlist_len(reply)-1))[3] = ' ';
- r = smartlist_join_strings(reply, "\r\n", 1, &sz);
- connection_write_to_buf(r, sz, conn);
- }
- SMARTLIST_FOREACH(reply, char *, cp, tor_free(cp));
- smartlist_free(reply);
- tor_free(r);
- return 0;
- }
- /** Return a newly allocated string listing all valid GETINFO fields as
- * required by GETINFO info/names. */
- static char *
- list_getinfo_options(void)
- {
- return tor_strdup(
- "accounting/bytes Number of bytes read/written so far in interval.\n"
- "accounting/bytes-left Number of bytes left to read/write in interval.\n"
- "accounting/enabled Is accounting currently enabled?\n"
- "accounting/hibernating Are we hibernating or awake?\n"
- "accounting/interval-end Time when interval ends.\n"
- "accounting/interval-start Time when interval starts.\n"
- "accounting/interval-wake Time to wake up in this interval.\n"
- "addr-mappings/all All current remapped addresses.\n"
- "addr-mappings/cache Addresses remapped by DNS cache.\n"
- "addr-mappings/configl Addresses remapped from configuration options.\n"
- "addr-mappings/control Addresses remapped by a controller.\n"
- "circuit-status Status of each current circuit.\n"
- "config/names List of configuration options, types, and documentation.\n"
- "desc/id/* Server descriptor by hex ID\n"
- "desc/name/* Server descriptor by nickname.\n"
- "helper-nodes Which nodes will we use as helpers?\n"
- "info/names List of GETINFO options, types, and documentation.\n"
- "network-status List of hex IDs, nicknames, server statuses.\n"
- "orconn-status Status of each current OR connection.\n"
- "stream-status Status of each current application stream.\n"
- "version The current version of Tor.\n");
- }
- /** Lookup the 'getinfo' entry <b>question</b>, and return
- * the answer in <b>*answer</b> (or NULL if key not recognized).
- * Return 0 if success, or -1 if internal error. */
- static int
- handle_getinfo_helper(const char *question, char **answer)
- {
- *answer = NULL; /* unrecognized key by default */
- if (!strcmp(question, "version")) {
- *answer = tor_strdup(VERSION);
- } else if (!strcmp(question, "config-file")) {
- *answer = tor_strdup(get_torrc_fname());
- } else if (!strcmpstart(question, "accounting/")) {
- return accounting_getinfo_helper(question, answer);
- } else if (!strcmpstart(question, "helper-nodes")) {
- return helper_nodes_getinfo_helper(question, answer);
- } else if (!strcmpstart(question, "config/")) {
- return config_getinfo_helper(question, answer);
- } else if (!strcmp(question, "info/names")) {
- *answer = list_getinfo_options();
- } else if (!strcmpstart(question, "desc/id/")) {
- routerinfo_t *ri = router_get_by_hexdigest(question+strlen("desc/id/"));
- if (ri && ri->cache_info.signed_descriptor)
- *answer = tor_strdup(ri->cache_info.signed_descriptor);
- } else if (!strcmpstart(question, "desc/name/")) {
- routerinfo_t *ri = router_get_by_nickname(question+strlen("desc/name/"),1);
- if (ri && ri->cache_info.signed_descriptor)
- *answer = tor_strdup(ri->cache_info.signed_descriptor);
- } else if (!strcmpstart(question, "unregistered-servers-")) {
- *answer = dirserver_getinfo_unregistered(question +
- strlen("unregistered-servers-"));
- } else if (!strcmp(question, "network-status")) {
- routerlist_t *routerlist = router_get_routerlist();
- if (!routerlist || !routerlist->routers ||
- list_server_status(routerlist->routers, answer) < 0) {
- return -1;
- }
- } else if (!strcmp(question, "circuit-status")) {
- circuit_t *circ;
- smartlist_t *status = smartlist_create();
- for (circ = _circuit_get_global_list(); circ; circ = circ->next) {
- char *s, *path;
- size_t slen;
- const char *state;
- if (! CIRCUIT_IS_ORIGIN(circ) || circ->marked_for_close)
- continue;
- path = circuit_list_path(circ,0);
- if (circ->state == CIRCUIT_STATE_OPEN)
- state = "BUILT";
- else if (strlen(path))
- state = "EXTENDED";
- else
- state = "LAUNCHED";
- slen = strlen(path)+strlen(state)+20;
- s = tor_malloc(slen+1);
- tor_snprintf(s, slen, "%lu %s %s",
- (unsigned long)circ->global_identifier,
- state, path);
- smartlist_add(status, s);
- tor_free(path);
- }
- *answer = smartlist_join_strings(status, "\r\n", 0, NULL);
- SMARTLIST_FOREACH(status, char *, cp, tor_free(cp));
- smartlist_free(status);
- } else if (!strcmp(question, "stream-status")) {
- connection_t **conns;
- int n_conns, i;
- char buf[256];
- smartlist_t *status = smartlist_create();
- get_connection_array(&conns, &n_conns);
- for (i=0; i < n_conns; ++i) {
- const char *state;
- char *s;
- size_t slen;
- circuit_t *circ;
- if (conns[i]->type != CONN_TYPE_AP ||
- conns[i]->marked_for_close ||
- conns[i]->state == AP_CONN_STATE_SOCKS_WAIT)
- continue;
- switch (conns[i]->state)
- {
- case AP_CONN_STATE_CONTROLLER_WAIT:
- case AP_CONN_STATE_CIRCUIT_WAIT:
- if (conns[i]->socks_request &&
- conns[i]->socks_request->command == SOCKS_COMMAND_RESOLVE)
- state = "NEWRESOLVE";
- else
- state = "NEW";
- break;
- case AP_CONN_STATE_RENDDESC_WAIT:
- case AP_CONN_STATE_CONNECT_WAIT:
- state = "SENTCONNECT"; break;
- case AP_CONN_STATE_RESOLVE_WAIT:
- state = "SENTRESOLVE"; break;
- case AP_CONN_STATE_OPEN:
- state = "SUCCEEDED"; break;
- default:
- warn(LD_BUG, "Asked for stream in unknown state %d",
- conns[i]->state);
- continue;
- }
- circ = circuit_get_by_edge_conn(conns[i]);
- write_stream_target_to_buf(conns[i], buf, sizeof(buf));
- slen = strlen(buf)+strlen(state)+32;
- s = tor_malloc(slen+1);
- tor_snprintf(s, slen, "%lu %s %lu %s",
- (unsigned long) conns[i]->global_identifier,state,
- circ?(unsigned long)circ->global_identifier : 0ul,
- buf);
- smartlist_add(status, s);
- }
- *answer = smartlist_join_strings(status, "\r\n", 0, NULL);
- SMARTLIST_FOREACH(status, char *, cp, tor_free(cp));
- smartlist_free(status);
- } else if (!strcmp(question, "orconn-status")) {
- connection_t **conns;
- int n_conns, i;
- smartlist_t *status = smartlist_create();
- get_connection_array(&conns, &n_conns);
- for (i=0; i < n_conns; ++i) {
- const char *state;
- char *s;
- size_t slen;
- if (conns[i]->type != CONN_TYPE_OR || conns[i]->marked_for_close)
- continue;
- if (conns[i]->state == OR_CONN_STATE_OPEN)
- state = "CONNECTED";
- else
- state = "LAUNCHED";
- slen = strlen(conns[i]->nickname)+strlen(state)+2;
- s = tor_malloc(slen+1);
- tor_snprintf(s, slen, "%s %s",conns[i]->nickname,state);
- smartlist_add(status, s);
- }
- *answer = smartlist_join_strings(status, "\r\n", 0, NULL);
- SMARTLIST_FOREACH(status, char *, cp, tor_free(cp));
- smartlist_free(status);
- } else if (!strcmpstart(question, "addr-mappings/")) {
- time_t min_e, max_e;
- smartlist_t *mappings;
- if (!strcmp(question, "addr-mappings/all")) {
- min_e = 0; max_e = TIME_MAX;
- } else if (!strcmp(question, "addr-mappings/cache")) {
- min_e = 2; max_e = TIME_MAX;
- } else if (!strcmp(question, "addr-mappings/config")) {
- min_e = 0; max_e = 0;
- } else if (!strcmp(question, "addr-mappings/control")) {
- min_e = 1; max_e = 1;
- } else {
- return 0;
- }
- mappings = smartlist_create();
- addressmap_get_mappings(mappings, min_e, max_e);
- *answer = smartlist_join_strings(mappings, "\n", 0, NULL);
- SMARTLIST_FOREACH(mappings, char *, cp, tor_free(cp));
- smartlist_free(mappings);
- }
- return 0;
- }
- /** Called when we receive a GETINFO command. Try to fetch all requested
- * information, and reply with information or error message. */
- static int
- handle_control_getinfo(connection_t *conn, uint32_t len, const char *body)
- {
- smartlist_t *questions = NULL;
- smartlist_t *answers = NULL;
- smartlist_t *unrecognized = NULL;
- char *msg = NULL, *ans = NULL;
- size_t msg_len;
- int v0 = STATE_IS_V0(conn->state);
- questions = smartlist_create();
- if (v0)
- smartlist_split_string(questions, body, "\n",
- SPLIT_SKIP_SPACE|SPLIT_IGNORE_BLANK, 0);
- else
- smartlist_split_string(questions, body, " ",
- SPLIT_SKIP_SPACE|SPLIT_IGNORE_BLANK, 0);
- answers = smartlist_create();
- unrecognized = smartlist_create();
- SMARTLIST_FOREACH(questions, const char *, q,
- {
- if (handle_getinfo_helper(q, &ans) < 0) {
- if (v0)
- send_control0_error(conn, ERR_INTERNAL, body);
- else
- connection_write_str_to_buf("551 Internal error\r\n", conn);
- goto done;
- }
- if (!ans) {
- if (v0) {
- send_control0_error(conn, ERR_UNRECOGNIZED_CONFIG_KEY, body);
- goto done;
- } else
- smartlist_add(unrecognized, (char*)q);
- } else {
- smartlist_add(answers, tor_strdup(q));
- smartlist_add(answers, ans);
- }
- });
- if (smartlist_len(unrecognized)) {
- int i;
- tor_assert(!v0);
- for (i=0; i < smartlist_len(unrecognized)-1; ++i)
- connection_printf_to_buf(conn,
- "552-Unrecognized key \"%s\"\r\n",
- (char*)smartlist_get(unrecognized, i));
- connection_printf_to_buf(conn,
- "552 Unrecognized key \"%s\"\r\n",
- (char*)smartlist_get(unrecognized, i));
- goto done;
- }
- if (v0) {
- msg = smartlist_join_strings2(answers, "\0", 1, 1, &msg_len);
- tor_assert(msg_len > 0); /* it will at least be terminated */
- send_control0_message(conn, CONTROL0_CMD_INFOVALUE,
- msg_len, msg);
- } else {
- int i;
- for (i = 0; i < smartlist_len(answers); i += 2) {
- char *k = smartlist_get(answers, i);
- char *v = smartlist_get(answers, i+1);
- /*XXXX Not an adequate test! XXXX011 */
- if (!strchr(v, '\n') && !strchr(v, '\r')) {
- connection_printf_to_buf(conn, "250-%s=", k);
- connection_write_str_to_buf(v, conn);
- connection_write_str_to_buf("\r\n", conn);
- } else {
- char *esc = NULL;
- size_t len;
- len = write_escaped_data(v, strlen(v), 1, &esc);
- connection_printf_to_buf(conn, "250+%s=\r\n", k);
- connection_write_to_buf(esc, len, conn);
- tor_free(esc);
- }
- }
- connection_write_str_to_buf("250 OK\r\n", conn);
- }
- done:
- if (answers) {
- SMARTLIST_FOREACH(answers, char *, cp, tor_free(cp));
- smartlist_free(answers);
- }
- if (questions) {
- SMARTLIST_FOREACH(questions, char *, cp, tor_free(cp));
- smartlist_free(questions);
- }
- smartlist_free(unrecognized);
- tor_free(msg);
- return 0;
- }
- /** Called when we get an EXTENDCIRCUIT message. Try to extend the listed
- * circuit, and report success or failure. */
- static int
- handle_control_extendcircuit(connection_t *conn, uint32_t len,
- const char *body)
- {
- smartlist_t *router_nicknames=NULL, *routers=NULL;
- uint32_t circ_id;
- circuit_t *circ = NULL;
- int zero_circ, v0;
- char reply[4];
- v0 = STATE_IS_V0(conn->state);
- router_nicknames = smartlist_create();
- if (v0) {
- if (len<5) {
- send_control0_error(conn, ERR_SYNTAX, "extendcircuit message too short");
- goto done;
- }
- smartlist_split_string(router_nicknames, body+4, ",", 0, 0);
- circ_id = ntohl(get_uint32(body));
- if (!circ_id) {
- /* start a new circuit */
- zero_circ = 1;
- } else {
- circ = circuit_get_by_global_id(circ_id);
- zero_circ = 0;
- if (!circ) {
- send_control0_error(conn, ERR_NO_CIRC,
- "No circuit found with given ID");
- goto done;
- }
- }
- } else { /* v1 */
- smartlist_t *args;
- args = smartlist_create();
- smartlist_split_string(args, body, " ",
- SPLIT_SKIP_SPACE|SPLIT_IGNORE_BLANK, 0);
- if (smartlist_len(args)<2) {
- connection_printf_to_buf(conn,
- "512 Missing argument to EXTENDCIRCUIT\r\n");
- SMARTLIST_FOREACH(args, char *, cp, tor_free(cp));
- smartlist_free(args);
- goto done;
- }
- zero_circ = !strcmp("0", (char*)smartlist_get(args,0));
- if (!zero_circ && !(circ = get_circ(smartlist_get(args,0)))) {
- connection_printf_to_buf(conn, "552 Unknown circuit \"%s\"\r\n",
- (char*)smartlist_get(args, 0));
- }
- smartlist_split_string(router_nicknames, smartlist_get(args,1), ",", 0, 0);
- SMARTLIST_FOREACH(args, char *, cp, tor_free(cp));
- smartlist_free(args);
- if (!zero_circ && !circ) {
- goto done;
- }
- }
- routers = smartlist_create();
- SMARTLIST_FOREACH(router_nicknames, const char *, n,
- {
- routerinfo_t *r = router_get_by_nickname(n, 1);
- if (!r) {
- if (v0)
- send_control0_error(conn, ERR_NO_ROUTER, n);
- else
- connection_printf_to_buf(conn, "552 No such router \"%s\"\r\n", n);
- goto done;
- }
- smartlist_add(routers, r);
- });
- if (!smartlist_len(routers)) {
- if (v0)
- send_control0_error(conn, ERR_SYNTAX, "No router names provided");
- else
- connection_write_str_to_buf("512 No router names provided\r\n", conn);
- goto done;
- }
- if (zero_circ) {
- /* start a new circuit */
- circ = circuit_init(CIRCUIT_PURPOSE_C_GENERAL, 0, 0, 0);
- }
- /* now circ refers to something that is ready to be extended */
- SMARTLIST_FOREACH(routers, routerinfo_t *, r,
- {
- extend_info_t *info = extend_info_from_router(r);
- circuit_append_new_exit(circ, info);
- extend_info_free(info);
- });
- /* now that we've populated the cpath, start extending */
- if (zero_circ) {
- if (circuit_handle_first_hop(circ) < 0) {
- circuit_mark_for_close(circ);
- if (v0)
- send_control0_error(conn, ERR_INTERNAL, "couldn't start circuit");
- else
- connection_write_str_to_buf("551 Couldn't start circuit\r\n", conn);
- goto done;
- }
- } else {
- if (circ->state == CIRCUIT_STATE_OPEN) {
- circuit_set_state(circ, CIRCUIT_STATE_BUILDING);
- if (circuit_send_next_onion_skin(circ) < 0) {
- info(LD_CONTROL,
- "send_next_onion_skin failed; circuit marked for closing.");
- circuit_mark_for_close(circ);
- if (v0)
- send_control0_error(conn, ERR_INTERNAL, "couldn't send onion skin");
- else
- connection_write_str_to_buf("551 Couldn't send onion skinr\n", conn);
- goto done;
- }
- }
- }
- if (v0) {
- set_uint32(reply, htonl(circ->global_identifier));
- send_control_done2(conn, reply, sizeof(reply));
- } else {
- connection_printf_to_buf(conn, "250 EXTENDED %lu\r\n",
- (unsigned long)circ->global_identifier);
- }
- done:
- SMARTLIST_FOREACH(router_nicknames, char *, n, tor_free(n));
- smartlist_free(router_nicknames);
- if (routers)
- smartlist_free(routers);
- return 0;
- }
- /** Called when we get an ATTACHSTREAM message. Try to attach the requested
- * stream, and report success or failure. */
- static int
- handle_control_attachstream(connection_t *conn, uint32_t len,
- const char *body)
- {
- connection_t *ap_conn = NULL;
- circuit_t *circ = NULL;
- int zero_circ;
- if (STATE_IS_V0(conn->state)) {
- uint32_t conn_id;
- uint32_t circ_id;
- if (len < 8) {
- send_control0_error(conn, ERR_SYNTAX, "attachstream message too short");
- return 0;
- }
- conn_id = ntohl(get_uint32(body));
- circ_id = ntohl(get_uint32(body+4));
- zero_circ = circ_id == 0;
- if (!(ap_conn = connection_get_by_global_id(conn_id))) {
- send_control0_error(conn, ERR_NO_STREAM,
- "No connection found with given ID");
- return 0;
- }
- if (circ_id && !(circ = circuit_get_by_global_id(circ_id))) {
- send_control0_error(conn, ERR_NO_CIRC, "No circuit found with given ID");
- return 0;
- }
- } else {
- smartlist_t *args;
- args = smartlist_create();
- smartlist_split_string(args, body, " ",
- SPLIT_SKIP_SPACE|SPLIT_IGNORE_BLANK, 0);
- if (smartlist_len(args)<2) {
- connection_printf_to_buf(conn,
- "512 Missing argument to ATTACHSTREAM\r\n");
- SMARTLIST_FOREACH(args, char *, cp, tor_free(cp));
- smartlist_free(args);
- return 0;
- }
- zero_circ = !strcmp("0", (char*)smartlist_get(args,1));
- if (!(ap_conn = get_stream(smartlist_get(args, 0)))) {
- connection_printf_to_buf(conn, "552 Unknown stream \"%s\"\r\n",
- (char*)smartlist_get(args, 0));
- } else if (!zero_circ && !(circ = get_circ(smartlist_get(args, 1)))) {
- connection_printf_to_buf(conn, "552 Unknown circuit \"%s\"\r\n",
- (char*)smartlist_get(args, 1));
- }
- SMARTLIST_FOREACH(args, char *, cp, tor_free(cp));
- smartlist_free(args);
- if (!ap_conn || (!zero_circ && !circ))
- return 0;
- }
- if (ap_conn->state != AP_CONN_STATE_CONTROLLER_WAIT) {
- if (STATE_IS_V0(conn->state)) {
- send_control0_error(conn, ERR_NO_STREAM,
- "Connection is not managed by controller.");
- } else {
- connection_write_str_to_buf(
- "555 Connection is not managed by controller.\r\n",
- conn);
- }
- return 0;
- }
- if (zero_circ) {
- ap_conn->state = AP_CONN_STATE_CIRCUIT_WAIT;
- if (connection_ap_handshake_attach_circuit(ap_conn)<0)
- connection_mark_unattached_ap(ap_conn, END_STREAM_REASON_CANT_ATTACH);
- send_control_done(conn);
- return 0;
- }
- if (circ->state != CIRCUIT_STATE_OPEN) {
- if (STATE_IS_V0(conn->state))
- send_control0_error(conn, ERR_INTERNAL,
- "Refuse to attach stream to non-open circ.");
- else
- connection_write_str_to_buf(
- "551 Can't attach stream to non-open circuit\r\n",
- conn);
- return 0;
- }
- if (connection_ap_handshake_attach_chosen_circuit(ap_conn, circ) != 1) {
- if (STATE_IS_V0(conn->state))
- send_control0_error(conn, ERR_INTERNAL, "Unable to attach stream.");
- else
- connection_write_str_to_buf("551 Unable to attach stream\r\n", conn);
- return 0;
- }
- send_control_done(conn);
- return 0;
- }
- /** Called when we get a POSTDESCRIPTOR message. Try to learn the provided
- * descriptor, and report success or failure. */
- static int
- handle_control_postdescriptor(connection_t *conn, uint32_t len,
- const char *body)
- {
- char *desc;
- int v0 = STATE_IS_V0(conn->state);
- const char *msg=NULL;
- if (v0)
- desc = (char*)body;
- else {
- const char *cp = memchr(body, '\n', len);
- tor_assert(cp);
- read_escaped_data(cp, len-(cp-body), 1, &desc);
- }
- switch (router_load_single_router(desc, &msg)) {
- case -1:
- if (!msg) msg = "Could not parse descriptor";
- if (v0)
- send_control0_error(conn,ERR_SYNTAX,msg);
- else
- connection_printf_to_buf(conn, "554 %s\r\n", msg);
- break;
- case 0:
- if (!msg) msg = "Descriptor not added";
- if (v0)
- send_control_done2(conn,msg,0);
- else
- connection_printf_to_buf(conn, "251 %s\r\n",msg);
- break;
- case 1:
- send_control_done(conn);
- break;
- }
- if (!v0)
- tor_free(desc);
- return 0;
- }
- /** Called when we receive a REDIRECTSTERAM command. Try to change the target
- * address of the named AP stream, and report success or failure. */
- static int
- handle_control_redirectstream(connection_t *conn, uint32_t len,
- const char *body)
- {
- connection_t *ap_conn = NULL;
- uint32_t conn_id;
- char *new_addr = NULL;
- uint16_t new_port = 0;
- if (STATE_IS_V0(conn->state)) {
- if (len < 6) {
- send_control0_error(conn, ERR_SYNTAX,
- "redirectstream message too short");
- return 0;
- }
- conn_id = ntohl(get_uint32(body));
- if (!(ap_conn = connection_get_by_global_id(conn_id))
- || ap_conn->state != CONN_TYPE_AP
- || !ap_conn->socks_request) {
- send_control0_error(conn, ERR_NO_STREAM,
- "No AP connection found with given ID");
- return 0;
- }
- new_addr = tor_strdup(body+4);
- } else {
- smartlist_t *args;
- args = smartlist_create();
- smartlist_split_string(args, body, " ",
- SPLIT_SKIP_SPACE|SPLIT_IGNORE_BLANK, 0);
- if (smartlist_len(args) < 2)
- connection_printf_to_buf(conn,
- "512 Missing argument to REDIRECTSTREAM\r\n");
- else if (!(ap_conn = get_stream(smartlist_get(args, 0)))
- || !ap_conn->socks_request) {
- connection_printf_to_buf(conn, "552 Unknown stream \"%s\"\r\n",
- (char*)smartlist_get(args, 0));
- } else {
- int ok;
- if (smartlist_len(args) > 2) { /* they included a port too */
- new_port = (uint16_t) tor_parse_ulong(smartlist_get(args, 2),
- 10, 1, 65535, &ok, NULL);
- }
- if (!ok) {
- connection_printf_to_buf(conn, "512 Cannot parse port \"%s\"\r\n",
- (char*)smartlist_get(args, 2));
- } else {
- new_addr = tor_strdup(smartlist_get(args, 1));
- }
- }
- SMARTLIST_FOREACH(args, char *, cp, tor_free(cp));
- smartlist_free(args);
- if (!new_addr)
- return 0;
- }
- strlcpy(ap_conn->socks_request->address, new_addr,
- sizeof(ap_conn->socks_request->address));
- if (new_port)
- ap_conn->socks_request->port = new_port;
- tor_free(new_addr);
- send_control_done(conn);
- return 0;
- }
- /** Called when we get a CLOSESTREAM command; try to close the named stream
- * and report success or failure. */
- static int
- handle_control_closestream(connection_t *conn, uint32_t len,
- const char *body)
- {
- connection_t *ap_conn=NULL;
- uint8_t reason=0;
- if (STATE_IS_V0(conn->state)) {
- uint32_t conn_id;
- if (len < 6) {
- send_control0_error(conn, ERR_SYNTAX, "closestream message too short");
- return 0;
- }
- conn_id = ntohl(get_uint32(body));
- reason = *(uint8_t*)(body+4);
- if (!(ap_conn = connection_get_by_global_id(conn_id))
- || ap_conn->state != CONN_TYPE_AP
- || !ap_conn->socks_request) {
- send_control0_error(conn, ERR_NO_STREAM,
- "No AP connection found with given ID");
- return 0;
- }
- } else {
- smartlist_t *args;
- int ok;
- args = smartlist_create();
- smartlist_split_string(args, body, " ",
- SPLIT_SKIP_SPACE|SPLIT_IGNORE_BLANK, 0);
- if (smartlist_len(args)<2)
- connection_printf_to_buf(conn,
- "512 Missing argument to CLOSESTREAM\r\n");
- else if (!(ap_conn = get_stream(smartlist_get(args, 0))))
- connection_printf_to_buf(conn, "552 Unknown stream \"%s\"\r\n",
- (char*)smartlist_get(args, 0));
- else {
- reason = (uint8_t) tor_parse_ulong(smartlist_get(args,1), 10, 0, 255,
- &ok, NULL);
- if (!ok) {
- connection_printf_to_buf(conn, "552 Unrecognized reason \"%s\"\r\n",
- (char*)smartlist_get(args, 1));
- ap_conn = NULL;
- }
- }
- SMARTLIST_FOREACH(args, char *, cp, tor_free(cp));
- smartlist_free(args);
- if (!ap_conn)
- return 0;
- }
- connection_mark_unattached_ap(ap_conn, reason);
- send_control_done(conn);
- return 0;
- }
- /** Called when we get a CLOSECIRCUIT command; try to close the named circuit
- * and report success or failure. */
- static int
- handle_control_closecircuit(connection_t *conn, uint32_t len,
- const char *body)
- {
- circuit_t *circ = NULL;
- int safe = 0;
- if (STATE_IS_V0(conn->state)) {
- uint32_t circ_id;
- if (len < 5) {
- send_control0_error(conn, ERR_SYNTAX, "closecircuit message too short");
- return 0;
- }
- circ_id = ntohl(get_uint32(body));
- safe = (*(uint8_t*)(body+4)) & 1;
- if (!(circ = circuit_get_by_global_id(circ_id))) {
- send_control0_error(conn, ERR_NO_CIRC,
- "No circuit found with given ID");
- return 0;
- }
- } else {
- smartlist_t *args;
- args = smartlist_create();
- smartlist_split_string(args, body, " ",
- SPLIT_SKIP_SPACE|SPLIT_IGNORE_BLANK, 0);
- if (smartlist_len(args)<1)
- connection_printf_to_buf(conn,
- "512 Missing argument to CLOSECIRCUIT\r\n");
- else if (!(circ=get_circ(smartlist_get(args, 0))))
- connection_printf_to_buf(conn, "552 Unknown circuit \"%s\"\r\n",
- (char*)smartlist_get(args, 0));
- else {
- int i;
- for (i=1; i < smartlist_len(args); ++i) {
- if (!strcasecmp(smartlist_get(args, i), "IfUnused"))
- safe = 1;
- else
- info(LD_CONTROL, "Skipping unknown option %s",
- (char*)smartlist_get(args,i));
- }
- }
- SMARTLIST_FOREACH(args, char *, cp, tor_free(cp));
- smartlist_free(args);
- if (!circ)
- return 0;
- }
- if (!safe || !circ->p_streams) {
- circuit_mark_for_close(circ);
- }
- send_control_done(conn);
- return 0;
- }
- /** Called when we get a v0 FRAGMENTHEADER or FRAGMENT command; try to append
- * the data to conn->incoming_cmd, setting conn->incoming_(type|len|cur_len)
- * as appropriate. If the command is malformed, drop it and all pending
- * fragments and report failure.
- */
- static int
- handle_control_fragments(connection_t *conn, uint16_t command_type,
- uint32_t body_len, char *body)
- {
- if (command_type == CONTROL0_CMD_FRAGMENTHEADER) {
- if (conn->incoming_cmd) {
- warn(LD_CONTROL, "Dropping incomplete fragmented command");
- tor_free(conn->incoming_cmd);
- }
- if (body_len < 6) {
- send_control0_error(conn, ERR_SYNTAX, "FRAGMENTHEADER too short.");
- return 0;
- }
- conn->incoming_cmd_type = ntohs(get_uint16(body));
- conn->incoming_cmd_len = ntohl(get_uint32(body+2));
- conn->incoming_cmd_cur_len = 0;
- conn->incoming_cmd = tor_malloc(conn->incoming_cmd_len);
- body += 6;
- body_len -= 6;
- } else if (command_type == CONTROL0_CMD_FRAGMENT) {
- if (!conn->incoming_cmd) {
- send_control0_error(conn, ERR_SYNTAX, "Out-of-place FRAGMENT");
- return 0;
- }
- } else {
- tor_assert(0);
- }
- if (conn->incoming_cmd_cur_len + body_len > conn->incoming_cmd_len) {
- tor_free(conn->incoming_cmd);
- send_control0_error(conn, ERR_SYNTAX,
- "Fragmented data exceeds declared length");
- return 0;
- }
- memcpy(conn->incoming_cmd + conn->incoming_cmd_cur_len,
- body, body_len);
- conn->incoming_cmd_cur_len += body_len;
- return 0;
- }
- /** Called when <b>conn</b> has no more bytes left on its outbuf. */
- int
- connection_control_finished_flushing(connection_t *conn)
- {
- tor_assert(conn);
- tor_assert(conn->type == CONN_TYPE_CONTROL);
- connection_stop_writing(conn);
- return 0;
- }
- /** Called when <b>conn</b> has gotten its socket closed. */
- int
- connection_control_reached_eof(connection_t *conn)
- {
- tor_assert(conn);
- tor_assert(conn->type == CONN_TYPE_CONTROL);
- info(LD_CONTROL,"Control connection reached EOF. Closing.");
- connection_mark_for_close(conn);
- return 0;
- }
- /** Called when data has arrived on a v1 control connection: Try to fetch
- * commands from conn->inbuf, and execute them.
- */
- static int
- connection_control_process_inbuf_v1(connection_t *conn)
- {
- size_t data_len;
- int cmd_len;
- char *args;
- tor_assert(conn);
- tor_assert(conn->type == CONN_TYPE_CONTROL);
- tor_assert(conn->state == CONTROL_CONN_STATE_OPEN_V1 ||
- conn->state == CONTROL_CONN_STATE_NEEDAUTH_V1);
- if (!conn->incoming_cmd) {
- conn->incoming_cmd = tor_malloc(1024);
- conn->incoming_cmd_len = 1024;
- conn->incoming_cmd_cur_len = 0;
- }
- again:
- while (1) {
- size_t last_idx;
- int r;
- /* First, fetch a line. */
- do {
- data_len = conn->incoming_cmd_len - conn->incoming_cmd_cur_len;
- r = fetch_from_buf_line(conn->inbuf,
- conn->incoming_cmd+conn->incoming_cmd_cur_len,
- &data_len);
- if (r == 0)
- /* Line not all here yet. Wait. */
- return 0;
- else if (r == -1) {
- while (conn->incoming_cmd_len < data_len+conn->incoming_cmd_cur_len)
- conn->incoming_cmd_len *= 2;
- conn->incoming_cmd = tor_realloc(conn->incoming_cmd,
- conn->incoming_cmd_len);
- }
- } while (r != 1);
- tor_assert(data_len);
- last_idx = conn->incoming_cmd_cur_len;
- conn->incoming_cmd_cur_len += data_len;
- /* We have appended a line to incoming_cmd. Is the command done? */
- if (last_idx == 0 && *conn->incoming_cmd != '+')
- /* One line command, didn't start with '+'. */
- break;
- if (last_idx+3 == conn->incoming_cmd_cur_len &&
- !memcmp(conn->incoming_cmd + last_idx, ".\r\n", 3)) {
- /* Just appended ".\r\n"; we're done. Remove it. */
- conn->incoming_cmd_cur_len -= 3;
- break;
- }
- /* Otherwise, read another line. */
- }
- data_len = conn->incoming_cmd_cur_len;
- /* Okay, we now have a command sitting on conn->incoming_cmd. See if we
- * recognize it.
- */
- cmd_len = 0;
- while ((size_t)cmd_len < data_len
- && !TOR_ISSPACE(conn->incoming_cmd[cmd_len]))
- ++cmd_len;
- data_len -= cmd_len;
- conn->incoming_cmd[cmd_len]='\0';
- args = conn->incoming_cmd+cmd_len+1;
- while (*args == ' ' || *args == '\t') {
- ++args;
- --data_len;
- }
- if (!strcasecmp(conn->incoming_cmd, "QUIT")) {
- connection_write_str_to_buf("250 closing connection\r\n", conn);
- connection_mark_for_close(conn);
- return 0;
- }
- if (conn->state == CONTROL_CONN_STATE_NEEDAUTH_V1 &&
- strcasecmp(conn->incoming_cmd, "AUTHENTICATE")) {
- connection_write_str_to_buf("514 Authentication required.\r\n", conn);
- conn->incoming_cmd_cur_len = 0;
- goto again;
- }
- if (!strcasecmp(conn->incoming_cmd, "SETCONF")) {
- if (handle_control_setconf(conn, data_len, args))
- return -1;
- } else if (!strcasecmp(conn->incoming_cmd, "RESETCONF")) {
- if (handle_control_resetconf(conn, data_len, args))
- return -1;
- } else if (!strcasecmp(conn->incoming_cmd, "GETCONF")) {
- if (handle_control_getconf(conn, data_len, args))
- return -1;
- } else if (!strcasecmp(conn->incoming_cmd, "SETEVENTS")) {
- if (handle_control_setevents(conn, data_len, args))
- return -1;
- } else if (!strcasecmp(conn->incoming_cmd, "AUTHENTICATE")) {
- if (handle_control_authenticate(conn, data_len, args))
- return -1;
- } else if (!strcasecmp(conn->incoming_cmd, "SAVECONF")) {
- if (handle_control_saveconf(conn, data_len, args))
- return -1;
- } else if (!strcasecmp(conn->incoming_cmd, "SIGNAL")) {
- if (handle_control_signal(conn, data_len, args))
- return -1;
- } else if (!strcasecmp(conn->incoming_cmd, "MAPADDRESS")) {
- if (handle_control_mapaddress(conn, data_len, args))
- return -1;
- } else if (!strcasecmp(conn->incoming_cmd, "GETINFO")) {
- if (handle_control_getinfo(conn, data_len, args))
- return -1;
- } else if (!strcasecmp(conn->incoming_cmd, "EXTENDCIRCUIT")) {
- if (handle_control_extendcircuit(conn, data_len, args))
- return -1;
- } else if (!strcasecmp(conn->incoming_cmd, "ATTACHSTREAM")) {
- if (handle_control_attachstream(conn, data_len, args))
- return -1;
- } else if (!strcasecmp(conn->incoming_cmd, "+POSTDESCRIPTOR")) {
- if (handle_control_postdescriptor(conn, data_len, args))
- return -1;
- } else if (!strcasecmp(conn->incoming_cmd, "REDIRECTSTREAM")) {
- if (handle_control_redirectstream(conn, data_len, args))
- return -1;
- } else if (!strcasecmp(conn->incoming_cmd, "CLOSESTREAM")) {
- if (handle_control_closestream(conn, data_len, args))
- return -1;
- } else if (!strcasecmp(conn->incoming_cmd, "CLOSECIRCUIT")) {
- if (handle_control_closecircuit(conn, data_len, args))
- return -1;
- } else {
- connection_printf_to_buf(conn, "510 Unrecognized command \"%s\"\r\n",
- conn->incoming_cmd);
- }
- conn->incoming_cmd_cur_len = 0;
- goto again;
- }
- /** Called when data has arrived on a v0 control connection: Try to fetch
- * commands from conn->inbuf, and execute them.
- */
- static int
- connection_control_process_inbuf_v0(connection_t *conn)
- {
- uint32_t body_len;
- uint16_t command_type;
- char *body=NULL;
- again:
- /* Try to suck a control message from the buffer. */
- switch (fetch_from_buf_control0(conn->inbuf, &body_len, &command_type, &body,
- conn->state == CONTROL_CONN_STATE_NEEDAUTH_V0))
- {
- case -2:
- tor_free(body);
- info(LD_CONTROL, "Detected v1 control protocol on connection (fd %d)",
- conn->s);
- conn->state = CONTROL_CONN_STATE_NEEDAUTH_V1;
- return connection_control_process_inbuf_v1(conn);
- case -1:
- tor_free(body);
- warn(LD_CONTROL, "Error in control command. Failing.");
- return -1;
- case 0:
- /* Control command not all here yet. Wait. */
- return 0;
- case 1:
- /* We got a command. Process it. */
- break;
- default:
- tor_assert(0);
- }
- /* We got a command. If we need authentication, only authentication
- * commands will be considered. */
- if (conn->state == CONTROL_CONN_STATE_NEEDAUTH_V0 &&
- command_type != CONTROL0_CMD_AUTHENTICATE) {
- info(LD_CONTROL, "Rejecting '%s' command; authentication needed.",
- control_cmd_to_string(command_type));
- send_control0_error(conn, ERR_UNAUTHORIZED, "Authentication required");
- tor_free(body);
- goto again;
- }
- if (command_type == CONTROL0_CMD_FRAGMENTHEADER ||
- command_type == CONTROL0_CMD_FRAGMENT) {
- if (handle_control_fragments(conn, command_type, body_len, body))
- return -1;
- tor_free(body);
- if (conn->incoming_cmd_cur_len != conn->incoming_cmd_len)
- goto again;
- command_type = conn->incoming_cmd_type;
- body_len = conn->incoming_cmd_len;
- body = conn->incoming_cmd;
- conn->incoming_cmd = NULL;
- } else if (conn->incoming_cmd) {
- warn(LD_CONTROL, "Dropping incomplete fragmented command");
- tor_free(conn->incoming_cmd);
- }
- /* Okay, we're willing to process the command. */
- switch (command_type)
- {
- case CONTROL0_CMD_SETCONF:
- if (handle_control_setconf(conn, body_len, body))
- return -1;
- break;
- case CONTROL0_CMD_GETCONF:
- if (handle_control_getconf(conn, body_len, body))
- return -1;
- break;
- case CONTROL0_CMD_SETEVENTS:
- if (handle_control_setevents(conn, body_len, body))
- return -1;
- break;
- case CONTROL0_CMD_AUTHENTICATE:
- if (handle_control_authenticate(conn, body_len, body))
- return -1;
- break;
- case CONTROL0_CMD_SAVECONF:
- if (handle_control_saveconf(conn, body_len, body))
- return -1;
- break;
- case CONTROL0_CMD_SIGNAL:
- if (handle_control_signal(conn, body_len, body))
- return -1;
- break;
- case CONTROL0_CMD_MAPADDRESS:
- if (handle_control_mapaddress(conn, body_len, body))
- return -1;
- break;
- case CONTROL0_CMD_GETINFO:
- if (handle_control_getinfo(conn, body_len, body))
- return -1;
- break;
- case CONTROL0_CMD_EXTENDCIRCUIT:
- if (handle_control_extendcircuit(conn, body_len, body))
- return -1;
- break;
- case CONTROL0_CMD_ATTACHSTREAM:
- if (handle_control_attachstream(conn, body_len, body))
- return -1;
- break;
- case CONTROL0_CMD_POSTDESCRIPTOR:
- if (handle_control_postdescriptor(conn, body_len, body))
- return -1;
- break;
- case CONTROL0_CMD_REDIRECTSTREAM:
- if (handle_control_redirectstream(conn, body_len, body))
- return -1;
- break;
- case CONTROL0_CMD_CLOSESTREAM:
- if (handle_control_closestream(conn, body_len, body))
- return -1;
- break;
- case CONTROL0_CMD_CLOSECIRCUIT:
- if (handle_control_closecircuit(conn, body_len, body))
- return -1;
- break;
- case CONTROL0_CMD_ERROR:
- case CONTROL0_CMD_DONE:
- case CONTROL0_CMD_CONFVALUE:
- case CONTROL0_CMD_EVENT:
- case CONTROL0_CMD_INFOVALUE:
- warn(LD_CONTROL, "Received client-only '%s' command; ignoring.",
- control_cmd_to_string(command_type));
- send_control0_error(conn, ERR_UNRECOGNIZED_TYPE,
- "Command type only valid from server to tor client");
- break;
- case CONTROL0_CMD_FRAGMENTHEADER:
- case CONTROL0_CMD_FRAGMENT:
- warn(LD_CONTROL, "Recieved command fragment out of order; ignoring.");
- send_control0_error(conn, ERR_SYNTAX, "Bad fragmentation on command.");
- default:
- warn(LD_CONTROL, "Received unrecognized command type %d; ignoring.",
- (int)command_type);
- send_control0_error(conn, ERR_UNRECOGNIZED_TYPE,
- "Unrecognized command type");
- break;
- }
- tor_free(body);
- goto again; /* There might be more data. */
- }
- /** Called when <b>conn</b> has received more bytes on its inbuf.
- */
- int
- connection_control_process_inbuf(connection_t *conn)
- {
- tor_assert(conn);
- tor_assert(conn->type == CONN_TYPE_CONTROL);
- if (STATE_IS_V0(conn->state))
- return connection_control_process_inbuf_v0(conn);
- else
- return connection_control_process_inbuf_v1(conn);
- }
- /** Something has happened to circuit <b>circ</b>: tell any interested
- * control connections. */
- int
- control_event_circuit_status(circuit_t *circ, circuit_status_event_t tp)
- {
- char *path, *msg;
- if (!EVENT_IS_INTERESTING(EVENT_CIRCUIT_STATUS))
- return 0;
- tor_assert(circ);
- tor_assert(CIRCUIT_IS_ORIGIN(circ));
- path = circuit_list_path(circ,0);
- if (EVENT_IS_INTERESTING0(EVENT_CIRCUIT_STATUS)) {
- size_t path_len = strlen(path);
- msg = tor_malloc(1+4+path_len+1); /* event, circid, path, NUL. */
- msg[0] = (uint8_t) tp;
- set_uint32(msg+1, htonl(circ->global_identifier));
- strlcpy(msg+5,path,path_len+1);
- send_control0_event(EVENT_CIRCUIT_STATUS, (uint32_t)(path_len+6), msg);
- tor_free(msg);
- }
- if (EVENT_IS_INTERESTING1(EVENT_CIRCUIT_STATUS)) {
- const char *status;
- switch (tp)
- {
- case CIRC_EVENT_LAUNCHED: status = "LAUNCHED"; break;
- case CIRC_EVENT_BUILT: status = "BUILT"; break;
- case CIRC_EVENT_EXTENDED: status = "EXTENDED"; break;
- case CIRC_EVENT_FAILED: status = "FAILED"; break;
- case CIRC_EVENT_CLOSED: status = "CLOSED"; break;
- default:
- warn(LD_BUG, "Unrecognized status code %d", (int)tp);
- return 0;
- }
- send_control1_event(EVENT_CIRCUIT_STATUS,
- "650 CIRC %lu %s %s\r\n",
- (unsigned long)circ->global_identifier,
- status, path);
- }
- tor_free(path);
- return 0;
- }
- /** Given an AP connection <b>conn</b> and a <b>len</b>-character buffer
- * <b>buf</b>, determine the address:port combination requested on
- * <b>conn</b>, and write it to <b>buf</b>. Return 0 on success, -1 on
- * failure. */
- static int
- write_stream_target_to_buf(connection_t *conn, char *buf, size_t len)
- {
- char buf2[256];
- if (conn->chosen_exit_name)
- if (tor_snprintf(buf2, sizeof(buf2), ".%s.exit", conn->chosen_exit_name)<0)
- return -1;
- if (tor_snprintf(buf, len, "%s%s%s:%d",
- conn->socks_request->address,
- conn->chosen_exit_name ? buf2 : "",
- !conn->chosen_exit_name &&
- connection_edge_is_rendezvous_stream(conn) ? ".onion" : "",
- conn->socks_request->port)<0)
- return -1;
- return 0;
- }
- /** Something has happened to the stream associated with AP connection
- * <b>conn</b>: tell any interested control connections. */
- int
- control_event_stream_status(connection_t *conn, stream_status_event_t tp)
- {
- char *msg;
- size_t len;
- char buf[256];
- tor_assert(conn->type == CONN_TYPE_AP);
- tor_assert(conn->socks_request);
- if (!EVENT_IS_INTERESTING(EVENT_STREAM_STATUS))
- return 0;
- write_stream_target_to_buf(conn, buf, sizeof(buf));
- if (EVENT_IS_INTERESTING0(EVENT_STREAM_STATUS)) {
- len = strlen(buf);
- msg = tor_malloc(5+len+1);
- msg[0] = (uint8_t) tp;
- set_uint32(msg+1, htonl(conn->global_identifier));
- strlcpy(msg+5, buf, len+1);
- send_control0_event(EVENT_STREAM_STATUS, (uint32_t)(5+len+1), msg);
- tor_free(msg);
- }
- if (EVENT_IS_INTERESTING1(EVENT_STREAM_STATUS)) {
- const char *status;
- circuit_t *circ;
- switch (tp)
- {
- case STREAM_EVENT_SENT_CONNECT: status = "SENTCONNECT"; break;
- case STREAM_EVENT_SENT_RESOLVE: status = "SENTRESOLVE"; break;
- case STREAM_EVENT_SUCCEEDED: status = "SUCCEEDED"; break;
- case STREAM_EVENT_FAILED: status = "FAILED"; break;
- case STREAM_EVENT_CLOSED: status = "CLOSED"; break;
- case STREAM_EVENT_NEW: status = "NEW"; break;
- case STREAM_EVENT_NEW_RESOLVE: status = "NEWRESOLVE"; break;
- case STREAM_EVENT_FAILED_RETRIABLE: status = "DETACHED"; break;
- default:
- warn(LD_BUG, "Unrecognized status code %d", (int)tp);
- return 0;
- }
- circ = circuit_get_by_edge_conn(conn);
- send_control1_event(EVENT_STREAM_STATUS,
- "650 STREAM %lu %s %lu %s\r\n",
- (unsigned long)conn->global_identifier, status,
- circ?(unsigned long)circ->global_identifier : 0ul,
- buf);
- /* XXX need to specify its intended exit, etc? */
- }
- return 0;
- }
- /** Something has happened to the OR connection <b>conn</b>: tell any
- * interested control connections. */
- int
- control_event_or_conn_status(connection_t *conn,or_conn_status_event_t tp)
- {
- char buf[HEX_DIGEST_LEN+3]; /* status, dollar, identity, NUL */
- size_t len;
- tor_assert(conn->type == CONN_TYPE_OR);
- if (!EVENT_IS_INTERESTING(EVENT_OR_CONN_STATUS))
- return 0;
- if (EVENT_IS_INTERESTING0(EVENT_OR_CONN_STATUS)) {
- buf[0] = (uint8_t)tp;
- strlcpy(buf+1,conn->nickname,sizeof(buf)-1);
- len = strlen(buf+1);
- send_control0_event(EVENT_OR_CONN_STATUS, (uint32_t)(len+1), buf);
- }
- if (EVENT_IS_INTERESTING1(EVENT_OR_CONN_STATUS)) {
- const char *status;
- switch (tp)
- {
- case OR_CONN_EVENT_LAUNCHED: status = "LAUNCHED"; break;
- case OR_CONN_EVENT_CONNECTED: status = "CONNECTED"; break;
- case OR_CONN_EVENT_FAILED: status = "FAILED"; break;
- case OR_CONN_EVENT_CLOSED: status = "CLOSED"; break;
- default:
- warn(LD_BUG, "Unrecognized status code %d", (int)tp);
- return 0;
- }
- send_control1_event(EVENT_OR_CONN_STATUS,
- "650 ORCONN %s %s\r\n",
- conn->nickname, status);
- }
- return 0;
- }
- /** A second or more has elapsed: tell any interested control
- * connections how much bandwidth we used. */
- int
- control_event_bandwidth_used(uint32_t n_read, uint32_t n_written)
- {
- char buf[8];
- if (EVENT_IS_INTERESTING0(EVENT_BANDWIDTH_USED)) {
- set_uint32(buf, htonl(n_read));
- set_uint32(buf+4, htonl(n_written));
- send_control0_event(EVENT_BANDWIDTH_USED, 8, buf);
- }
- if (EVENT_IS_INTERESTING1(EVENT_BANDWIDTH_USED)) {
- send_control1_event(EVENT_BANDWIDTH_USED,
- "650 BW %lu %lu\r\n",
- (unsigned long)n_read,
- (unsigned long)n_written);
- }
- return 0;
- }
- /** Called when we are sending a log message to the controllers: suspend
- * sending further log messages to the controllers until we're done. Used by
- * CONN_LOG_PROTECT. */
- void
- disable_control_logging(void)
- {
- ++disable_log_messages;
- }
- /** We're done sending a log message to the controllers: re-enable controller
- * logging. Used by CONN_LOG_PROTECT. */
- void
- enable_control_logging(void)
- {
- if (--disable_log_messages < 0)
- tor_assert(0);
- }
- /** We got a log message: tell any interested control connections. */
- void
- control_event_logmsg(int severity, unsigned int domain, const char *msg)
- {
- int oldlog, event;
- if (disable_log_messages)
- return;
- oldlog = EVENT_IS_INTERESTING0(EVENT_LOG_OBSOLETE) &&
- (severity == LOG_NOTICE || severity == LOG_WARN || severity == LOG_ERR);
- event = log_severity_to_event(severity);
- if (event<0 || !EVENT_IS_INTERESTING0(event))
- event = 0;
- if (oldlog || event) {
- size_t len = strlen(msg);
- ++disable_log_messages;
- if (event)
- send_control0_event(event, (uint32_t)(len+1), msg);
- if (oldlog)
- send_control0_event(EVENT_LOG_OBSOLETE, (uint32_t)(len+1), msg);
- --disable_log_messages;
- }
- event = log_severity_to_event(severity);
- if (event >= 0 && EVENT_IS_INTERESTING1(event)) {
- char *b = NULL;
- const char *s;
- if (strchr(msg, '\n')) {
- char *cp;
- b = tor_strdup(msg);
- for (cp = b; *cp; ++cp)
- if (*cp == '\r' || *cp == '\n')
- *cp = ' ';
- }
- switch (severity) {
- case LOG_DEBUG: s = "DEBUG"; break;
- case LOG_INFO: s = "INFO"; break;
- case LOG_NOTICE: s = "NOTICE"; break;
- case LOG_WARN: s = "WARN"; break;
- case LOG_ERR: s = "ERR"; break;
- default: s = "UnknownLogSeverity"; break;
- }
- ++disable_log_messages;
- send_control1_event(event, "650 %s %s\r\n", s, b?b:msg);
- --disable_log_messages;
- tor_free(b);
- }
- }
- /** Called whenever we receive new router descriptors: tell any
- * interested control connections. <b>routers</b> is a list of
- * DIGEST_LEN-byte identity digests.
- */
- int
- control_event_descriptors_changed(smartlist_t *routers)
- {
- size_t len;
- char *msg;
- smartlist_t *identities;
- char buf[HEX_DIGEST_LEN+1];
- if (!EVENT_IS_INTERESTING(EVENT_NEW_DESC))
- return 0;
- identities = smartlist_create();
- SMARTLIST_FOREACH(routers, routerinfo_t *, r,
- {
- base16_encode(buf,sizeof(buf),r->cache_info.identity_digest,DIGEST_LEN);
- smartlist_add(identities, tor_strdup(buf));
- });
- if (EVENT_IS_INTERESTING0(EVENT_NEW_DESC)) {
- msg = smartlist_join_strings(identities, ",", 0, &len);
- send_control0_event(EVENT_NEW_DESC, len+1, msg);
- tor_free(msg);
- }
- if (EVENT_IS_INTERESTING1(EVENT_NEW_DESC)) {
- msg = smartlist_join_strings(identities, " ", 0, &len);
- send_control1_event(EVENT_NEW_DESC, "650 NEWDESC %s\r\n", msg);
- tor_free(msg);
- }
- SMARTLIST_FOREACH(identities, char *, cp, tor_free(cp));
- smartlist_free(identities);
- return 0;
- }
- /** Called whenever an address mapping on <b>from<b> from changes to <b>to</b>.
- * <b>expires</b> values less than 3 are special; see connection_edge.c. */
- int
- control_event_address_mapped(const char *from, const char *to, time_t expires)
- {
- if (!EVENT_IS_INTERESTING1(EVENT_ADDRMAP))
- return 0;
- if (expires < 3)
- send_control1_event(EVENT_ADDRMAP,
- "650 ADDRMAP %s %s NEVER\r\n", from, to);
- else {
- char buf[ISO_TIME_LEN+1];
- format_local_iso_time(buf,expires);
- send_control1_event(EVENT_ADDRMAP, "650 ADDRMAP %s %s \"%s\"\r\n",
- from, to, buf);
- }
- return 0;
- }
- /** The authoritative dirserver has received a new descriptor that
- * has passed basic syntax checks and is properly self-signed.
- *
- * Notify any interested party of the new descriptor and what has
- * been done with it, and also optionally give an explanation/reason. */
- int
- control_event_or_authdir_new_descriptor(const char *action,
- const char *descriptor,
- const char *msg)
- {
- char firstline[1024];
- char *buf;
- int totallen;
- char *esc = NULL;
- size_t esclen;
- if (!EVENT_IS_INTERESTING(EVENT_AUTHDIR_NEWDESCS))
- return 0;
- tor_snprintf(firstline, sizeof(firstline),
- "650+AUTHDIR_NEWDESC=\r\n%s\r\n%s\r\n",
- action,
- msg ? msg : "");
- /* Escape the server descriptor properly */
- esclen = write_escaped_data(descriptor, strlen(descriptor), 1, &esc);
- totallen = strlen(firstline) + esclen + 1;
- buf = tor_malloc(totallen);
- strlcpy(buf, firstline, totallen);
- strlcpy(buf+strlen(firstline), esc, totallen);
- send_control1_event_string(EVENT_AUTHDIR_NEWDESCS, buf);
- tor_free(esc);
- tor_free(buf);
- return 0;
- }
- /** Choose a random authentication cookie and write it to disk.
- * Anybody who can read the cookie from disk will be considered
- * authorized to use the control connection. */
- int
- init_cookie_authentication(int enabled)
- {
- char fname[512];
- if (!enabled) {
- authentication_cookie_is_set = 0;
- return 0;
- }
- tor_snprintf(fname, sizeof(fname), "%s/control_auth_cookie",
- get_options()->DataDirectory);
- crypto_rand(authentication_cookie, AUTHENTICATION_COOKIE_LEN);
- authentication_cookie_is_set = 1;
- if (write_bytes_to_file(fname, authentication_cookie,
- AUTHENTICATION_COOKIE_LEN, 1)) {
- warn(LD_FS,"Error writing authentication cookie.");
- return -1;
- }
- return 0;
- }
|