1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
------------------------------------------------------------------------------
-- Ada Web Server --
-- --
-- Copyright (C) 2004-2006 --
-- AdaCore --
-- --
-- This library is free software; you can redistribute it and/or modify --
-- it under the terms of the GNU General Public License as published by --
-- the Free Software Foundation; either version 2 of the License, or (at --
-- your option) any later version. --
-- --
-- This library is distributed in the hope that it will be useful, but --
-- WITHOUT ANY WARRANTY; without even the implied warranty of --
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU --
-- General Public License for more details. --
-- --
-- You should have received a copy of the GNU General Public License --
-- along with this library; if not, write to the Free Software Foundation, --
-- Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. --
-- --
------------------------------------------------------------------------------
-- ~ MAIN [SSL]
with Ada.Exceptions;
with Ada.Streams;
with Ada.Text_IO;
with AWS.Net.SSL;
with Get_Free_Port;
procedure Plain2SSL is
use AWS.Net;
use Ada.Streams;
use Ada.Text_IO;
Client, Server, Peer : Socket_Type'Class := Socket (False);
Port : Positive := 3456;
Sample : Stream_Element_Array (1 .. 1000);
procedure Test (Source, Target : Socket_Type'Class);
----------
-- Test --
----------
procedure Test (Source, Target : Socket_Type'Class) is
Buffer : Stream_Element_Array (Sample'Range);
First : Stream_Element_Offset := Buffer'First;
Last : Stream_Element_Offset;
task Send_Task;
task body Send_Task is
begin
-- We have to send in task for SSL negotiation
Send (Source, Sample);
end Send_Task;
begin
loop
Receive (Target, Buffer (First .. Buffer'Last), Last);
exit when Last = Buffer'Last;
First := Last + 1;
end loop;
if Buffer = Sample then
Put_Line ("Ok.");
else
Put_Line ("Error.");
end if;
end Test;
begin
for J in Sample'Range loop
Sample (J) := Stream_Element
(J mod (Stream_Element_Offset (Stream_Element'Last)));
end loop;
Get_Free_Port (Port);
Bind (Server, Port);
Listen (Server);
Connect (Client, "localhost", Port);
Accept_Socket (Server, Peer);
Set_Timeout (Client, 2.0);
Set_Timeout (Peer, 2.0);
Test (Client, Peer);
Test (Peer, Client);
declare
-- Convert to SSL connection
SSL_Client : SSL.Socket_Type := SSL.Secure_Client (Client);
SSL_Peer : SSL.Socket_Type := SSL.Secure_Server (Peer);
begin
-- Between SSL
Test (SSL_Client, SSL_Peer);
Test (SSL_Peer, SSL_Client);
-- Provoke data errors
Test (SSL_Client, Peer);
Test (SSL_Peer, Client);
-- Provoke SSL errors
begin
Test (Client, SSL_Peer);
exception
when E : Socket_Error =>
Put_Line (Ada.Exceptions.Exception_Message (E));
end;
SSL.Shutdown (SSL_Client);
SSL.Shutdown (SSL_Peer);
end;
end Plain2SSL;
|
