rtyler
/
ada-playground
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add some beginning work to make an epoll(7) based echo server

This commit is contained in:
R. Tyler Croy 2013-03-24 17:15:43 -07:00
parent 889546daab
commit 6f7d8d3cd0
4 changed files with 219 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
epollecho/epollecho.gpr Normal file
View File

@ -0,0 +1,5 @@
project Epollecho is
for Source_Dirs use ("src");
for Object_Dir use "obj";
for Main use ("main.adb");
end Epollecho;

9
epollecho/src/epoll.adb Normal file
View File

@ -0,0 +1,9 @@
package body Epoll is
function Create (Size : Integer) return Epoll_Fd_Type is
begin
return Create (Interfaces.C.int (Size));
end Create;
end Epoll;

75
epollecho/src/epoll.ads Normal file
View File

@ -0,0 +1,75 @@
with GNAT.Sockets,
Interfaces.C,
System;
package Epoll is
subtype Epoll_Fd_Type is Integer;
type Epoll_Events_Type is (Epoll_In,
Epoll_Pri,
Epoll_Out,
Epoll_Et,
Epoll_In_And_Et
);
for Epoll_Events_Type use (Epoll_In => 1,
Epoll_Pri => 2,
Epoll_Out => 4,
Epoll_Et => 2147483648,
Epoll_In_And_Et => 2147483649);
for Epoll_Events_Type'Size use Interfaces.C.int'Size;
type Epoll_Ctl_Type is (Epoll_Ctl_Add,
Epoll_Ctl_Del,
Epoll_Ctl_Mod);
for Epoll_Ctl_Type use (Epoll_Ctl_Add => 1,
Epoll_Ctl_Del => 2,
Epoll_Ctl_Mod => 3);
type Data_Type (Discriminant : Interfaces.C.unsigned := 0) is record
case Discriminant is
when 0 =>
Ptr : System.Address;
when 1 =>
FD : aliased GNAT.Sockets.Socket_Type; -- Interfaces.C.int
when 2 =>
U32 : aliased Interfaces.Unsigned_32;
when others =>
U64 : aliased Interfaces.Unsigned_64;
end case;
end record;
pragma Convention (C_Pass_By_Copy, Data_Type);
pragma Unchecked_Union (Data_Type);
type Event_Type is record
Events : aliased Epoll_Events_Type;
Data : Data_Type;
end record;
pragma Convention (C_Pass_By_Copy, Event_Type);
type Event_Array_Type is array (Positive range <>) of aliased Event_Type;
pragma Convention (C, Event_Array_Type);
--for Event_Array_Type'Component_Size use Event_Type'Size;
function Create (Size : Interfaces.C.int) return Epoll_Fd_Type;
pragma Import (C, Create, "epoll_create");
function Create (Size : Integer) return Epoll_Fd_Type;
function Control (Epfd : Epoll_Fd_Type;
Op : Epoll_Ctl_Type;
Fd : Epoll_Fd_Type;
Events : access Event_Type) return Interfaces.C.int;
pragma Import (C, Control, "epoll_ctl");
function Wait (Epfd : Epoll_Fd_Type;
Events : access Event_Type;
Max_Events : Interfaces.C.int;
Timeout : Interfaces.C.int) return Interfaces.C.int;
pragma Import (C, Wait, "epoll_wait");
end Epoll;

130
epollecho/src/main.adb Normal file
View File

@ -0,0 +1,130 @@
with Ada.Text_IO,
Ada.Command_Line,
Ada.Streams,
Interfaces.C;
with GNAT.Sockets;
with Epoll;
procedure Main is
Event : aliased Epoll.Event_Type;
Events : Epoll.Event_Array_Type;
First_Event : aliased Epoll.Event_Type := Events (Events'First);
ServerSock : GNAT.Sockets.Socket_Type;
ServerAddr : GNAT.Sockets.Sock_Addr_Type;
Listen_Addr : constant String := "127.0.0.1";
Listen_Port : constant Integer := 2046;
EpollFD : Epoll.Epoll_Fd_Type;
Retval, Num_FDs : Interfaces.C.int;
use Ada.Text_IO,
Interfaces.C,
GNAT.Sockets;
procedure Read_Data (S : in GNAT.Sockets.Socket_Type) is
Channel : Stream_Access := Stream (S);
Data : Ada.Streams.Stream_Element_Array (1 .. 1);
Offset : Ada.Streams.Stream_Element_Count;
use Ada.Streams;
begin
while true loop
Ada.Streams.Read (Channel.All, Data, Offset);
exit when Offset = 0;
Put (Character'Val (Data (1)));
end loop;
Put_Line (".. closing connection");
GNAT.Sockets.Close_Socket (S);
end Read_Data;
begin
Put_Line ("Starting epoll-based echo server...");
ServerAddr.Addr := Inet_Addr (Listen_Addr);
ServerAddr.Port := Port_Type (Listen_Port);
Create_Socket (ServerSock);
Set_Socket_Option (ServerSock, Socket_Level, (Reuse_Address, True));
Bind_Socket (ServerSock, ServerAddr);
Put_Line (".. bound to socket");
Listen_Socket (ServerSock);
Put_Line (".. listening for connections");
EpollFD := Epoll.Create (Events'Last);
if EpollFD = -1 then
Put_Line ("Failed to create epoll(7) file descriptor");
Ada.Command_Line.Set_Exit_Status (1);
return;
end if;
Event.Events := Epoll.Epoll_In;
Event.Data.FD := ServerSock;
Retval := Epoll.Control (EpollFD, Epoll.Epoll_Ctl_Add, To_C (ServerSock), Event'Access);
if Retval = -1 then
Put_Line ("Epoll.Control call failed, not sure why");
Ada.Command_Line.Set_Exit_Status (1);
return;
end if;
loop
Put_Line ("Waiting..");
Num_FDs := Epoll.Wait (EpollFD, First_Event'Access, 10, -1);
if Num_FDs = -1 then
Put_Line ("Failure on Epoll.Wait, exiting");
Ada.Command_Line.Set_Exit_Status (1);
return;
end if;
for I in 0 .. Num_FDs loop
Put_Line ("Activity on" & Num_FDs'Img & " sockets");
declare
Index : constant Integer := Integer (I);
Polled_Event : Epoll.Event_Type := Events (Index);
ClientSock : GNAT.Sockets.Socket_Type;
Socket_Request : GNAT.Sockets.Request_Type :=
GNAT.Sockets.Request_Type'(Name => Non_Blocking_IO,
Enabled => True);
begin
Put_Line ("Socket with activity" & Image (Polled_Event.Data.FD));
if Polled_Event.Data.FD = ServerSock then
Put_Line ("Polled_Event is new connection, let's accept");
Accept_Socket (ServerSock, ClientSock, ServerAddr);
Control_Socket (Socket => ClientSock,
Request => Socket_Request);
Event.Events := Epoll.Epoll_In_And_Et;
Event.Data.FD := ClientSock;
Put_Line ("Socket opening up: " & Image (ClientSock));
Retval := Epoll.Control (EpollFD,
Epoll.Epoll_Ctl_Add,
To_C (ClientSock),
Event'Access);
if Retval = -1 then
Put_Line ("Failed to add accepted socket to epollfd");
Ada.Command_Line.Set_Exit_Status (1);
return;
end if;
else
Put ("Received data on socket: ");
Put (Image (Polled_Event.Data.FD));
Put (To_C (Polled_Event.Data.FD)'Img);
New_Line;
Read_Data (Polled_Event.Data.FD);
end if;
end;
end loop;
end loop;
end Main;