1 /* 2 * Copyright (C) 2008 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 17 #include <stdlib.h> 18 #include <errno.h> 19 #include <fcntl.h> 20 21 #include <sys/socket.h> 22 #include <sys/stat.h> 23 #include <sys/types.h> 24 #include <sys/wait.h> 25 26 #include <linux/netlink.h> 27 #include <linux/rtnetlink.h> 28 #include <linux/pkt_sched.h> 29 30 #define LOG_TAG "ThrottleController" 31 #include <cutils/log.h> 32 33 34 #include "ThrottleController.h" 35 36 static char TC_PATH[] = "/system/bin/tc"; 37 38 extern "C" int logwrap(int argc, const char **argv, int background); 39 extern "C" int ifc_init(void); 40 extern "C" int ifc_up(const char *name); 41 extern "C" int ifc_down(const char *name); 42 43 int ThrottleController::runTcCmd(const char *cmd) { 44 char buffer[255]; 45 46 strncpy(buffer, cmd, sizeof(buffer)-1); 47 48 const char *args[32]; 49 char *next = buffer; 50 char *tmp; 51 52 args[0] = TC_PATH; 53 int i = 1; 54 55 while ((tmp = strsep(&next, " "))) { 56 args[i++] = tmp; 57 if (i == 32) { 58 LOGE("tc argument overflow"); 59 errno = E2BIG; 60 return -1; 61 } 62 } 63 args[i] = NULL; 64 65 return logwrap(i, args, 0); 66 } 67 68 int ThrottleController::setInterfaceThrottle(const char *iface, int rxKbps, int txKbps) { 69 char cmd[512]; 70 char ifn[65]; 71 int rc; 72 73 memset(ifn, 0, sizeof(ifn)); 74 strncpy(ifn, iface, sizeof(ifn)-1); 75 76 if (txKbps == -1) { 77 reset(ifn); 78 return 0; 79 } 80 81 /* 82 * 83 * Target interface configuration 84 * 85 */ 86 87 /* 88 * Add root qdisc for the interface 89 */ 90 sprintf(cmd, "qdisc add dev %s root handle 1: htb default 1 r2q 1000", ifn); 91 if (runTcCmd(cmd)) { 92 LOGE("Failed to add root qdisc (%s)", strerror(errno)); 93 goto fail; 94 } 95 96 /* 97 * Add our egress throttling class 98 */ 99 sprintf(cmd, "class add dev %s parent 1: classid 1:1 htb rate %dkbit", ifn, txKbps); 100 if (runTcCmd(cmd)) { 101 LOGE("Failed to add egress throttling class (%s)", strerror(errno)); 102 goto fail; 103 } 104 105 /* 106 * Bring up the IFD device 107 */ 108 ifc_init(); 109 if (ifc_up("ifb0")) { 110 LOGE("Failed to up ifb0 (%s)", strerror(errno)); 111 goto fail; 112 } 113 114 /* 115 * Add root qdisc for IFD 116 */ 117 sprintf(cmd, "qdisc add dev ifb0 root handle 1: htb default 1 r2q 1000"); 118 if (runTcCmd(cmd)) { 119 LOGE("Failed to add root ifb qdisc (%s)", strerror(errno)); 120 goto fail; 121 } 122 123 /* 124 * Add our ingress throttling class 125 */ 126 sprintf(cmd, "class add dev ifb0 parent 1: classid 1:1 htb rate %dkbit", rxKbps); 127 if (runTcCmd(cmd)) { 128 LOGE("Failed to add ingress throttling class (%s)", strerror(errno)); 129 goto fail; 130 } 131 132 /* 133 * Add ingress qdisc for pkt redirection 134 */ 135 sprintf(cmd, "qdisc add dev %s ingress", ifn); 136 if (runTcCmd(cmd)) { 137 LOGE("Failed to add ingress qdisc (%s)", strerror(errno)); 138 goto fail; 139 } 140 141 /* 142 * Add filter to link <ifn> -> ifb0 143 */ 144 sprintf(cmd, "filter add dev %s parent ffff: protocol ip prio 10 u32 match " 145 "u32 0 0 flowid 1:1 action mirred egress redirect dev ifb0", ifn); 146 if (runTcCmd(cmd)) { 147 LOGE("Failed to add ifb filter (%s)", strerror(errno)); 148 goto fail; 149 } 150 151 return 0; 152 fail: 153 reset(ifn); 154 return -1; 155 } 156 157 void ThrottleController::reset(const char *iface) { 158 char cmd[128]; 159 160 sprintf(cmd, "qdisc del dev %s root", iface); 161 runTcCmd(cmd); 162 sprintf(cmd, "qdisc del dev %s ingress", iface); 163 runTcCmd(cmd); 164 165 runTcCmd("qdisc del dev ifb0 root"); 166 } 167 168 int ThrottleController::getInterfaceRxThrottle(const char *iface, int *rx) { 169 *rx = 0; 170 return 0; 171 } 172 173 int ThrottleController::getInterfaceTxThrottle(const char *iface, int *tx) { 174 *tx = 0; 175 return 0; 176 } 177
