BandwidthController.h revision 0dad7c2f1f6994fbe5e85b9e1fc72d29d6453211
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 IptOp {IptOpInsert, IptOpReplace, IptOpDelete}; 41 42 int64_t sharedQuotaBytes; 43 std::list<std::string> sharedQuotaIfaces; 44 45 std::list<QuotaInfo> quotaIfaces; 46 47 std::list<int /*appUid*/> naughtyAppUids; 48 int maninpulateNaughtyApps(int numUids, char *appStrUids[], bool doAdd); 49 50 int prepCostlyIface(const char *ifn, bool isShared); 51 int cleanupCostlyIface(const char *ifn, bool isShared); 52 53 std::string makeIptablesNaughtyCmd(IptOp op, int uid); 54 std::string makeIptablesQuotaCmd(IptOp op, char *costName, int64_t quota); 55 56 /* Runs for both ipv4 and ipv6 iptables */ 57 int runCommands(int numCommands, const char *commands[], bool allowFailure); 58 /* Runs for both ipv4 and ipv6 iptables, appends -j REJECT --reject-with ... */ 59 static int runIpxtablesCmd(const char *cmd, bool appendReject); 60 static int runIptablesCmd(const char *cmd, bool appendReject, bool isIp6); 61 62private: 63 static const char *cleanupCommands[]; 64 static const char *setupCommands[]; 65 static const char *basicAccountingCommands[]; 66 static const int MAX_CMD_LEN; 67 static const int MAX_IFACENAME_LEN; 68 static const int MAX_CMD_ARGS; 69 static const char IPTABLES_PATH[]; 70 static const char IP6TABLES_PATH[]; 71 72}; 73 74#endif 75