src/wit-import/wit_importer.hh

/*
 * WiPal - A library and a set of tools to manipulate wireless traces.
 * Copyright (C) 2007, 2009  Universite Pierre et Marie Curie - Paris 6
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA  02110-1301  USA
 *
 * Author: Thomas Claveirole <thomas.claveirole@lip6.fr>
 */
#ifndef WIT_IMPORTER_HH_
# define WIT_IMPORTER_HH_

# include <string>
# include <map>
# include <vector>

# include <boost/date_time/posix_time/posix_time.hpp>
# include <boost/shared_ptr.hpp>

# include <wipal/tool/datafile.hh>
# include <wipal/pcap/stream.hh>
# include <wipal/wifi/addr.hh>
# include <wipal/wifi/addr_mapping.hh>
# include <wipal/phy/prism_header.hh>
# include <wipal/wifi/dissector/dissector.hh>
# include <wipal/wifi/dissector/default_hooks.hh>

namespace wit
{

  struct importer
  {
    enum error
      {
        success                 = 0,
        malformed_frame         = 1,
        malformed_prism         = 2,
        malformed_802           = 3,
        malformed_other         = 4,
        driver_reported         = 5,
        unexpected_section      = 6,
        bad_length              = 7,
        bad_type_subtype        = 8
      };

    importer(const std::string& prefix);

    template <class T>
    void        operator () (const wpl::pkt::packet<T>& d);

  protected:
    static const std::string    db_null;

    enum
      {
        channels_max            = 14
      };

    /*----------------.
    | importer::hooks |
    `----------------*/

    struct hooks: public wpl::wifi::dissector_default_hooks
    {
      hooks(wpl::wifi::addr_mapping& m);

      std::string       addrs() const;
      std::string       beacon_timestamp() const;
      std::string       fragnum() const;
      std::string       seqnum() const;
      std::string       error() const;

    protected:

      void      invalid_type_or_subtype_hook(const void*, size_t, status);
      void      truncated_frame_hook(const void*, size_t, status);
      void      addr_hook(const void*, size_t,
                          unsigned, const wpl::wifi::addr&);
      void      seq_ctl_hook(const void*, size_t, unsigned, unsigned);
      void      beacon_hook(const wpl::wifi::mgt::header*, size_t);
      void      end_of_frame_hook(const void*, size_t, status);

    private:

      wpl::wifi::addr_mapping&  mapping_;
      std::string               addrs_;
      std::string               beacon_timestamp_;
      std::string               fragnum_;
      std::string               seqnum_;
      std::string               error_;
    };


    static std::string  frames_data_filename(const std::string&, unsigned);

    static std::string  mapping_filename(const std::string&,  unsigned);

    std::string         handle_time_data(const wpl::pkt::metadata&);

    void                handle_malformed_prism();

    const wpl::prism::header*
    handle_prism_header(const std::string& time_string,
                        const void*,
                        size_t caplen);

    unsigned                                    last_channel_;

    boost::posix_time::ptime                    last_frame_;

    unsigned                                    last_id_[channels_max];

    std::vector<wpl::tool::datafile>            frames_data_;

    std::vector<wpl::wifi::addr_mapping>        mappings_;
  };

} // End of namespace wit.

# include "wit_importer.hxx"

#endif // ! WIT_IMPORTER_HH_

---