用NIO实现的一个端口映射的原型

elgs

用NIO实现的一个端口映射的原型

2. /*
3. * Created on 2004-11-11
4. */
5. package niotest;
6. import java.io.IOException;
7. import java.net.InetSocketAddress;
8. import java.nio.ByteBuffer;
9. import java.nio.channels.SelectionKey;
10. import java.nio.channels.Selector;
11. import java.nio.channels.ServerSocketChannel;
12. import java.nio.channels.SocketChannel;
13. import java.util.Iterator;
14. /**
15. * @author elgs
16. */
17. public class Server
18. {
19.     private static Selector   selector = null;
20.     private static ByteBuffer buf      = ByteBuffer.allocateDirect(1024 * 100);
21.     private static boolean    dbg      = false;
22.     public static void main(String[] args) throws IOException
23.     {
24.         selector = Selector.open();
25.         ServerSocketChannel ssChannel = ServerSocketChannel.open();
26.         ssChannel.configureBlocking(false);
27.         ssChannel.socket().bind(new InetSocketAddress(80));
28.         ssChannel.register(selector, SelectionKey.OP_ACCEPT);
29.         System.out.println("Listening on port 80.
30. ");
31.         while (selector.select() > 0)
32.         {
33.             for (Iterator it = selector.selectedKeys().iterator(); it.hasNext();)
34.             {
35.                 // Get the selection key
36.                 SelectionKey selKey = (SelectionKey) it.next();
37.                 // Remove it from the list to indicate that it is being
38.                 // processed
39.                 it.remove();
40.                 try
41.                 {
42.                     processSelectionKey(selKey);
43.                 }
44.                 catch (Exception e)
45.                 {
46.                     e.printStackTrace();
47.                 }
48.             }
49.         }
50.     }
51.     public static void processSelectionKey(SelectionKey selKey)
52.             throws IOException
53.     {
54.         // Since the ready operations are cumulative,
55.         // need to check readiness for each operation
56.         if (selKey.isValid() && selKey.isAcceptable())
57.         {
58.             // Get channel with connection request
59.             ServerSocketChannel ssChannel = (ServerSocketChannel) selKey
60.                     .channel();
61.             SocketChannel sChannel = ssChannel.accept();
62.             sChannel.configureBlocking(false);
63.             // If serverSocketChannel is non-blocking, sChannel may
64.             // be null
65.             if (sChannel != null)
66.             {
67.                 SelectionKey clientKey = sChannel.register(selector,
68.                         SelectionKey.OP_READ | SelectionKey.OP_WRITE);
69.                 SocketChannel sServerChannel = SocketChannel.open();
70.                 sServerChannel.configureBlocking(false);
71.                 sServerChannel.connect(new InetSocketAddress("172.16.31.130",
72.                         80));
73.                 //172.16.31.130
74.                 while (!sServerChannel.finishConnect())
75.                 {
76.                     if (dbg)
77.                         System.out.print(".");
78.                 }
79.                 if (dbg)
80.                     System.out.println("Connected OK!");
81.                 SelectionKey serverKey = sServerChannel.register(selector,
82.                         sServerChannel.validOps());
83.                 clientKey.attach(serverKey);
84.                 serverKey.attach(clientKey);
85.             }
86.         }
87.         if (selKey.isValid() && selKey.isReadable())
88.         {
89.             //          Get channel with bytes to read
90.             SocketChannel sChannel = (SocketChannel) selKey.channel();
91.             SelectionKey palKey = (SelectionKey) selKey.attachment();
92.             if (palKey != null)
93.             {
94.                 SocketChannel palChannel = (SocketChannel) palKey.channel();
95.                 if (palKey.isValid() && selKey.isValid())
96.                 {
97.                     // Clear the buffer and read bytes from socket
98.                     buf.clear();
99.                     int count = 0;
100.                     try
101.                     {
102.                         while ((count = sChannel.read(buf)) > 0)
103.                         {
104.                             buf.flip(); // Make buffer readable
105.                             while (buf.hasRemaining())
106.                             {
107.                                 palChannel.write(buf);
108.                             }
109.                             buf.clear();
110.                         }
111.                     }
112.                     catch (IOException e)
113.                     {
114.                         sChannel.close();
115.                         palChannel.close();
116.                         if (dbg)
117.                             System.out.println("Forced disconnected!");
118.                     }
119.                     if (count == -1)
120.                     {
121.                         sChannel.close();
122.                         palChannel.close();
123.                         if (dbg)
124.                             System.out.println("Disconnected OK!");
125.                     }
126.                 }
127.                 else
128.                 {
129.                     sChannel.close();
130.                     palChannel.close();
131.                     if (dbg)
132.                         System.out.println("Not Connected!");
133.                 }
134.             }
135.             else
136.             {
137.                 sChannel.close();
138.                 if (dbg)
139.                     System.out.println("Timeout disconnected OK!");
140.             }
141.         }
142.     }
143. }

复制代码

本文来自ChinaUnix博客，如果查看原文请点：http://blog.chinaunix.net/u/4960/showart_15791.html