From 1a62694a3b1d7693ae625e2b6a50a34c84d355eb Mon Sep 17 00:00:00 2001
From: clowwindy <clowwindy42@gmail.com>
Date: Fri, 10 Jul 2015 15:02:33 +0800
Subject: [PATCH] add udp source port test

---
 .travis.yml           |  2 +-
 tests/jenkins.sh      |  4 +++-
 tests/test_udp_src.py | 31 +++++++++++++++++++++++++++++++
 tests/test_udp_src.sh | 23 +++++++++++++++++++++++
 4 files changed, 58 insertions(+), 2 deletions(-)
 create mode 100644 tests/test_udp_src.py
 create mode 100755 tests/test_udp_src.sh

diff --git a/.travis.yml b/.travis.yml
index f29cb96..535bc9a 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -13,7 +13,7 @@ before_install:
   - sudo dd if=/dev/urandom of=/usr/share/nginx/www/file bs=1M count=10
   - sudo sh -c "echo '127.0.0.1    localhost' > /etc/hosts"
   - sudo service nginx restart
-  - pip install pep8 pyflakes nose coverage
+  - pip install pep8 pyflakes nose coverage PySocks
   - sudo tests/socksify/install.sh
   - sudo tests/libsodium/install.sh
   - sudo tests/setup_tc.sh
diff --git a/tests/jenkins.sh b/tests/jenkins.sh
index 71d5b1c..5e4e61f 100755
--- a/tests/jenkins.sh
+++ b/tests/jenkins.sh
@@ -24,6 +24,8 @@ function run_test {
     return 0
 }
 
+pip install PySocks
+
 python --version
 coverage erase
 mkdir tmp
@@ -69,7 +71,7 @@ if [ -f /proc/sys/net/ipv4/tcp_fastopen ] ; then
 fi
 
 run_test tests/test_large_file.sh
-
+run_test tests/test_udp_src.sh
 run_test tests/test_command.sh
 
 coverage combine && coverage report --include=shadowsocks/*
diff --git a/tests/test_udp_src.py b/tests/test_udp_src.py
new file mode 100644
index 0000000..83c615d
--- /dev/null
+++ b/tests/test_udp_src.py
@@ -0,0 +1,31 @@
+#!/usr/bin/python
+
+import socket
+import socks
+
+if __name__ == '__main__':
+    sock_out = socks.socksocket(socket.AF_INET, socket.SOCK_DGRAM,
+                                socket.SOL_UDP)
+    sock_out.set_proxy(socks.SOCKS5, '127.0.0.1', 1081)
+    sock_out.bind(('127.0.0.1', 9000))
+
+    sock_in1 = socket.socket(socket.AF_INET, socket.SOCK_DGRAM,
+                             socket.SOL_UDP)
+    sock_in2 = socket.socket(socket.AF_INET, socket.SOCK_DGRAM,
+                             socket.SOL_UDP)
+
+    sock_in1.bind(('127.0.0.1', 9001))
+    sock_in2.bind(('127.0.0.1', 9002))
+
+    sock_out.sendto('data', ('127.0.0.1', 9001))
+    result1 = sock_in1.recvfrom(8)
+
+    sock_out.sendto('data', ('127.0.0.1', 9002))
+    result2 = sock_in2.recvfrom(8)
+
+    sock_out.close()
+    sock_in1.close()
+    sock_in2.close()
+
+    # make sure they're from the same source port
+    assert result1 == result2
diff --git a/tests/test_udp_src.sh b/tests/test_udp_src.sh
new file mode 100755
index 0000000..876f242
--- /dev/null
+++ b/tests/test_udp_src.sh
@@ -0,0 +1,23 @@
+#!/bin/bash
+
+PYTHON="coverage run -p -a"
+
+mkdir -p tmp
+
+$PYTHON shadowsocks/local.py -c tests/aes.json &
+LOCAL=$!
+
+$PYTHON shadowsocks/server.py -c tests/aes.json --forbidden-ip "" &
+SERVER=$!
+
+sleep 3
+
+python tests/test_udp_src.py
+r=$?
+
+kill -s SIGINT $LOCAL
+kill -s SIGINT $SERVER
+
+sleep 2
+
+exit $r