MBO: Add support to ignore association disallowed set by AP
[mech_eap.git] / hostapd / hlr_auc_gw.c
index 3ef4365..2117d34 100644 (file)
@@ -1,6 +1,6 @@
 /*
  * HLR/AuC testing gateway for hostapd EAP-SIM/AKA database/authenticator
- * Copyright (c) 2005-2007, 2012-2013, Jouni Malinen <j@w1.fi>
+ * Copyright (c) 2005-2007, 2012-2016, Jouni Malinen <j@w1.fi>
  *
  * This software may be distributed under the terms of the BSD license.
  * See README for more details.
@@ -18,6 +18,9 @@
  * SIM-REQ-AUTH <IMSI> <max_chal>
  * SIM-RESP-AUTH <IMSI> Kc1:SRES1:RAND1 Kc2:SRES2:RAND2 [Kc3:SRES3:RAND3]
  * SIM-RESP-AUTH <IMSI> FAILURE
+ * GSM-AUTH-REQ <IMSI> RAND1:RAND2[:RAND3]
+ * GSM-AUTH-RESP <IMSI> Kc1:SRES1:Kc2:SRES2[:Kc3:SRES3]
+ * GSM-AUTH-RESP <IMSI> FAILURE
  *
  * EAP-AKA / UMTS query/response:
  * AKA-REQ-AUTH <IMSI>
@@ -62,6 +65,7 @@ static char *milenage_file = NULL;
 static int update_milenage = 0;
 static int sqn_changes = 0;
 static int ind_len = 5;
+static int stdout_debug = 1;
 
 /* GSM triplets */
 struct gsm_triplet {
@@ -83,6 +87,7 @@ struct milenage_parameters {
        u8 amf[2];
        u8 sqn[6];
        int set;
+       size_t res_len;
 };
 
 static struct milenage_parameters *milenage_db = NULL;
@@ -92,6 +97,7 @@ static struct milenage_parameters *milenage_db = NULL;
 #define EAP_AKA_RAND_LEN 16
 #define EAP_AKA_AUTN_LEN 16
 #define EAP_AKA_AUTS_LEN 14
+#define EAP_AKA_RES_MIN_LEN 4
 #define EAP_AKA_RES_MAX_LEN 16
 #define EAP_AKA_IK_LEN 16
 #define EAP_AKA_CK_LEN 16
@@ -120,7 +126,8 @@ static int db_table_create_milenage(sqlite3 *db)
                "  ki CHAR(32) NOT NULL,"
                "  opc CHAR(32) NOT NULL,"
                "  amf CHAR(4) NOT NULL,"
-               "  sqn CHAR(12) NOT NULL"
+               "  sqn CHAR(12) NOT NULL,"
+               "  res_len INTEGER"
                ");";
 
        printf("Adding database table for milenage information\n");
@@ -186,6 +193,10 @@ static int get_milenage_cb(void *ctx, int argc, char *argv[], char *col[])
                        printf("Invalid sqn value in database\n");
                        return -1;
                }
+
+               if (os_strcmp(col[i], "res_len") == 0 && argv[i]) {
+                       m->res_len = atoi(argv[i]);
+               }
        }
 
        return 0;
@@ -202,8 +213,7 @@ static struct milenage_parameters * db_get_milenage(const char *imsi_txt)
        os_snprintf(db_tmp_milenage.imsi, sizeof(db_tmp_milenage.imsi),
                    "%llu", imsi);
        os_snprintf(cmd, sizeof(cmd),
-                   "SELECT ki,opc,amf,sqn FROM milenage WHERE imsi=%llu;",
-                   imsi);
+                   "SELECT * FROM milenage WHERE imsi=%llu;", imsi);
        if (sqlite3_exec(sqlite_db, cmd, get_milenage_cb, &db_tmp_milenage,
                         NULL) != SQLITE_OK)
                return NULL;
@@ -218,6 +228,9 @@ static int db_update_milenage_sqn(struct milenage_parameters *m)
 {
        char cmd[128], val[13], *pos;
 
+       if (sqlite_db == NULL)
+               return 0;
+
        pos = val;
        pos += wpa_snprintf_hex(pos, sizeof(val), m->sqn, 6);
        *pos = '\0';
@@ -271,7 +284,7 @@ static int read_gsm_triplets(const char *fname)
 
        f = fopen(fname, "r");
        if (f == NULL) {
-               printf("Could not open GSM tripler data file '%s'\n", fname);
+               printf("Could not open GSM triplet data file '%s'\n", fname);
                return -1;
        }
 
@@ -299,66 +312,40 @@ static int read_gsm_triplets(const char *fname)
                }
 
                /* IMSI */
