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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151 | ------------------------------------------------------------------------------
-- Ada Web Server --
-- --
-- Copyright (C) 2011-2012, AdaCore --
-- --
-- This is free software; you can redistribute it and/or modify it --
-- under terms of the GNU General Public License as published by the --
-- Free Software Foundation; either version 3, or (at your option) any --
-- later version. This software 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 --
-- distributed with this software; see file COPYING3. If not, go --
-- to http://www.gnu.org/licenses for a complete copy of the license. --
------------------------------------------------------------------------------
-- Compile this application, then run it with:
--
-- $ ./download_manager [<Mb>]
--
-- <Mb> is the size in Mb of the file to download, default to 100Mb
--
with Ada.Command_Line;
with Ada.Numerics.Discrete_Random;
with Ada.Text_IO;
with Ada.Streams.Stream_IO;
with GNAT.MD5;
with AWS.Config.Set;
with AWS.Messages;
with AWS.MIME;
with AWS.Resources.Streams.Disk;
with AWS.Response;
with AWS.Server;
with AWS.Services.Download;
with AWS.Services.Dispatchers.Linker;
with AWS.Services.Dispatchers.URI;
with AWS.Status;
with AWS.Translator;
procedure Download_Manager is
use Ada;
use Ada.Streams;
use Ada.Text_IO;
use AWS;
function CB (Request : Status.Data) return Response.Data;
Filename : constant String := "dm_file.data";
Size : Positive := 100;
procedure Create_Filename;
-- Generate file content and print the MD5 signature, this can be used to
-- check if the file has properly been downloaded.
--------
-- CB --
--------
function CB (Request : Status.Data) return Response.Data is
URI : constant String := Status.URI (Request);
Stream : Resources.Streams.Stream_Access;
begin
if URI = "/" then
return Response.File (MIME.Text_HTML, "main.html");
elsif URI = "/download_file" then
Stream := new Resources.Streams.Disk.Stream_Type;
Resources.Streams.Disk.Open
(Resources.Streams.Disk.Stream_Type (Stream.all), Filename);
return Services.Download.Build (Request, Filename, Stream);
else
return Response.Acknowledge (Messages.S404, "Not found");
end if;
end CB;
---------------------
-- Create_Filename --
---------------------
procedure Create_Filename is
use GNAT;
subtype Buffer is String (1 .. 1_024);
package Char_Random is new Numerics.Discrete_Random (Character);
use Char_Random;
Rand_Generator : Char_Random.Generator;
Ctx : MD5.Context;
File : Stream_IO.File_Type;
Str : Buffer;
begin
Reset (Rand_Generator);
Stream_IO.Create (File, Stream_IO.Out_File, Filename);
for K in 1 .. Size * 1_000 loop
for K in Buffer'Range loop
Str (K) := Random (Rand_Generator);
end loop;
MD5.Update (Ctx, Str);
Stream_IO.Write (File, Translator.To_Stream_Element_Array (Str));
end loop;
Stream_IO.Close (File);
Text_IO.Put_Line ("File MD5 is " & MD5.Digest (Ctx));
end Create_Filename;
URI : Services.Dispatchers.URI.Handler;
Handler : Services.Dispatchers.Linker.Handler;
Conf : Config.Object := Config.Get_Current;
WS : Server.HTTP;
begin
if Command_Line.Argument_Count = 1 then
Size := Positive'Value (Command_Line.Argument (1));
end if;
Create_Filename;
Config.Set.Server_Port (Conf, 8080);
Config.Set.Reuse_Address (Conf, True);
Services.Dispatchers.URI.Register
(URI, "/", CB'Unrestricted_Access);
Services.Dispatchers.URI.Register
(URI, "/download_file", CB'Unrestricted_Access);
Text_IO.Put_Line ("Start download server...");
Services.Download.Start (URI, Handler, 1);
Text_IO.Put_Line ("Start main server...");
Server.Start (WS, Handler, Conf);
Text_IO.Put_Line ("Press Q to quit...");
Server.Wait (Server.Q_Key_Pressed);
Server.Shutdown (WS);
Services.Download.Stop;
end Download_Manager;
|