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