RSNEAPOL is working. Key flags accessors are not implemented yet.

include/eapol.h

@@ -7,13 +7,21 @@
 
 
 namespace Tins {
+    
+    /** \cond 
+     * Forward declaration. Avoid header inclusion.
+     */
+    class RSNInformation;
+    /** \endcond */
+    
     /**
      * \brief Class that represents the EAP encapsulation over LAN.
      */
     class EAPOL : public PDU {
     public:
         enum EAPOLTYPE {
-            RC4 = 1
+            RC4 = 1,
+            RSN,
         };
         
         /**
@@ -158,7 +166,7 @@ namespace Tins {
          * \brief Getter for the replay counter field.
          * \return The replay counter field.
          */
-        uint64_t replay_counter() const { return _header.replay_counter; }
+        uint64_t replay_counter() const { return Utils::net_to_host_ll(_header.replay_counter); }
         
         /**
          * \brief Getter for the key IV field.
@@ -269,7 +277,155 @@ namespace Tins {
      */
     class RSNEAPOL : public EAPOL {
     public:
-    
+        /**
+         * \brief Creates an instance of RSNEAPOL.
+         */
+        RSNEAPOL();
+        
+        /**
+         * \brief Destructor.
+         * 
+         * Memory allocated for the key field is freed(if any).
+         */
+        ~RSNEAPOL();
+        
+        /* Getters */
+        
+        /**
+         * \brief Getter for the key length field.
+         * \return The key length field.
+         */
+        uint16_t key_length() const { return Utils::net_to_host_s(_header.key_length); }
+        
+        /**
+         * \brief Getter for the replay counter field.
+         * \return The replay counter field.
+         */
+        uint64_t replay_counter() const { return Utils::net_to_host_ll(_header.replay_counter); }
+        
+        /**
+         * \brief Getter for the key IV field.
+         * \return The key IV field.
+         */
+        const uint8_t *key_iv() const { return _header.key_iv; }
+        
+        /**
+         * \brief Getter for the nonce field.
+         * \return The nonce field.
+         */
+        const uint8_t *nonce() const { return _header.nonce; }
+        
+        /**
+         * \brief Getter for the rsc field.
+         * \return The rsc field.
+         */
+        uint64_t rsc() const { return Utils::net_to_host_ll(_header.rsc); }
+        
+        /**
+         * \brief Getter for the id field.
+         * \return The id field.
+         */
+        uint64_t id() const { return Utils::net_to_host_ll(_header.id); }
+        
+        /**
+         * \brief Getter for the mic field.
+         * \return The mic field.
+         */
+        const uint8_t *mic() const { return _header.mic; }
+        
+        /**
+         * \brief Getter for the wpa length field.
+         * \return The wpa length field.
+         */
+        uint16_t wpa_length() const { return Utils::net_to_host_s(_header.wpa_length); }
+        
+        /**
+         * \brief Getter for the key field.
+         * \return The key field.
+         */
+        const uint8_t *key() const { return _key; }
+        
+        /**
+         * \brief Returns the header size.
+         *
+         * This metod overrides PDU::header_size. This size includes the
+         * payload and options size.
+         *
+         * \sa PDU::header_size
+         */
+        uint32_t header_size() const;
+        
+        /* Setters */
+        
+        /**
+         * \brief Sets the key length field.
+         * \param new_key_length The new key length to be set.
+         */
+        void key_length(uint16_t new_key_length);
+        
+        /**
+         * \brief Sets the replay counter field.
+         * \param new_replay_counter The new replay counter to be set.
+         */
+        void replay_counter(uint16_t new_replay_counter);
+        
+        /**
+         * \brief Sets the key IV field.
+         * \param new_key_iv The new key IV to be set.
+         */
+        void key_iv(const uint8_t *new_key_iv);
+        
+        /**
+         * \brief Sets the nonce field.
+         * 
+         * This method sets the nonce field. This field is 32 bytes long,
+         * therefore the input buffer should be at least that length.
+         * \param new_nonce The new nonce to be set.
+         */
+        void nonce(const uint8_t *new_nonce);
+        
+        /**
+         * \brief Sets the rsc field.
+         * \param new_rsc The new rsc to be set.
+         */
+        void rsc(uint64_t new_rsc);
+        
+        /**
+         * \brief Sets the id field.
+         * \param new_id The new id to be set.
+         */
+        void id(uint64_t new_id);
+        
+        /**
+         * \brief Sets the mic field.
+         * 
+         * This method sets the mic field. This field is 16 bytes long,
+         * therefore the input buffer should be at least that length.
+         * \param new_mic The new mic to be set.
+         */
+        void mic(const uint8_t *new_mic);
+        
+        /**
+         * \brief Sets the wpa length field.
+         * \param new_wpa_length The new wpa length to be set.
+         */
+        void wpa_length(uint16_t new_wpa_length);
+        
+        /**
+         * \brief Sets the key field.
+         * \param new_key The new key to be set.
+         */
+        void key(const uint8_t *new_key, uint32_t sz);
+        
+        /**
+         * \brief Sets RSN information for this EAPOL PDU.
+         * 
+         * This method copies the RSN information and copies it in the
+         * key field. Therefore, if a key has been set, this will remove it.
+         * \param rsn The RSN information to be set.
+         * \sa RSNInformation.
+         */
+        void rsn_information(const RSNInformation &rsn);
     private:
         struct rsnhdr {
             uint16_t key_mic:1,
@@ -282,7 +438,7 @@ namespace Tins {
                 key_type:1,
                 key_index:2,
                 install:1,
-                key_ack:1;             
+                key_ack:1;       
             uint16_t key_length;
             uint64_t replay_counter;
             uint8_t nonce[32], key_iv[16];
@@ -290,6 +446,13 @@ namespace Tins {
             uint8_t mic[16];
             uint16_t wpa_length;
         } __attribute__((__packed__));
+        
+        void write_body(uint8_t *buffer, uint32_t total_sz);
+        
+        
+        rsnhdr _header;
+        uint8_t *_key;
+        uint32_t _key_size;
     };
 };