File: ipfilter.h

package info (click to toggle)
eiskaltdcpp 2.4.2-1.4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 25,788 kB
  • sloc: cpp: 97,597; ansic: 5,004; perl: 1,897; xml: 1,440; sh: 1,313; php: 661; javascript: 257; makefile: 39
file content (81 lines) | stat: -rw-r--r-- 2,245 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
/***************************************************************************
 *                                                                         *
 *   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 3 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/

#pragma once

#include <string>

#include "dcpp/stdinc.h"
#include "dcpp/Singleton.h"

enum eDIRECTION {
    eDIRECTION_IN = 0,
    eDIRECTION_OUT,
    eDIRECTION_BOTH
};

enum eTableAction {
    etaDROP=0,
    etaACPT
};

typedef struct _IPFilterElem {
    uint32_t ip;
    uint32_t mask;

    eDIRECTION direction;
    eTableAction action;
} IPFilterElem;

typedef std::unordered_map<uint32_t, IPFilterElem*> IPHash;
typedef std::vector<IPFilterElem*> IPList;

class IPFilter :
        public dcpp::Singleton<IPFilter>
{
    friend class dcpp::Singleton<IPFilter>;

public:
    static uint32_t StringToUint32(const std::string&);
    static std::string Uint32ToString(uint32_t);
    static uint32_t MaskToCIDR(uint32_t);
    static uint32_t MaskForBits(uint32_t);
    static bool ParseString(std::string, uint32_t&, uint32_t&, eTableAction&);

    void load();
    void shutdown();
    void loadList();
    void saveList();

    const IPList &getRules();
    const IPHash &getHash ();

    bool addToRules(const std::string &exp, eDIRECTION direction);
    void remFromRules(std::string exp, eTableAction);
    void changeRuleDirection(std::string exp, eDIRECTION, eTableAction);
    void clearRules();

    void moveRuleUp(uint32_t, eTableAction);
    void moveRuleDown(uint32_t, eTableAction);

    bool OK(const std::string &exp, eDIRECTION direction);

    void exportTo(std::string path, std::string& error);
    void importFrom(std::string path, std::string& error);

    void step(uint32_t, eTableAction, bool down = true);

private:
    IPFilter();
    virtual ~IPFilter();


    IPHash list_ip;
    IPList rules;
};