Tràn ngăn xếp khi sử dụng mô hình System.Net.Sockets.Socket.AcceptAsync

Question

Đối với phương thức System.Net.Sockets.Socket.AcceptAsync của C# và .NET, một phương pháp sẽ được yêu cầu để xử lý giá trị trả lại là "sai" để xử lý trạng thái sẵn có ngay lập tức SocketAsyncEventArgs từ kết nối được xử lý đồng bộ. Microsoft cung cấp các ví dụ (được tìm thấy trên trang lớp học System.Net.Sockets.SocketAsyncEventArgs) sẽ gây tràn ngăn xếp nếu có một số lượng lớn các kết nối đang chờ xử lý, có thể được khai thác trên bất kỳ hệ thống nào triển khai mô hình xử lý của chúng.

Các ý tưởng khác để giải quyết vấn đề này là tạo vòng lặp gọi phương thức xử lý, với điều kiện là giá trị Socket.AcceptAsync trả về bằng false và để ngắt vòng lặp (để cho phép xử lý trì hoãn) nếu giá trị chỉ ra rằng hoạt động đang được hoàn thành không đồng bộ (đúng). Tuy nhiên, giải pháp này cũng gây ra lỗ hổng tràn ngăn xếp do thực tế là cuộc gọi lại được liên kết với số SocketAsyncEventArgs được chuyển đến Socket.AcceptAsync có ở cuối phương thức, gọi tới số Socket.AcceptAsync, cũng có vòng lặp cho các kết nối có sẵn, đồng bộ ngay lập tức .

Như bạn có thể thấy, đây là một vấn đề khá vững chắc và tôi chưa tìm được giải pháp tốt nào không liên quan đến System.Threading.ThreadPool và tạo tấn phương pháp khác và xử lý lịch biểu. Theo như tôi thấy, mô hình socket không đồng bộ liên quan đến Socket.AcceptAsync đòi hỏi nhiều hơn những gì được thể hiện trong các ví dụ trên MSDN.

Có ai có giải pháp sạch và hiệu quả để xử lý ngay các kết nối đang chờ xử lý được chấp nhận đồng bộ từ Socket.AcceptAsync mà không cần tạo chuỗi riêng để xử lý các kết nối và không sử dụng đệ quy không?

Answer 1

tôi sẽ không sử dụng AcceptAsync, nhưng thay vì BeginAccept/EndAccept và triển khai đúng mẫu không đồng bộ chính xác, nghĩa là, kiểm tra CompletedSynchronously để tránh các cuộc gọi lại trong chuỗi gọi lại về các thao tác đã hoàn tất.

cũng Xem AsyncCallBack CompletedSynchronously

---

Chỉnh sửa liên quan đến nhu cầu sử dụng AcceptAsync:

Các MSDN documentation dứt khoát nói rằng callback sẽ KHÔNG được gọi cho các hoạt động mà hoàn thành đồng bộ. Điều này khác với mẫu không đồng bộ chung, nơi gọi lại luôn được gọi.

Trả về true nếu hoạt động I/O là đang chờ xử lý. SocketAsyncEventArgs. Sự kiện đã hoàn thành trên thông số e sẽ được nâng lên khi hoàn tất thao tác . Trả về sai nếu hoạt động I/O hoàn thành đồng bộ. SocketAsyncEventArgs.Các sự kiện đã hoàn thành trên tham số e sẽ không được nâng lên và đối tượng e được chuyển thành tham số có thể được kiểm tra ngay sau khi gọi phương thức để truy xuất kết quả .

Tôi hiện không thấy cách vòng lặp không giải quyết được sự cố tràn ngăn xếp. Có lẽ bạn có thể cụ thể hơn về mã gây ra sự cố?

---

Chỉnh sửa 2: Tôi đang nghĩ đến việc mã như thế này (chỉ liên quan đến AcceptAsync, phần còn lại là chỉ để có được một ứng dụng làm việc để thử nó ra với):

static void Main(string[] args) { 
    Socket listenSocket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp); 
    listenSocket.Bind(new IPEndPoint(IPAddress.Loopback, 4444)); 
    listenSocket.Listen(100); 
    SocketAsyncEventArgs e = new SocketAsyncEventArgs(); 
    e.Completed += AcceptCallback; 
    if (!listenSocket.AcceptAsync(e)) { 
     AcceptCallback(listenSocket, e); 
    } 
    Console.ReadKey(true); 
} 

private static void AcceptCallback(object sender, SocketAsyncEventArgs e) { 
    Socket listenSocket = (Socket)sender; 
    do { 
     try { 
      Socket newSocket = e.AcceptSocket; 
      Debug.Assert(newSocket != null); 
      // do your magic here with the new socket 
      newSocket.Send(Encoding.ASCII.GetBytes("Hello socket!")); 
      newSocket.Disconnect(false); 
      newSocket.Close(); 
     } catch { 
      // handle any exceptions here; 
     } finally { 
      e.AcceptSocket = null; // to enable reuse 
     } 
    } while (!listenSocket.AcceptAsync(e)); 
} 

Answer 2

Tôi đã không nhìn cẩn thận, nhưng nó có mùi như thế này có thể hữu ích (xem phần gọi là "chồng lặn"):

http://blogs.msdn.com/b/mjm/archive/2005/05/04/414793.aspx

Answer 3

Tôi có giải quyết vấn đề này bằng cách thay đổi vị trí của vòng lặp. Thay vì đệ quy gọi trình xử lý chấp nhận từ bên trong chính nó, gói mã trong một vòng lặp do-while với điều kiện là "! Socket.AcceptAsync (args)" ngăn chặn tràn ngăn xếp.

Lý do đằng sau việc này là bạn sử dụng chuỗi gọi lại để xử lý các kết nối có sẵn ngay lập tức, trước khi bận tâm chờ đợi không đồng bộ cho các kết nối khác đi qua. Đó là tái sử dụng một thread gộp lại, có hiệu quả.

Tôi đánh giá cao câu trả lời nhưng vì một lý do nào đó mà không ai trong số họ nhấp vào tôi và không thực sự giải quyết vấn đề. Tuy nhiên, có vẻ như một cái gì đó trong đó đã kích hoạt tâm trí của tôi để đưa ra ý tưởng đó. Nó tránh làm việc thủ công với lớp ThreadPool và không sử dụng đệ quy.

Tất nhiên, nếu ai đó có giải pháp tốt hơn hoặc thậm chí là giải pháp thay thế, tôi rất sẵn lòng nghe nó.

Answer 4

newSocket.Send(Encoding.ASCII.GetBytes("Hello socket!")); 
newSocket.Disconnect(false); 
newSocket.Close(); 

Vấn đề với đoạn mã trên là điều này sẽ chặn hoạt động chấp nhận tiếp theo của bạn.

Cách tốt hơn là như thế này:

while (true) 
{ 
    if (e.SocketError == SocketError.Success) 
    { 
     //ReadEventArg object user token 
     SocketAsyncEventArgs readEventArgs = m_readWritePool.Pop(); 
     Socket socket = ((AsyncUserToken)readEventArgs.UserToken).Socket = e.AcceptSocket; 

     if (!socket.ReceiveAsync(readEventArgs)) 
     ThreadPool.QueueUserWorkItem(new WaitCallback(ProcessReceiveEx), readEventArgs); . 
    } 
    else 
    { 
     HadleBadAccept(e); 
    } 

    e.AcceptSocket = null; 

    m_maxNumberAcceptedClients.WaitOne(); 
    if (listenSocket.AcceptAsync(e)) 
     break; 
}