-               pos2 = strchr(pos, ':');
-               if (pos2 == NULL) {
-                       printf("%s:%d - Invalid IMSI (%s)\n",
-                              fname, line, pos);
-                       ret = -1;
-                       break;
-               }
-               *pos2 = '\0';
-               if (strlen(pos) >= sizeof(g->imsi)) {
-                       printf("%s:%d - Too long IMSI (%s)\n",
-                              fname, line, pos);
+               pos2 = NULL;
+               pos = str_token(buf, ":", &pos2);
+               if (!pos || os_strlen(pos) >= sizeof(g->imsi)) {
+                       printf("%s:%d - Invalid IMSI\n", fname, line);
                        ret = -1;
                        break;
                }
                os_strlcpy(g->imsi, pos, sizeof(g->imsi));
-               pos = pos2 + 1;
 
                /* Kc */
-               pos2 = strchr(pos, ':');
-               if (pos2 == NULL) {
-                       printf("%s:%d - Invalid Kc (%s)\n", fname, line, pos);
+               pos = str_token(buf, ":", &pos2);
+               if (!pos || os_strlen(pos) != 16 || hexstr2bin(pos, g->kc, 8)) {
+                       printf("%s:%d - Invalid Kc\n", fname, line);
                        ret = -1;
                        break;
                }
-               *pos2 = '\0';
-               if (strlen(pos) != 16 || hexstr2bin(pos, g->kc, 8)) {
-                       printf("%s:%d - Invalid Kc (%s)\n", fname, line, pos);
-                       ret = -1;
-                       break;
-               }
-               pos = pos2 + 1;
 
                /* SRES */
-               pos2 = strchr(pos, ':');
-               if (pos2 == NULL) {
-                       printf("%s:%d - Invalid SRES (%s)\n", fname, line,
-                              pos);
+               pos = str_token(buf, ":", &pos2);
+               if (!pos || os_strlen(pos) != 8 ||
+                   hexstr2bin(pos, g->sres, 4)) {
+                       printf("%s:%d - Invalid SRES\n", fname, line);
                        ret = -1;
                        break;
                }
-               *pos2 = '\0';
-               if (strlen(pos) != 8 || hexstr2bin(pos, g->sres, 4)) {
-                       printf("%s:%d - Invalid SRES (%s)\n", fname, line,
-                              pos);
-                       ret = -1;
-                       break;
-               }
-               pos = pos2 + 1;
 
                /* RAND */
-               pos2 = strchr(pos, ':');
-               if (pos2)
-                       *pos2 = '\0';
-               if (strlen(pos) != 32 || hexstr2bin(pos, g->_rand, 16)) {
-                       printf("%s:%d - Invalid RAND (%s)\n", fname, line,
-                              pos);
+               pos = str_token(buf, ":", &pos2);
+               if (!pos || os_strlen(pos) != 32 ||
+                   hexstr2bin(pos, g->_rand, 16)) {
+                       printf("%s:%d - Invalid RAND\n", fname, line);
                        ret = -1;
                        break;
                }
-               pos = pos2 + 1;
 
                g->next = gsm_db;
                gsm_db = g;
@@ -417,7 +404,7 @@ static int read_milenage(const char *fname)
        while (fgets(buf, sizeof(buf), f)) {
                line++;
 
-               /* Parse IMSI Ki OPc AMF SQN */
+               /* Parse IMSI Ki OPc AMF SQN [RES_len] */
                buf[sizeof(buf) - 1] = '\0';
                if (buf[0] == '#')
                        continue;
@@ -437,78 +424,62 @@ static int read_milenage(const char *fname)
                }
 
                /* IMSI */
-               pos2 = strchr(pos, ' ');
-               if (pos2 == NULL) {
-                       printf("%s:%d - Invalid IMSI (%s)\n",
-                              fname, line, pos);
-                       ret = -1;
-                       break;
-               }
-               *pos2 = '\0';
-               if (strlen(pos) >= sizeof(m->imsi)) {
-                       printf("%s:%d - Too long IMSI (%s)\n",
-                              fname, line, pos);
+               pos2 = NULL;
+               pos = str_token(buf, " ", &pos2);
+               if (!pos || os_strlen(pos) >= sizeof(m->imsi)) {
+                       printf("%s:%d - Invalid IMSI\n", fname, line);
                        ret = -1;
                        break;
                }
                os_strlcpy(m->imsi, pos, sizeof(m->imsi));
-               pos = pos2 + 1;
 
                /* Ki */
-               pos2 = strchr(pos, ' ');
-               if (pos2 == NULL) {
-                       printf("%s:%d - Invalid Ki (%s)\n", fname, line, pos);
-                       ret = -1;
-                       break;
-               }
-               *pos2 = '\0';
-               if (strlen(pos) != 32 || hexstr2bin(pos, m->ki, 16)) {
-                       printf("%s:%d - Invalid Ki (%s)\n", fname, line, pos);
+               pos = str_token(buf, " ", &pos2);
+               if (!pos || os_strlen(pos) != 32 ||
+                   hexstr2bin(pos, m->ki, 16)) {
+                       printf("%s:%d - Invalid Ki\n", fname, line);
                        ret = -1;
                        break;
                }
-               pos = pos2 + 1;
 
                /* OPc */
-               pos2 = strchr(pos, ' ');
-               if (pos2 == NULL) {
-                       printf("%s:%d - Invalid OPc (%s)\n", fname, line, pos);
-                       ret = -1;
-                       break;
-               }
-               *pos2 = '\0';
-               if (strlen(pos) != 32 || hexstr2bin(pos, m->opc, 16)) {
-                       printf("%s:%d - Invalid OPc (%s)\n", fname, line, pos);
+               pos = str_token(buf, " ", &pos2);
+               if (!pos || os_strlen(pos) != 32 ||
+                   hexstr2bin(pos, m->opc, 16)) {
+                       printf("%s:%d - Invalid OPc\n", fname, line);
                        ret = -1;
                        break;
                }
-               pos = pos2 + 1;
 
                /* AMF */
-               pos2 = strchr(pos, ' ');
-               if (pos2 == NULL) {
-                       printf("%s:%d - Invalid AMF (%s)\n", fname, line, pos);
+               pos = str_token(buf, " ", &pos2);
+               if (!pos || os_strlen(pos) != 4 || hexstr2bin(pos, m->amf, 2)) {
+                       printf("%s:%d - Invalid AMF\n", fname, line);
                        ret = -1;
                        break;
                }
-               *pos2 = '\0';
-               if (strlen(pos) != 4 || hexstr2bin(pos, m->amf, 2)) {
-                       printf("%s:%d - Invalid AMF (%s)\n", fname, line, pos);
-                       ret = -1;
-                       break;
-               }
-               pos = pos2 + 1;
 
                /* SQN */
-               pos2 = strchr(pos, ' ');
-               if (pos2)
-                       *pos2 = '\0';
-               if (strlen(pos) != 12 || hexstr2bin(pos, m->sqn, 6)) {
-                       printf("%s:%d - Invalid SEQ (%s)\n", fname, line, pos);
+               pos = str_token(buf, " ", &pos2);
+               if (!pos || os_strlen(pos) != 12 ||
+                   hexstr2bin(pos, m->sqn, 6)) {
+                       printf("%s:%d - Invalid SEQ\n", fname, line);
                        ret = -1;
                        break;
                }
-               pos = pos2 + 1;
+
+               pos = str_token(buf, " ", &pos2);
+               if (pos) {
+                       m->res_len = atoi(pos);
+                       if (m->res_len &&
+                           (m->res_len < EAP_AKA_RES_MIN_LEN ||
+                            m->res_len > EAP_AKA_RES_MAX_LEN)) {
+                               printf("%s:%d - Invalid RES_len\n",
+                                      fname, line);
+                               ret = -1;
+                               break;
+                       }
+               }
 
                m->next = milenage_db;
                milenage_db = m;
@@ -525,7 +496,7 @@ static int read_milenage(const char *fname)
 static void update_milenage_file(const char *fname)
 {
        FILE *f, *f2;
-       char buf[500], *pos;
+       char name[500], buf[500], *pos;
        char *end = buf + sizeof(buf);
        struct milenage_parameters *m;
        size_t imsi_len;
@@ -536,10 +507,10 @@ static void update_milenage_file(const char *fname)
                return;
        }
 
-       snprintf(buf, sizeof(buf), "%s.new", fname);
-       f2 = fopen(buf, "w");
+       snprintf(name, sizeof(name), "%s.new", fname);
+       f2 = fopen(name, "w");
        if (f2 == NULL) {
-               printf("Could not write Milenage data file '%s'\n", buf);
+               printf("Could not write Milenage data file '%s'\n", name);
                fclose(f);
                return;
        }
@@ -581,14 +552,14 @@ static void update_milenage_file(const char *fname)
        fclose(f2);
        fclose(f);
 
-       snprintf(buf, sizeof(buf), "%s.bak", fname);
-       if (rename(fname, buf) < 0) {
+       snprintf(name, sizeof(name), "%s.bak", fname);
+       if (rename(fname, name) < 0) {
                perror("rename");
                return;
        }
 
-       snprintf(buf, sizeof(buf), "%s.new", fname);
-       if (rename(buf, fname) < 0) {
+       snprintf(name, sizeof(name), "%s.new", fname);
+       if (rename(name, fname) < 0) {
                perror("rename");
                return;
        }
@@ -615,31 +586,30 @@ static struct milenage_parameters * get_milenage(const char *imsi)
 }
 
 
-static void sim_req_auth(int s, struct sockaddr_un *from, socklen_t fromlen,
-                        char *imsi)
+static int sim_req_auth(char *imsi, char *resp, size_t resp_len)
 {
        int count, max_chal, ret;
        char *pos;
-       char reply[1000], *rpos, *rend;
+       char *rpos, *rend;
        struct milenage_parameters *m;
        struct gsm_triplet *g;
 
-       reply[0] = '\0';
+       resp[0] = '\0';
 
        pos = strchr(imsi, ' ');
        if (pos) {
                *pos++ = '\0';
                max_chal = atoi(pos);
-               if (max_chal < 1 || max_chal < EAP_SIM_MAX_CHAL)
+               if (max_chal < 1 || max_chal > EAP_SIM_MAX_CHAL)
                        max_chal = EAP_SIM_MAX_CHAL;
        } else
                max_chal = EAP_SIM_MAX_CHAL;
 
-       rend = &reply[sizeof(reply)];
-       rpos = reply;
+       rend = resp + resp_len;
+       rpos = resp;
        ret = snprintf(rpos, rend - rpos, "SIM-RESP-AUTH %s", imsi);
        if (ret < 0 || ret >= rend - rpos)
-               return;
+               return -1;
        rpos += ret;
 
        m = get_milenage(imsi);
@@ -647,7 +617,7 @@ static void sim_req_auth(int s, struct sockaddr_un *from, socklen_t fromlen,
                u8 _rand[16], sres[4], kc[8];
                for (count = 0; count < max_chal; count++) {
                        if (random_get_bytes(_rand, 16) < 0)
-                               return;
+                               return -1;
                        gsm_milenage(m->opc, m->ki, _rand, sres, kc);
                        *rpos++ = ' ';
                        rpos += wpa_snprintf_hex(rpos, rend - rpos, kc, 8);
@@ -657,7 +627,7 @@ static void sim_req_auth(int s, struct sockaddr_un *from, socklen_t fromlen,
                        rpos += wpa_snprintf_hex(rpos, rend - rpos, _rand, 16);
                }
                *rpos = '\0';
-               goto send;
+               return 0;
        }
 
        count = 0;
@@ -681,15 +651,61 @@ static void sim_req_auth(int s, struct sockaddr_un *from, socklen_t fromlen,
                printf("No GSM triplets found for %s\n", imsi);
                ret = snprintf(rpos, rend - rpos, " FAILURE");
                if (ret < 0 || ret >= rend - rpos)
-                       return;
+                       return -1;
                rpos += ret;
        }
 
-send:
-       printf("Send: %s\n", reply);
-       if (sendto(s, reply, rpos - reply, 0,
-                  (struct sockaddr *) from, fromlen) < 0)
-               perror("send");
+       return 0;
+}
+
+
+static int gsm_auth_req(char *imsi, char *resp, size_t resp_len)
+{
+       int count, ret;
+       char *pos, *rpos, *rend;
+       struct milenage_parameters *m;
+
+       resp[0] = '\0';
+
+       pos = os_strchr(imsi, ' ');
+       if (!pos)
+               return -1;
+       *pos++ = '\0';
+
+       rend = resp + resp_len;
+       rpos = resp;
+       ret = os_snprintf(rpos, rend - rpos, "GSM-AUTH-RESP %s", imsi);
+       if (os_snprintf_error(rend - rpos, ret))
+               return -1;
+       rpos += ret;
+
+       m = get_milenage(imsi);
+       if (m) {
+               u8 _rand[16], sres[4], kc[8];
+               for (count = 0; count < EAP_SIM_MAX_CHAL; count++) {
+                       if (hexstr2bin(pos, _rand, 16) != 0)
+                               return -1;
+                       gsm_milenage(m->opc, m->ki, _rand, sres, kc);
+                       *rpos++ = count == 0 ? ' ' : ':';
+                       rpos += wpa_snprintf_hex(rpos, rend - rpos, kc, 8);
+                       *rpos++ = ':';
+                       rpos += wpa_snprintf_hex(rpos, rend - rpos, sres, 4);
+                       pos += 16 * 2;
+                       if (*pos != ':')
+                               break;
+                       pos++;
+               }
+               *rpos = '\0';
+               return 0;
+       }
+
+       printf("No GSM triplets found for %s\n", imsi);
+       ret = os_snprintf(rpos, rend - rpos, " FAILURE");
+       if (os_snprintf_error(rend - rpos, ret))
+               return -1;
+       rpos += ret;
+
+       return 0;
 }
 
 
@@ -715,11 +731,10 @@ static void inc_sqn(u8 *sqn)
 }
 
 
-static void aka_req_auth(int s, struct sockaddr_un *from, socklen_t fromlen,
-                        char *imsi)
+static int aka_req_auth(char *imsi, char *resp, size_t resp_len)
 {
        /* AKA-RESP-AUTH <IMSI> <RAND> <AUTN> <IK> <CK> <RES> */
-       char reply[1000], *pos, *end;
+       char *pos, *end;
        u8 _rand[EAP_AKA_RAND_LEN];
        u8 autn[EAP_AKA_AUTN_LEN];
        u8 ik[EAP_AKA_IK_LEN];
@@ -733,18 +748,24 @@ static void aka_req_auth(int s, struct sockaddr_un *from, socklen_t fromlen,
        m = get_milenage(imsi);
        if (m) {
                if (random_get_bytes(_rand, EAP_AKA_RAND_LEN) < 0)
-                       return;
+                       return -1;
                res_len = EAP_AKA_RES_MAX_LEN;
                inc_sqn(m->sqn);
 #ifdef CONFIG_SQLITE
                db_update_milenage_sqn(m);
 #endif /* CONFIG_SQLITE */
                sqn_changes = 1;
-               printf("AKA: Milenage with SQN=%02x%02x%02x%02x%02x%02x\n",
-                      m->sqn[0], m->sqn[1], m->sqn[2],
-                      m->sqn[3], m->sqn[4], m->sqn[5]);
+               if (stdout_debug) {
+                       printf("AKA: Milenage with SQN=%02x%02x%02x%02x%02x%02x\n",
+                              m->sqn[0], m->sqn[1], m->sqn[2],
+                              m->sqn[3], m->sqn[4], m->sqn[5]);
+               }
                milenage_generate(m->opc, m->amf, m->ki, m->sqn, _rand,
                                  autn, ik, ck, res, &res_len);
+               if (m->res_len >= EAP_AKA_RES_MIN_LEN &&
+                   m->res_len <= EAP_AKA_RES_MAX_LEN &&
+                   m->res_len < res_len)
+                       res_len = m->res_len;
        } else {
                printf("Unknown IMSI: %s\n", imsi);
 #ifdef AKA_USE_FIXED_TEST_VALUES
@@ -760,18 +781,18 @@ static void aka_req_auth(int s, struct sockaddr_un *from, socklen_t fromlen,
 #endif /* AKA_USE_FIXED_TEST_VALUES */
        }
 
-       pos = reply;
-       end = &reply[sizeof(reply)];
+       pos = resp;
+       end = resp + resp_len;
        ret = snprintf(pos, end - pos, "AKA-RESP-AUTH %s ", imsi);
        if (ret < 0 || ret >= end - pos)
-               return;
+               return -1;
        pos += ret;
        if (failed) {
                ret = snprintf(pos, end - pos, "FAILURE");
                if (ret < 0 || ret >= end - pos)
-                       return;
+                       return -1;
                pos += ret;
-               goto done;
+               return 0;
        }
        pos += wpa_snprintf_hex(pos, end - pos, _rand, EAP_AKA_RAND_LEN);
        *pos++ = ' ';
@@ -783,65 +804,87 @@ static void aka_req_auth(int s, struct sockaddr_un *from, socklen_t fromlen,
        *pos++ = ' ';
        pos += wpa_snprintf_hex(pos, end - pos, res, res_len);
 
-done:
-       printf("Send: %s\n", reply);
-
-       if (sendto(s, reply, pos - reply, 0, (struct sockaddr *) from,
-                  fromlen) < 0)
-               perror("send");
+       return 0;
 }
 
 
-static void aka_auts(int s, struct sockaddr_un *from, socklen_t fromlen,
-                    char *imsi)
+static int aka_auts(char *imsi, char *resp, size_t resp_len)
 {
        char *auts, *__rand;
        u8 _auts[EAP_AKA_AUTS_LEN], _rand[EAP_AKA_RAND_LEN], sqn[6];
        struct milenage_parameters *m;
 
+       resp[0] = '\0';
+
        /* AKA-AUTS <IMSI> <AUTS> <RAND> */
 
        auts = strchr(imsi, ' ');
        if (auts == NULL)
-               return;
+               return -1;
        *auts++ = '\0';
 
        __rand = strchr(auts, ' ');
        if (__rand == NULL)
-               return;
+               return -1;
        *__rand++ = '\0';
 
-       printf("AKA-AUTS: IMSI=%s AUTS=%s RAND=%s\n", imsi, auts, __rand);
+       if (stdout_debug) {
+               printf("AKA-AUTS: IMSI=%s AUTS=%s RAND=%s\n",
+                      imsi, auts, __rand);
+       }
        if (hexstr2bin(auts, _auts, EAP_AKA_AUTS_LEN) ||
            hexstr2bin(__rand, _rand, EAP_AKA_RAND_LEN)) {
                printf("Could not parse AUTS/RAND\n");
-               return;
+               return -1;
        }
 
        m = get_milenage(imsi);
        if (m == NULL) {
                printf("Unknown IMSI: %s\n", imsi);
-               return;
+               return -1;
        }
 
        if (milenage_auts(m->opc, m->ki, _rand, _auts, sqn)) {
                printf("AKA-AUTS: Incorrect MAC-S\n");
        } else {
                memcpy(m->sqn, sqn, 6);
-               printf("AKA-AUTS: Re-synchronized: "
-                      "SQN=%02x%02x%02x%02x%02x%02x\n",
-                      sqn[0], sqn[1], sqn[2], sqn[3], sqn[4], sqn[5]);
+               if (stdout_debug) {
+                       printf("AKA-AUTS: Re-synchronized: "
+                              "SQN=%02x%02x%02x%02x%02x%02x\n",
+                              sqn[0], sqn[1], sqn[2], sqn[3], sqn[4], sqn[5]);
+               }
 #ifdef CONFIG_SQLITE
                db_update_milenage_sqn(m);
 #endif /* CONFIG_SQLITE */
                sqn_changes = 1;
        }
+
+       return 0;
+}
+
+
+static int process_cmd(char *cmd, char *resp, size_t resp_len)
+{
+       if (os_strncmp(cmd, "SIM-REQ-AUTH ", 13) == 0)
+               return sim_req_auth(cmd + 13, resp, resp_len);
+
+       if (os_strncmp(cmd, "GSM-AUTH-REQ ", 13) == 0)
+               return gsm_auth_req(cmd + 13, resp, resp_len);
+
+       if (os_strncmp(cmd, "AKA-REQ-AUTH ", 13) == 0)
+               return aka_req_auth(cmd + 13, resp, resp_len);
+
+       if (os_strncmp(cmd, "AKA-AUTS ", 9) == 0)
+               return aka_auts(cmd + 9, resp, resp_len);
+
+       printf("Unknown request: %s\n", cmd);
+       return -1;
 }
 
 
 static int process(int s)
 {
-       char buf[1000];
+       char buf[1000], resp[1000];
        struct sockaddr_un from;
        socklen_t fromlen;
        ssize_t res;
@@ -863,14 +906,21 @@ static int process(int s)
 
        printf("Received: %s\n", buf);
 
-       if (strncmp(buf, "SIM-REQ-AUTH ", 13) == 0)
-               sim_req_auth(s, &from, fromlen, buf + 13);
-       else if (strncmp(buf, "AKA-REQ-AUTH ", 13) == 0)
-               aka_req_auth(s, &from, fromlen, buf + 13);
-       else if (strncmp(buf, "AKA-AUTS ", 9) == 0)
-               aka_auts(s, &from, fromlen, buf + 9);
-       else
-               printf("Unknown request: %s\n", buf);
+       if (process_cmd(buf, resp, sizeof(resp)) < 0) {
+               printf("Failed to process request\n");
+               return -1;
+       }
+
+       if (resp[0] == '\0') {
+               printf("No response\n");
+               return 0;
+       }
+
+       printf("Send: %s\n", resp);
+
+       if (sendto(s, resp, os_strlen(resp), 0, (struct sockaddr *) &from,
+                  fromlen) < 0)
+               perror("send");
 
        return 0;
 }
@@ -898,8 +948,10 @@ static void cleanup(void)
                os_free(prev);
        }
 
-       close(serv_sock);
-       unlink(socket_path);
+       if (serv_sock >= 0)
+               close(serv_sock);
+       if (socket_path)
+               unlink(socket_path);
 
 #ifdef CONFIG_SQLITE
        if (sqlite_db) {
@@ -921,12 +973,12 @@ static void usage(void)
 {
        printf("HLR/AuC testing gateway for hostapd EAP-SIM/AKA "
               "database/authenticator\n"
-              "Copyright (c) 2005-2007, 2012-2013, Jouni Malinen <j@w1.fi>\n"
+              "Copyright (c) 2005-2016, Jouni Malinen <j@w1.fi>\n"
               "\n"
               "usage:\n"
               "hlr_auc_gw [-hu] [-s<socket path>] [-g<triplet file>] "
               "[-m<milenage file>] \\\n"
-              "        [-D<DB file>] [-i<IND len in bits>]\n"
+              "        [-D<DB file>] [-i<IND len in bits>] [command]\n"
               "\n"
               "options:\n"
               "  -h = show this usage help\n"
@@ -936,7 +988,15 @@ static void usage(void)
               "  -g<triplet file> = path for GSM authentication triplets\n"
               "  -m<milenage file> = path for Milenage keys\n"
               "  -D<DB file> = path to SQLite database\n"
-              "  -i<IND len in bits> = IND length for SQN (default: 5)\n",
+              "  -i<IND len in bits> = IND length for SQN (default: 5)\n"
+              "\n"
+              "If the optional command argument, like "
+              "\"AKA-REQ-AUTH <IMSI>\" is used, a single\n"
+              "command is processed with response sent to stdout. Otherwise, "
+              "hlr_auc_gw opens\n"
+              "a control interface and processes commands sent through it "
+              "(e.g., by EAP server\n"
+              "in hostapd).\n",
               default_socket_path);
 }
 
@@ -946,6 +1006,7 @@ int main(int argc, char *argv[])
        int c;
        char *gsm_triplet_file = NULL;
        char *sqlite_db_file = NULL;
+       int ret = 0;
 
        if (os_program_init())
                return -1;
@@ -1009,18 +1070,31 @@ int main(int argc, char *argv[])
        if (milenage_file && read_milenage(milenage_file) < 0)
                return -1;
 
-       serv_sock = open_socket(socket_path);
-       if (serv_sock < 0)
-               return -1;
+       if (optind == argc) {
+               serv_sock = open_socket(socket_path);
+               if (serv_sock < 0)
+                       return -1;
 
-       printf("Listening for requests on %s\n", socket_path);
+               printf("Listening for requests on %s\n", socket_path);
 
-       atexit(cleanup);
-       signal(SIGTERM, handle_term);
-       signal(SIGINT, handle_term);
+               atexit(cleanup);
+               signal(SIGTERM, handle_term);
+               signal(SIGINT, handle_term);
 
-       for (;;)
-               process(serv_sock);
+               for (;;)
+                       process(serv_sock);
+       } else {
+               char buf[1000];
+               socket_path = NULL;
+               stdout_debug = 0;
+               if (process_cmd(argv[optind], buf, sizeof(buf)) < 0) {
+                       printf("FAIL\n");
+                       ret = -1;
+               } else {
+                       printf("%s\n", buf);
+               }
+               cleanup();
+       }
 
 #ifdef CONFIG_SQLITE
        if (sqlite_db) {
@@ -1031,5 +1105,5 @@ int main(int argc, char *argv[])
 
        os_program_deinit();
 
-       return 0;
+       return ret;
 }