diff --git a/src/common.h b/src/common.h index 4f51175..3dd22b3 100644 --- a/src/common.h +++ b/src/common.h @@ -104,7 +104,7 @@ struct query { }; enum connection { - CONN_RAW_UDP, + CONN_RAW_UDP = 0, CONN_DNS_NULL, CONN_MAX }; diff --git a/src/user.c b/src/user.c index 61a16e8..6391573 100644 --- a/src/user.c +++ b/src/user.c @@ -199,7 +199,7 @@ user_set_conn_type(int userid, enum connection c) if (userid < 0 || userid >= usercount) return; - if (c < 0 || c >= CONN_MAX) + if (c < CONN_RAW_UDP || c >= CONN_MAX) return; users[userid].conn = c;