Merge "InetDiagSocketTest: enable UDP unit tests on all devices"

    private ConnectivityManager mCm;

    private Context mContext;

    private final static int SOCKET_TIMEOUT_MS = 100;

    private boolean mInetDiagUdpEnabled;

    @Before

    public void setUp() throws Exception {

        Instrumentation instrumentation = InstrumentationRegistry.getInstrumentation();

        mContext = instrumentation.getTargetContext();

        mCm = (ConnectivityManager) mContext.getSystemService(Context.CONNECTIVITY_SERVICE);

        int expectedUid = Process.myUid();

        UdpConnection udp = new UdpConnection("127.0.0.1", "127.0.0.2");

        int uid = mCm.getConnectionOwnerUid(udp.protocol, udp.local, udp.remote);

        mInetDiagUdpEnabled = (uid == expectedUid);

    }

    private class Connection {

        checkConnectionOwnerUid(tcp.protocol, tcp.local, new InetSocketAddress(0), false);

        tcp.close();

        /**

         * TODO: STOPSHIP: Always test for UDP, do not allow opt-out.

         */

        if (!mInetDiagUdpEnabled) return;

        /**

         * For UDP connections, either a complete match {protocol, local, remote} or a

         * partial match {protocol, local} should return a valid UID.