Augmented Traffic Control
Full documentation for the project is available at http://facebook.github.io/augmented-traffic-control/.
Augmented Traffic Control (ATC) is a tool to simulate network conditions. It allows controlling the connection that a device has to the internet. Aspects of the connection that can be controlled include:
- packet loss
- corrupted packets
- packets ordering
In order to be able to shape the network traffic, ATC must be running on a device that routes the traffic and sees the real IP address of the device, like your network gateway for instance.
ATC is made of multiple components that interact together:
atcd: The ATC daemon which is responsible for setting/unsetting traffic shaping.
atcdexposes a Thrift interface to interact with it.
django-atc-api: A Django app based on Django Rest Framework that provides a RESTful interface to
django-atc-demo-ui: A Django app that provides a simple Web UI to use
atcfrom a mobile phone.
django-atc-profile-storage: A Django app that can be used to save shaping profiles, making it easier to re-use them later without manually re-entering those settings.
ATC in sub-components, it make it easier to hack on it or build on top of it. While
django-atc-demo-ui is shipped as part of
ATC's main repository to allow people to be able to use
ATC out of the box, by providing a REST API to
atcd, it makes it relatively easy to interact with
atcd via the command line and opens the path for the community to be able to build creative command line tools, web UI or mobile apps that interact with
Most requirements are handled automatically by pip, the packaging system used by ATC, and each
ATC package may have different requirements and the README.md files of the respective packages should be checked for more details. Anyhow, some requirements apply to the overall codebase:
- Python 2.7+
- Django 1.7+
The fact that
ATC is splitted in multiple packages allows for multiple deployment scenarii. However, deploying all the packages on the same host is the simplest and most likely fitting most use cases.
To get more details on how to install/configure each packages, please refer to the packages'respective READMEs.
The easiest way to install
ATC is by using
for pkg in \ atc_thrift atcd django-atc-api \ django-atc-demo-ui django-atc-profile-storage do pip install $pkg done
Now that we have all the packages installed, we need to create a new Django project in which we will use our Django app.
django-admin startproject atcui cd atcui
Now that we have our django project, we need to configure it to use our apps and we need to tell it how to route to our apps.
atcui/settings.py and enable the
ATC apps by adding to
INSTALLED_APPS = ( ... # Django ATC API 'rest_framework', 'atc_api', # Django ATC Demo UI 'bootstrap_themes', 'django_static_jquery', 'atc_demo_ui', # Django ATC Profile Storage 'atc_profile_storage', )
atcui/urls.py and enable routing to the
ATC apps by adding the routes to
... ... from django.views.generic.base import RedirectView urlpatterns = patterns('', ... # Django ATC API url(r'^api/v1/', include('atc_api.urls')), # Django ATC Demo UI url(r'^atc_demo_ui/', include('atc_demo_ui.urls')), # Django ATC profile storage url(r'^api/v1/profiles/', include('atc_profile_storage.urls')), url(r'^$', RedirectView.as_view(url='/atc_demo_ui/', permanent=False)), )
Finally, let's update the Django DB:
python manage.py migrate
All require packages should now be installed and configured. We now need to run the daemon and the UI interface. While we will run
ATC straight from the command line in this example, you can refer to example sysvinit and upstart scripts.
atcd modifies network related settings and as such needs to run in privileged mode:
eth0 is your interface to connect to the internet and
eth1, your interface to connec to your lan, this should just work. If your setting is slightly different, use the command line arguments
--atcd-lan to adapt to your configuration.
The UI on the other hand is a standard Django Web app and can be run as a normal user. Make sure you are in the directory that was created when you ran
django-admin startproject atcui and run:
python manage.py runserver 0.0.0.0:8000
You should now be able to access the web UI at http://localhost:8000
ATC Code Structure
ATC source code is available under the atc directory, it is currently composed of:
- atc_thrift the thrift interface's library
- atcd the ATC daemon that runs on the router doing the traffic shaping
- django-atc-api A django app that provides a RESTful interface to
- django-atc-demo-ui A django app that provides a simple demo UI leveraging the RESTful API
- django-atc-profile-storage A django app that allows saving shaping profiles to DB allowing users to select their favorite profile from a list instead of re-entering all the profile details every time.
The chef directory contains 2 chef cookbooks:
- atc A cookbook to deploy ATC. It also allows to deploy ATC in a Virtual Box VM in order to develop on ATC.
- atclient Set up a Linux Desktop VM that can be used to test shaping end to end.
atc_thrift defines the thrift interface to communicate with the
django-atc-api is a django app that provide a REST API to the
atcd daemon. Web applications, command line tools can use the API in order to shape/unshape traffic.
django-atc-demo-ui is a simple Web UI to enable/disable traffic shaping. The UI is mostly written in React
django-atc-profile-storage allows saving profiles to DB. A typical use case will be to save a list of predefined/often used shaping settings that you want to be able to accessing in just a few clicks/taps.
Developing on ATC
To make ATC development easier, we use Virtual Box and Vagrant to provision and run a VM that will run the ATC daemon and the ATC UI from your git checkout.
Interacting with ATC will only shape the traffic within the VM and not on the host.
Setting up the environment
You will need to install VirtualBox, Vagrant and a couple of plugins:
- Chef DK
- Install some vagrant plugins:
- vagrant plugin install vagrant-berkshelf --plugin-version '>= 2.0.1'
- vagrant plugin install vagrant-omnibus
- Clone this repo: git clone email@example.com:facebook/augmented-traffic-control.git atc
Once in the repo, go to the
chef/atc directory and run:
vagrant up trusty
This will take some time before it completes, once the VM is provision, SSH into it:
vagrant ssh trusty
You should now be able to access ATC at: http://localhost:8080/
Hacking on the code
Hacking on ATC is done from the host and tested in the VM. In order to reflect the changes, you will need to start the services manually.
atcui have their python libraries installed in a python virtualenv so you will need to activate the environment in order to be able to run the services.
The virtualenv is installed in /usr/local/atc/venv/bin/activate .
Running the daemon
atcd daemon is running under the root user privileges, all operations below needs to be done as root.
To run the daemon manually, first make sure it is not running in the background:
service atcd stop
And run the daemon:
Once you are happy with your changes and you want to test them, you will need to kill the daemon and restart it in order to apply the changes.
Running the API/UI
This is a django project and, when running the django built-in HTTP server, will detect code changes and reload automatically.
To run the HTTP REST API and UI:
cd /var/django && python manage.py runserver 0.0.0.0:8000