Author: avos
Date: Thu Oct 22 17:08:12 2015
New Revision: 289762
URL: https://svnweb.freebsd.org/changeset/base/289762

Log:
  Unbreak regression testing tools for net80211 crypto cipher modules
  
  Approved by:  adrian (mentor)
  Differential Revision:        https://reviews.freebsd.org/D3441

Modified:
  head/tools/regression/net80211/ccmp/test_ccmp.c
  head/tools/regression/net80211/tkip/test_tkip.c
  head/tools/regression/net80211/wep/test_wep.c

Modified: head/tools/regression/net80211/ccmp/test_ccmp.c
==============================================================================
--- head/tools/regression/net80211/ccmp/test_ccmp.c     Thu Oct 22 16:46:30 
2015        (r289761)
+++ head/tools/regression/net80211/ccmp/test_ccmp.c     Thu Oct 22 17:08:12 
2015        (r289762)
@@ -53,6 +53,7 @@
 #include <sys/socket.h>
 
 #include <net/if.h>
+#include <net/if_var.h>
 #include <net/if_media.h>
 
 #include <net80211/ieee80211_var.h>
@@ -591,32 +592,32 @@ printtest(const struct ciphertest *t)
 }
 
 static int
-runtest(struct ieee80211com *ic, struct ciphertest *t)
+runtest(struct ieee80211vap *vap, struct ciphertest *t)
 {
-       struct ieee80211_key key;
+       struct ieee80211_key *key = &vap->iv_nw_keys[t->keyix];
        struct mbuf *m = NULL;
        const struct ieee80211_cipher *cip;
-       u_int8_t mac[IEEE80211_ADDR_LEN];
+       int hdrlen;
 
        printf("%s: ", t->name);
 
        /*
         * Setup key.
         */
-       memset(&key, 0, sizeof(key));
-       key.wk_flags = IEEE80211_KEY_XMIT | IEEE80211_KEY_RECV;
-       key.wk_cipher = &ieee80211_cipher_none;
-       if (!ieee80211_crypto_newkey(ic, t->cipher,
-           IEEE80211_KEY_XMIT | IEEE80211_KEY_RECV, &key)) {
+       memset(key, 0, sizeof(*key));
+       key->wk_flags = IEEE80211_KEY_XMIT | IEEE80211_KEY_RECV;
+       key->wk_cipher = &ieee80211_cipher_none;
+       if (!ieee80211_crypto_newkey(vap, t->cipher,
+           IEEE80211_KEY_XMIT | IEEE80211_KEY_RECV, key)) {
                printf("FAIL: ieee80211_crypto_newkey failed\n");
                goto bad;
        }
 
-       memcpy(key.wk_key, t->key, t->key_len);
-       key.wk_keylen = t->key_len;
-       key.wk_keyrsc = 0;
-       key.wk_keytsc = t->pn-1;        /* PN-1 since we do encap */
-       if (!ieee80211_crypto_setkey(ic, &key, mac)) {
+       memcpy(key->wk_key, t->key, t->key_len);
+       key->wk_keylen = t->key_len;
+       memset(key->wk_keyrsc, 0, sizeof(key->wk_keyrsc));
+       key->wk_keytsc = t->pn-1;       /* PN-1 since we do encap */
+       if (!ieee80211_crypto_setkey(vap, key)) {
                printf("FAIL: ieee80211_crypto_setkey failed\n");
                goto bad;
        }
@@ -624,17 +625,18 @@ runtest(struct ieee80211com *ic, struct 
        /*
         * Craft frame from plaintext data.
         */
-       cip = key.wk_cipher;
+       cip = key->wk_cipher;
        m = m_getcl(M_NOWAIT, MT_HEADER, M_PKTHDR);
        m->m_data += cip->ic_header;
        memcpy(mtod(m, void *), t->plaintext, t->plaintext_len);
        m->m_len = t->plaintext_len;
        m->m_pkthdr.len = m->m_len;
+       hdrlen = ieee80211_anyhdrsize(mtod(m, void *));
 
        /*
         * Encrypt frame w/ MIC.
         */
-       if (!cip->ic_encap(&key, m, t->keyix<<6)) {
+       if (!cip->ic_encap(key, m)) {
                printtest(t);
                printf("FAIL: ccmp encap failed\n");
                goto bad;
@@ -660,7 +662,7 @@ runtest(struct ieee80211com *ic, struct 
        /*
         * Decrypt frame; strip MIC.
         */
-       if (!cip->ic_decap(&key, m)) {
+       if (!cip->ic_decap(key, m, hdrlen)) {
                printf("FAIL: ccmp decap failed\n");
                printtest(t);
                cmpfail(mtod(m, const void *), m->m_len,
@@ -680,17 +682,17 @@ runtest(struct ieee80211com *ic, struct 
                printf("FAIL: decap botch; data does not compare\n");
                printtest(t);
                cmpfail(mtod(m, const void *), m->m_pkthdr.len,
-                       t->plaintext, t_plaintext_len);
+                       t->plaintext, t->plaintext_len);
                goto bad;
        }
        m_freem(m);
-       ieee80211_crypto_delkey(ic, &key);
+       ieee80211_crypto_delkey(vap, key);
        printf("PASS\n");
        return 1;
 bad:
        if (m != NULL)
                m_freem(m);
-       ieee80211_crypto_delkey(ic, &key);
+       ieee80211_crypto_delkey(vap, key);
        return 0;
 }
 
@@ -704,26 +706,38 @@ static    int debug = 0;
 static int
 init_crypto_ccmp_test(void)
 {
-#define        N(a)    (sizeof(a)/sizeof(a[0]))
        struct ieee80211com ic;
+       struct ieee80211vap vap;
+       struct ifnet ifp;
        int i, pass, total;
 
        memset(&ic, 0, sizeof(ic));
-       if (debug)
-               ic.ic_debug = IEEE80211_MSG_CRYPTO;
+       memset(&vap, 0, sizeof(vap));
+       memset(&ifp, 0, sizeof(ifp));
+
        ieee80211_crypto_attach(&ic);
 
+       /* some minimal initialization */
+       strncpy(ifp.if_xname, "test_ccmp", sizeof(ifp.if_xname));
+       vap.iv_ic = &ic;
+       vap.iv_ifp = &ifp;
+       if (debug)
+               vap.iv_debug = IEEE80211_MSG_CRYPTO;
+       ieee80211_crypto_vattach(&vap);
+
        pass = 0;
        total = 0;
-       for (i = 0; i < N(ccmptests); i++)
+       for (i = 0; i < nitems(ccmptests); i++)
                if (tests & (1<<i)) {
                        total++;
-                       pass += runtest(&ic, &ccmptests[i]);
+                       pass += runtest(&vap, &ccmptests[i]);
                }
        printf("%u of %u 802.11i AES-CCMP test vectors passed\n", pass, total);
+
+       ieee80211_crypto_vdetach(&vap);
        ieee80211_crypto_detach(&ic);
+
        return (pass == total ? 0 : -1);
-#undef N
 }
 
 static int

Modified: head/tools/regression/net80211/tkip/test_tkip.c
==============================================================================
--- head/tools/regression/net80211/tkip/test_tkip.c     Thu Oct 22 16:46:30 
2015        (r289761)
+++ head/tools/regression/net80211/tkip/test_tkip.c     Thu Oct 22 17:08:12 
2015        (r289762)
@@ -43,6 +43,7 @@
 #include <sys/socket.h>
 
 #include <net/if.h>
+#include <net/if_var.h>
 #include <net/if_media.h>
 
 #include <net80211/ieee80211_var.h>
@@ -141,7 +142,6 @@ struct tkip_ctx {
        struct ieee80211com *tc_ic;     /* for diagnostics */
 
        uint16_t tx_ttak[5];
-       int     tx_phase1_done;
        uint8_t tx_rc4key[16];
 
        uint16_t rx_ttak[5];
@@ -179,34 +179,34 @@ cmpfail(const void *gen, size_t genlen, 
 }
 
 static int
-runtest(struct ieee80211com *ic, struct ciphertest *t)
+runtest(struct ieee80211vap *vap, struct ciphertest *t)
 {
        struct tkip_ctx *ctx;
-       struct ieee80211_key key;
+       struct ieee80211_key *key = &vap->iv_nw_keys[t->keyix];
        struct mbuf *m = NULL;
        const struct ieee80211_cipher *cip;
-       u_int8_t mac[IEEE80211_ADDR_LEN];
        u_int len;
+       int hdrlen;
 
        printf("%s: ", t->name);
 
        /*
         * Setup key.
         */
-       memset(&key, 0, sizeof(key));
-       key.wk_flags = IEEE80211_KEY_XMIT | IEEE80211_KEY_RECV;
-       key.wk_cipher = &ieee80211_cipher_none;
-       if (!ieee80211_crypto_newkey(ic, IEEE80211_CIPHER_TKIP,
-           IEEE80211_KEY_XMIT | IEEE80211_KEY_RECV, &key)) {
+       memset(key, 0, sizeof(*key));
+       key->wk_flags = IEEE80211_KEY_XMIT | IEEE80211_KEY_RECV;
+       key->wk_cipher = &ieee80211_cipher_none;
+       if (!ieee80211_crypto_newkey(vap, t->cipher,
+           IEEE80211_KEY_XMIT | IEEE80211_KEY_RECV, key)) {
                printf("FAIL: ieee80211_crypto_newkey failed\n");
                goto bad;
        }
 
-       memcpy(key.wk_key, t->key, t->key_len);
-       key.wk_keylen = 128/NBBY;
-       key.wk_keyrsc = 0;
-       key.wk_keytsc = t->pn;
-       if (!ieee80211_crypto_setkey(ic, &key, mac)) {
+       memcpy(key->wk_key, t->key, t->key_len);
+       key->wk_keylen = 128/NBBY;
+       memset(key->wk_keyrsc, 0, sizeof(key->wk_keyrsc));
+       key->wk_keytsc = t->pn;
+       if (!ieee80211_crypto_setkey(vap, key)) {
                printf("FAIL: ieee80211_crypto_setkey failed\n");
                goto bad;
        }
@@ -214,18 +214,19 @@ runtest(struct ieee80211com *ic, struct 
        /*
         * Craft frame from plaintext data.
         */
-       cip = key.wk_cipher;
+       cip = key->wk_cipher;
        m = m_getcl(M_NOWAIT, MT_HEADER, M_PKTHDR);
        m->m_data += cip->ic_header;
        len = t->plaintext_len - IEEE80211_WEP_MICLEN;
        memcpy(mtod(m, void *), t->plaintext, len);
        m->m_len = len;
        m->m_pkthdr.len = m->m_len;
+       hdrlen = ieee80211_anyhdrsize(mtod(m, void *));
 
        /*
         * Add MIC.
         */
-       if (!ieee80211_crypto_enmic(ic, &key, m)) {
+       if (!ieee80211_crypto_enmic(vap, key, m, 1)) {
                printf("FAIL: tkip enmic failed\n");
                goto bad;
        }
@@ -247,14 +248,14 @@ runtest(struct ieee80211com *ic, struct 
        /*
         * Encrypt frame w/ MIC.
         */
-       if (!cip->ic_encap(&key, m, t->keyix<<6)) {
+       if (!cip->ic_encap(key, m)) {
                printf("FAIL: tkip encap failed\n");
                goto bad;
        }
        /*
         * Verify: phase1, phase2, frame length, frame contents.
         */
-       ctx = key.wk_private;
+       ctx = key->wk_private;
        if (memcmp(ctx->tx_ttak, t->phase1, t->phase1_len)) {
                printf("FAIL: encrypt phase1 botch\n");
                cmpfail(ctx->tx_ttak, sizeof(ctx->tx_ttak),
@@ -281,7 +282,7 @@ runtest(struct ieee80211com *ic, struct 
        /*
         * Decrypt frame.
         */
-       if (!cip->ic_decap(&key, m)) {
+       if (!cip->ic_decap(key, m, hdrlen)) {
                printf("tkip decap failed\n");
                /*
                 * Check reason for failure: phase1, phase2, frame data (ICV).
@@ -319,17 +320,19 @@ runtest(struct ieee80211com *ic, struct 
        /*
         * De-MIC decrypted frame.
         */
-       if (!ieee80211_crypto_demic(ic, &key, m)) {
+       if (!ieee80211_crypto_demic(vap, key, m, 1)) {
                printf("FAIL: tkip demic failed\n");
                goto bad;
        }
        /* XXX check frame length and contents... */
+       m_freem(m);
+       ieee80211_crypto_delkey(vap, key);
        printf("PASS\n");
        return 1;
 bad:
        if (m != NULL)
                m_freem(m);
-       ieee80211_crypto_delkey(ic, &key);
+       ieee80211_crypto_delkey(vap, key);
        return 0;
 }
 
@@ -343,26 +346,38 @@ static    int tests = -1;
 static int
 init_crypto_tkip_test(void)
 {
-#define        N(a)    (sizeof(a)/sizeof(a[0]))
        struct ieee80211com ic;
+       struct ieee80211vap vap;
+       struct ifnet ifp;
        int i, pass, total;
 
        memset(&ic, 0, sizeof(ic));
-       if (debug)
-               ic.ic_debug = IEEE80211_MSG_CRYPTO;
+       memset(&vap, 0, sizeof(vap));
+       memset(&ifp, 0, sizeof(ifp));
+
        ieee80211_crypto_attach(&ic);
 
+       /* some minimal initialization */
+       strncpy(ifp.if_xname, "test_ccmp", sizeof(ifp.if_xname));
+       vap.iv_ic = &ic;
+       vap.iv_ifp = &ifp;
+       if (debug)
+               vap.iv_debug = IEEE80211_MSG_CRYPTO;
+       ieee80211_crypto_vattach(&vap);
+
        pass = 0;
        total = 0;
-       for (i = 0; i < N(tkiptests); i++)
+       for (i = 0; i < nitems(tkiptests); i++)
                if (tests & (1<<i)) {
                        total++;
-                       pass += runtest(&ic, &tkiptests[i]);
+                       pass += runtest(&vap, &tkiptests[i]);
                }
        printf("%u of %u 802.11i TKIP test vectors passed\n", pass, total);
+
+       ieee80211_crypto_vdetach(&vap);
        ieee80211_crypto_detach(&ic);
+
        return (pass == total ? 0 : -1);
-#undef N
 }
 
 static int

Modified: head/tools/regression/net80211/wep/test_wep.c
==============================================================================
--- head/tools/regression/net80211/wep/test_wep.c       Thu Oct 22 16:46:30 
2015        (r289761)
+++ head/tools/regression/net80211/wep/test_wep.c       Thu Oct 22 17:08:12 
2015        (r289762)
@@ -53,6 +53,7 @@
 #include <sys/socket.h>
 
 #include <net/if.h>
+#include <net/if_var.h>
 #include <net/if_media.h>
 
 #include <net80211/ieee80211_var.h>
@@ -178,54 +179,55 @@ cmpfail(const void *gen, size_t genlen, 
 }
 
 struct wep_ctx_hw {                    /* for use with h/w support */
-       struct ieee80211com *wc_ic;     /* for diagnostics */
-       u_int32_t       wc_iv;          /* initial vector for crypto */
+       struct ieee80211vap *wc_vap;    /* for diagnostics+statistics */
+       struct ieee80211com *wc_ic;
+       uint32_t        wc_iv;          /* initial vector for crypto */
 };
 
 static int
-runtest(struct ieee80211com *ic, struct ciphertest *t)
+runtest(struct ieee80211vap *vap, struct ciphertest *t)
 {
-       struct ieee80211_key key;
+       struct ieee80211_key *key = &vap->iv_nw_keys[t->keyix];
        struct mbuf *m = NULL;
        const struct ieee80211_cipher *cip;
-       u_int8_t mac[IEEE80211_ADDR_LEN];
        struct wep_ctx_hw *ctx;
+       int hdrlen;
 
        printf("%s: ", t->name);
 
        /*
         * Setup key.
         */
-       memset(&key, 0, sizeof(key));
-       key.wk_flags = IEEE80211_KEY_XMIT | IEEE80211_KEY_RECV;
-       key.wk_cipher = &ieee80211_cipher_none;
-       if (!ieee80211_crypto_newkey(ic, t->cipher,
-           IEEE80211_KEY_XMIT | IEEE80211_KEY_RECV, &key)) {
+       memset(key, 0, sizeof(*key));
+       key->wk_flags = IEEE80211_KEY_XMIT | IEEE80211_KEY_RECV;
+       key->wk_cipher = &ieee80211_cipher_none;
+       if (!ieee80211_crypto_newkey(vap, t->cipher,
+           IEEE80211_KEY_XMIT | IEEE80211_KEY_RECV, key)) {
                printf("FAIL: ieee80211_crypto_newkey failed\n");
                goto bad;
        }
 
-       memcpy(key.wk_key, t->key, t->key_len);
-       key.wk_keylen = t->key_len;
-       if (!ieee80211_crypto_setkey(ic, &key, mac)) {
+       memcpy(key->wk_key, t->key, t->key_len);
+       key->wk_keylen = t->key_len;
+       if (!ieee80211_crypto_setkey(vap, key)) {
                printf("FAIL: ieee80211_crypto_setkey failed\n");
                goto bad;
        }
-       cip = key.wk_cipher;
 
        /*
         * Craft frame from plaintext data.
         */
-       cip = key.wk_cipher;
+       cip = key->wk_cipher;
        m = m_getcl(M_NOWAIT, MT_HEADER, M_PKTHDR);
        memcpy(mtod(m, void *), t->encrypted, t->encrypted_len);
        m->m_len = t->encrypted_len;
        m->m_pkthdr.len = m->m_len;
+       hdrlen = ieee80211_anyhdrsize(mtod(m, void *));
 
        /*
         * Decrypt frame.
         */
-       if (!cip->ic_decap(&key, m)) {
+       if (!cip->ic_decap(key, m, hdrlen)) {
                printf("FAIL: wep decap failed\n");
                cmpfail(mtod(m, const void *), m->m_pkthdr.len,
                        t->plaintext, t->plaintext_len);
@@ -249,9 +251,11 @@ runtest(struct ieee80211com *ic, struct 
        /*
         * Encrypt frame.
         */
-       ctx = (struct wep_ctx_hw *) key.wk_private;
+       ctx = (struct wep_ctx_hw *) key->wk_private;
+       ctx->wc_vap = vap;
+       ctx->wc_ic = vap->iv_ic;
        memcpy(&ctx->wc_iv, t->iv, sizeof(t->iv));      /* for encap/encrypt */
-       if (!cip->ic_encap(&key, m, t->keyix<<6)) {
+       if (!cip->ic_encap(key, m)) {
                printf("FAIL: wep encap failed\n");
                goto bad;
        }
@@ -271,13 +275,13 @@ runtest(struct ieee80211com *ic, struct 
                goto bad;
        }
        m_freem(m);
-       ieee80211_crypto_delkey(ic, &key);
+       ieee80211_crypto_delkey(vap, key);
        printf("PASS\n");
        return 1;
 bad:
        if (m != NULL)
                m_freem(m);
-       ieee80211_crypto_delkey(ic, &key);
+       ieee80211_crypto_delkey(vap, key);
        return 0;
 }
 
@@ -291,25 +295,38 @@ static    int debug = 0;
 static int
 init_crypto_wep_test(void)
 {
-#define        N(a)    (sizeof(a)/sizeof(a[0]))
        struct ieee80211com ic;
+       struct ieee80211vap vap;
+       struct ifnet ifp;
        int i, pass, total;
 
        memset(&ic, 0, sizeof(ic));
-       if (debug)
-               ic.ic_debug = IEEE80211_MSG_CRYPTO;
+       memset(&vap, 0, sizeof(vap));
+       memset(&ifp, 0, sizeof(ifp));
+
        ieee80211_crypto_attach(&ic);
+
+       /* some minimal initialization */
+       strncpy(ifp.if_xname, "test_ccmp", sizeof(ifp.if_xname));
+       vap.iv_ic = &ic;
+       vap.iv_ifp = &ifp;
+       if (debug)
+               vap.iv_debug = IEEE80211_MSG_CRYPTO;
+       ieee80211_crypto_vattach(&vap);
+
        pass = 0;
        total = 0;
-       for (i = 0; i < N(weptests); i++)
+       for (i = 0; i < nitems(weptests); i++)
                if (tests & (1<<i)) {
                        total++;
-                       pass += runtest(&ic, &weptests[i]);
+                       pass += runtest(&vap, &weptests[i]);
                }
        printf("%u of %u 802.11i WEP test vectors passed\n", pass, total);
+
+       ieee80211_crypto_vdetach(&vap);
        ieee80211_crypto_detach(&ic);
+
        return (pass == total ? 0 : -1);
-#undef N
 }
 
 static int
_______________________________________________
svn-src-all@freebsd.org mailing list
https://lists.freebsd.org/mailman/listinfo/svn-src-all
To unsubscribe, send any mail to "svn-src-all-unsubscr...@freebsd.org"

Reply via email to