BandwidthController.h revision 26e0d49fa743d7881104196a9eda733bd2aac92f 
1/*
2 * Copyright (C) 2011 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16#ifndef _BANDWIDTH_CONTROLLER_H
17#define _BANDWIDTH_CONTROLLER_H
18
19#include <list>
20#include <string>
21#include <utility>  // for pair
22class BandwidthController {
23public:
24    BandwidthController();
25    int enableBandwidthControl(void);
26    int disableBandwidthControl(void);
27
28    int setInterfaceSharedQuota(const char *iface, int64_t bytes);
29    int removeInterfaceSharedQuota(const char *iface);
30
31    int setInterfaceQuota(const char *iface, int64_t bytes);
32    int removeInterfaceQuota(const char *iface);
33
34    int addNaughtyApps(int numUids, char *appUids[]);
35    int removeNaughtyApps(int numUids, char *appUids[]);
36
37
38protected:
39    typedef std::pair<std::string /*ifaceName*/, int64_t /*quota*/> QuotaInfo;
40    enum IptIpVer { IptIpV4, IptIpV6 };
41    enum IptOp { IptOpInsert, IptOpReplace, IptOpDelete };
42    enum IptRejectOp { IptRejectAdd, IptRejectNoAdd };
43    enum NaughtyAppOp { NaughtyAppOpAdd, NaughtyAppOpRemove };
44    enum QuotaType { QuotaUnique, QuotaShared };
45    enum RunCmdErrHandling { RunCmdFailureBad, RunCmdFailureOk };
46
47    int64_t sharedQuotaBytes;
48    std::list<std::string> sharedQuotaIfaces;
49
50    std::list<QuotaInfo> quotaIfaces;
51
52    std::list<int /*appUid*/> naughtyAppUids;
53    int maninpulateNaughtyApps(int numUids, char *appStrUids[], NaughtyAppOp appOp);
54
55    int prepCostlyIface(const char *ifn, QuotaType quotaType);
56    int cleanupCostlyIface(const char *ifn, QuotaType quotaType);
57
58    std::string makeIptablesNaughtyCmd(IptOp op, int uid);
59    std::string makeIptablesQuotaCmd(IptOp op, const char *costName, int64_t quota);
60
61    /* Runs for both ipv4 and ipv6 iptables */
62    int runCommands(int numCommands, const char *commands[], RunCmdErrHandling cmdErrHandling);
63    /* Runs for both ipv4 and ipv6 iptables, appends -j REJECT --reject-with ...  */
64    static int runIpxtablesCmd(const char *cmd, IptRejectOp rejectHandling);
65    static int runIptablesCmd(const char *cmd, IptRejectOp rejectHandling, IptIpVer iptIpVer);
66
67    // Provides strncpy() + check overflow.
68    static int StrncpyAndCheck(char *buffer, const char *src, size_t buffSize);
69
70private:
71    static const char *cleanupCommands[];
72    static const char *setupCommands[];
73    static const char *basicAccountingCommands[];
74    static const int MAX_CMD_LEN;
75    static const int MAX_IFACENAME_LEN;
76    static const int MAX_CMD_ARGS;
77    static const char IPTABLES_PATH[];
78    static const char IP6TABLES_PATH[];
79
80};
81
82#endif
83