diff --git a/src/or/control.c b/src/or/control.c
index 43970fc0e7..b7bd35b6c1 100644
--- a/src/or/control.c
+++ b/src/or/control.c
@@ -169,8 +169,8 @@ 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_setcircuitpurpose(connection_t *conn, uint32_t len,
- const char *body);
+static int handle_control_setpurpose(connection_t *conn, int for_circuits,
+ 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,
@@ -1576,19 +1576,22 @@ handle_control_getinfo(connection_t *conn, uint32_t len, const char *body)
return 0;
}
-/** If string contains a recognized circuit purpose,
+/** If string contains a recognized purpose (for
+ * circuits if for_circuits is 1, else for routers),
* possibly prefaced with the string "purpose=", then assign it
* and return 0. Otherwise return -1. */
static int
-get_purpose(char *string, uint8_t *purpose)
+get_purpose(char *string, int for_circuits, uint8_t *purpose)
{
if (!strcmpstart(string, "purpose="))
string += strlen("purpose=");
if (!strcmp(string, "general"))
- *purpose = CIRCUIT_PURPOSE_C_GENERAL;
+ *purpose = for_circuits ? CIRCUIT_PURPOSE_C_GENERAL :
+ ROUTER_PURPOSE_GENERAL;
else if (!strcmp(string, "controller"))
- *purpose = CIRCUIT_PURPOSE_CONTROLLER;
+ *purpose = for_circuits ? CIRCUIT_PURPOSE_CONTROLLER :
+ ROUTER_PURPOSE_GENERAL;
else { /* not a recognized purpose */
return -1;
}
@@ -1656,7 +1659,7 @@ handle_control_extendcircuit(connection_t *conn, uint32_t len,
goto done;
}
if (zero_circ && smartlist_len(args)>2) {
- if (get_purpose(smartlist_get(args,2), &intended_purpose) < 0) {
+ if (get_purpose(smartlist_get(args,2), 1, &intended_purpose) < 0) {
connection_printf_to_buf(conn, "552 Unknown purpose \"%s\"\r\n",
(char *)smartlist_get(args,2));
goto done;
@@ -1739,36 +1742,50 @@ handle_control_extendcircuit(connection_t *conn, uint32_t len,
return 0;
}
-/** Called when we get a SETCIRCUITPURPOSE message. If we can find
- * the circuit and it's a valid purpose, change it. */
+/** Called when we get a SETCIRCUITPURPOSE (if for_circuits
+ * is 1) or SETROUTERPURPOSE message. If we can find
+ * the circuit/router and it's a valid purpose, change it. */
static int
-handle_control_setcircuitpurpose(connection_t *conn, uint32_t len,
- const char *body)
+handle_control_setpurpose(connection_t *conn, int for_circuits,
+ uint32_t len, const char *body)
{
- circuit_t *circ;
+ circuit_t *circ = NULL;
+ routerinfo_t *ri = NULL;
uint8_t new_purpose;
smartlist_t *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 SETCIRCUITPURPOSE\r\n");
+ "512 Missing argument to SET%sPURPOSE\r\n",
+ for_circuits ? "CIRCUIT" : "ROUTER");
goto done;
}
- if (!(circ = get_circ(smartlist_get(args,0)))) {
- connection_printf_to_buf(conn, "552 Unknown circuit \"%s\"\r\n",
- (char*)smartlist_get(args, 0));
- goto done;
+ if (for_circuits) {
+ if (!(circ = get_circ(smartlist_get(args,0)))) {
+ connection_printf_to_buf(conn, "552 Unknown circuit \"%s\"\r\n",
+ (char*)smartlist_get(args, 0));
+ goto done;
+ }
+ } else {
+ if (!(ri = router_get_by_nickname(smartlist_get(args,0), 0))) {
+ connection_printf_to_buf(conn, "552 Unknown router \"%s\"\r\n",
+ (char*)smartlist_get(args, 0));
+ goto done;
+ }
}
- if (get_purpose(smartlist_get(args,1), &new_purpose) < 0) {
+ if (get_purpose(smartlist_get(args,1), for_circuits, &new_purpose) < 0) {
connection_printf_to_buf(conn, "552 Unknown purpose \"%s\"\r\n",
(char *)smartlist_get(args,1));
goto done;
}
- circ->purpose = new_purpose;
+ if (for_circuits)
+ circ->purpose = new_purpose;
+ else
+ ri->purpose = new_purpose;
connection_write_str_to_buf("250 OK\r\n", conn);
done:
@@ -1891,16 +1908,32 @@ handle_control_postdescriptor(connection_t *conn, uint32_t len,
char *desc;
int v0 = STATE_IS_V0(conn->state);
const char *msg=NULL;
+ uint8_t purpose = ROUTER_PURPOSE_GENERAL;
if (v0)
desc = (char*)body;
else {
- const char *cp = memchr(body, '\n', len);
+ char *cp = memchr(body, '\n', len);
+ smartlist_t *args = smartlist_create();
tor_assert(cp);
+ *cp++ = '\0';
+ smartlist_split_string(args, body, " ",
+ SPLIT_SKIP_SPACE|SPLIT_IGNORE_BLANK, 0);
+ if (smartlist_len(args)) {
+ if (get_purpose(smartlist_get(args,0), 0, &purpose) < 0) {
+ connection_printf_to_buf(conn, "552 Unknown purpose \"%s\"\r\n",
+ (char *)smartlist_get(args,0));
+ SMARTLIST_FOREACH(args, char *, cp, tor_free(cp));
+ smartlist_free(args);
+ return 0;
+ }
+ }
+ SMARTLIST_FOREACH(args, char *, cp, tor_free(cp));
+ smartlist_free(args);
read_escaped_data(cp, len-(cp-body), 1, &desc);
}
- switch (router_load_single_router(desc, &msg)) {
+ switch (router_load_single_router(desc, purpose, &msg)) {
case -1:
if (!msg) msg = "Could not parse descriptor";
if (v0)
@@ -2296,7 +2329,10 @@ connection_control_process_inbuf_v1(connection_t *conn)
if (handle_control_extendcircuit(conn, data_len, args))
return -1;
} else if (!strcasecmp(conn->incoming_cmd, "SETCIRCUITPURPOSE")) {
- if (handle_control_setcircuitpurpose(conn, data_len, args))
+ if (handle_control_setpurpose(conn, 1, data_len, args))
+ return -1;
+ } else if (!strcasecmp(conn->incoming_cmd, "SETROUTERPURPOSE")) {
+ if (handle_control_setpurpose(conn, 0, data_len, args))
return -1;
} else if (!strcasecmp(conn->incoming_cmd, "ATTACHSTREAM")) {
if (handle_control_attachstream(conn, data_len, args))
diff --git a/src/or/or.h b/src/or/or.h
index c188f94a86..d51b566a31 100644
--- a/src/or/or.h
+++ b/src/or/or.h
@@ -817,6 +817,12 @@ typedef struct {
unsigned int is_stable:1; /** Do we think this is a stable OR? */
unsigned int is_possible_guard:1; /**< Do we think this is an OK guard? */
+/** Tor can use this desc for circuit-building. */
+#define ROUTER_PURPOSE_GENERAL 0
+/** Tor should avoid using this desc for circuit-building. */
+#define ROUTER_PURPOSE_CONTROLLER 1
+ uint8_t purpose; /** Should Tor use this desc for circuit-building? */
+
/* The below items are used only by authdirservers for
* reachability testing. */
/** When was the last time we could reach this OR? */
@@ -2322,7 +2328,8 @@ void routerlist_remove_old_routers(void);
void networkstatus_list_clean(time_t now);
int router_add_to_routerlist(routerinfo_t *router, const char **msg,
int from_cache, int from_fetch);
-int router_load_single_router(const char *s, const char **msg);
+int router_load_single_router(const char *s, uint8_t purpose,
+ const char **msg);
void router_load_routers_from_string(const char *s, int from_cache,
smartlist_t *requested_fingerprints);
typedef enum {
diff --git a/src/or/routerlist.c b/src/or/routerlist.c
index fe6febf632..004b97b3ef 100644
--- a/src/or/routerlist.c
+++ b/src/or/routerlist.c
@@ -647,7 +647,7 @@ router_nickname_is_in_list(routerinfo_t *router, const char *list)
}
/** Add every router from our routerlist that is currently running to
- * sl.
+ * sl, so that we can pick a node for a circuit.
*/
static void
router_add_running_routers_to_smartlist(smartlist_t *sl, int allow_unverified,
@@ -660,6 +660,7 @@ router_add_running_routers_to_smartlist(smartlist_t *sl, int allow_unverified,
SMARTLIST_FOREACH(routerlist->routers, routerinfo_t *, router,
{
if (router->is_running &&
+ router->purpose == ROUTER_PURPOSE_GENERAL &&
(router->is_verified ||
(allow_unverified &&
!router_is_unreliable(router, need_uptime,
@@ -1817,7 +1818,7 @@ routerlist_remove_old_routers(void)
* This is used only by the controller.
*/
int
-router_load_single_router(const char *s, const char **msg)
+router_load_single_router(const char *s, uint8_t purpose, const char **msg)
{
routerinfo_t *ri;
int r;
@@ -1830,6 +1831,7 @@ router_load_single_router(const char *s, const char **msg)
*msg = "Couldn't parse router descriptor.";
return -1;
}
+ ri->purpose = purpose;
if (router_is_me(ri)) {
log_warn(LD_DIR, "Router's identity key matches mine; dropping.");
*msg = "Router's identity key matches mine.";