WebSVN – Kolibri OS – Diff – /programs/network/downloader/downloader.asm


;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;;                                                                 ;;
;; Copyright (C) KolibriOS team 2014-2017. All rights reserved.    ;;
;; Distributed under terms of the GNU General Public License       ;;
;;                                                                 ;;
;;  downloader.asm - HTTP client for KolibriOS                     ;;
;;                                                                 ;;
;;      hidnplayr@kolibrios.org                                    ;;
;;                                                                 ;;
;;          GNU GENERAL PUBLIC LICENSE                             ;;
;;             Version 2, June 1991                                ;;
;;                                                                 ;;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 
URLMAXLEN       = 65535
 
__DEBUG__       = 1
__DEBUG_LEVEL__ = 1
 
 
format binary as ""
use32
        org     0x0
 
        db      'MENUET01'      ; header
        dd      0x01            ; header version
        dd      START           ; entry point
        dd      IM_END          ; image size
        dd      I_END+0x1000    ; required memory
        dd      I_END+0x1000    ; esp
        dd      url
        dd      0x0             ; I_Path
 
 
include '../../macros.inc'
include '../../proc32.inc'
include '../../dll.inc'
include '../../debug-fdo.inc'
include '../../develop/libraries/box_lib/trunk/box_lib.mac'
include '../../develop/libraries/http/http.inc'
 
virtual at 0
        http_msg http_msg
end virtual
 
 
START:
        mcall   68, 11                  ; init heap so we can allocate memory dynamically
 
; load libraries
        stdcall dll.Load, @IMPORT
        test    eax, eax
        jnz     mainloop.exit
 
; wanted events
        mcall   40, EVM_REDRAW + EVM_KEY + EVM_BUTTON + EVM_MOUSE + EVM_MOUSE_FILTER
 
; prepare filename buffers
        mov     edi, fname_buf
        mov     esi, download_file_path
  @@:
        lodsb
        stosb
        test    al, al
        jnz     @r
 
; Initialise OpenDialog
        invoke  OpenDialog_Init, OpenDialog_data
 
; If user provided parameters, start download right away!
        cmp     byte[url], 0
        jne     display_url_and_download
 
        mov     [OpenDialog_data.draw_window], draw_window
 
redraw:
        call    draw_window
 
mainloop:
        mcall   10      ; wait here for event
        cmp     eax, EV_REDRAW
        je      redraw
        cmp     eax, EV_KEY
        je      .key
        cmp     eax, EV_BUTTON
        je      .button
        cmp     eax, EV_MOUSE
        je      .mouse
        jmp     mainloop
 
  .key:
        mcall   2       ; read key
        invoke  edit_box_key, edit1
        cmp     ax, 13 shl 8
        je      download
        jmp     mainloop
 
  .button:
        mcall   17      ; get id
        cmp     ah, 1   ; button id=1 ?
        je      .exit
 
        cmp     [btn_text], sz_download
        je      download
 
        cmp     [btn_text], sz_open
        je      open_file
 
  .exit:
        mcall   -1      ; exit
 
  .mouse:
        invoke  edit_box_mouse, edit1
        jmp     mainloop
 
 
open_file:
        mcall   70, fileopen
        jmp     mainloop
 
display_url_and_download:
        xor     al, al
        mov     ecx, 4096
        mov     edi, url
        repne scasb
        sub     edi, url+1
        mov     [edit1.size], edi
 
download:
; Extract the filename from URL
        mov     edi, url
        xor     al, al
        mov     ecx, URLMAXLEN
        repne scasb
        mov     esi, edi
        dec     esi
        dec     esi
        std
  .loop:
        lodsb
        cmp     al, '/'
        je      .done
        test    al, al
        jnz     .loop
  .done:
        cld
        mov     ecx, edi
        sub     ecx, esi
        inc     esi
        inc     esi
        mov     edi, filename_area
        rep movsb
 
; invoke OpenDialog
        invoke  OpenDialog_Start, OpenDialog_data
        mcall   40, EVM_REDRAW + EVM_BUTTON + EVM_STACK
        call    draw_window
 
; Create the local file
        mov     [fileinfo], 2           ; create/write to file
        xor     eax, eax
        mov     [fileinfo.offset], eax
        mov     [fileinfo.offset+4], eax
        mov     [fileinfo.size], eax
        mcall   70, fileinfo
        test    eax, eax
        jnz     create_error
 
; Start the download
        invoke  HTTP_get, url, 0, FLAG_STREAM or FLAG_REUSE_BUFFER, 0
        test    eax, eax
        jz      get_error
 
        mov     [identifier], eax
        mov     [offset], 0
        mov     [btn_text], sz_cancel
        mov     [status], sz_downloading
        or      [edit1.flags], ed_figure_only
        and     [edit1.flags], not ed_focus
        push    [sc.work]
        pop     [edit1.color]
        call    draw_window
 
        jmp     download_loop
 
get_error:
        mov     [btn_text], sz_exit
        mov     [status], sz_err_http
        jmp     redraw
 
create_error:
        mov     [btn_text], sz_exit
        mov     [status], sz_err_create
        jmp     redraw
 
download_loop:
        mcall   10
        cmp     eax, EV_REDRAW
        je      .redraw
        cmp     eax, EV_BUTTON
        je      .button
 
        invoke  HTTP_receive, [identifier]
        test    eax, eax
        jz      got_data
        jmp     download_loop
 
  .redraw:
        call    draw_window
        jmp     download_loop
 
  .button:
        jmp     http_free
 
got_data:
        mov     ebp, [identifier]
        test    [ebp + http_msg.flags], 0xffff0000      ; error?
        jnz     http_error
 
        cmp     [fileinfo], 3                           ; Did we write before?
        je      .write
 
        test    [ebp + http_msg.flags], FLAG_CONTENT_LENGTH
        jz      .first_write
 
        mov     eax, [ebp + http_msg.content_length]
        mov     [pb.max], eax
 
        DEBUGF  1, "new file size=%u\n", eax
        mov     [fileinfo], 4                           ; set end of file
        mov     [fileinfo.offset], eax                  ; new file size
        mcall   70, fileinfo
        test    eax, eax
        jnz     write_error
 
 
  .first_write:
        mov     [fileinfo], 3                           ; write to existing file
  .write:
        mov     ecx, [ebp + http_msg.content_received]
        sub     ecx, [offset]
        jz      .no_data                                ; more then 0 data bytes?
 
        mov     [fileinfo.size], ecx
        mov     eax, [ebp + http_msg.content_ptr]
        mov     [fileinfo.buffer], eax
        mov     ebx, [offset]
        mov     [fileinfo.offset], ebx
        DEBUGF  1, "Writing to disk: size=%u offset=%u\n", ecx, ebx
        mcall   70, fileinfo
        test    eax, eax                                ; check error code
        jnz     write_error
        cmp     ebx, ecx                                ; check if all bytes were written to disk
        jne     write_error
 
        mov     eax, [ebp + http_msg.content_received]
        mov     [offset], eax
        mov     [pb.value], eax
 
        invoke  progressbar_draw, pb
 
  .no_data:
        test    [ebp + http_msg.flags], FLAG_GOT_ALL_DATA
        jz      download_loop
 
; Download completed successfully
        mov     [status], sz_complete
        mov     [pb.progress_color], 0x0000c800         ; green
        mov     [btn_text], sz_open
        jmp     http_free
 
write_error:
        mov     [status], sz_err_full
        mov     [pb.progress_color], 0x00c80000         ; red
        mov     [btn_text], sz_exit
        jmp     http_free
 
http_error:
        mov     [status], sz_err_http
        mov     [pb.progress_color], 0x00c80000         ; red
        mov     [btn_text], sz_exit
;        jmp     http_free
 
http_free:
        mcall   40, EVM_REDRAW + EVM_BUTTON
        push    [ebp + http_msg.content_received]
        pop     [pb.value]
 
        mov     ecx, [ebp + http_msg.content_ptr]
        test    ecx, ecx
        jz      @f
        mcall   68, 13                                  ; free the buffer
  @@:
        invoke  HTTP_free, [identifier]                 ; free headers and connection
        jmp     redraw
 
draw_window:
        mcall   12, 1   ; start window draw
 
; get system colors
        mcall   48, 3, sc, 40
 
; draw window
        mov     edx, [sc.work]
        or      edx, 0x34000000
        mcall   0, <50, 420>, <350, 120>, , 0, title
 
; draw button
        mcall   8, <320,75>, <50,24>, 22, [sc.work_button]
 
; draw button text
        mov     ecx, [sc.work_button_text]
        or      ecx, 90000000h
        mcall   4, <325,56>, , [btn_text]
 
; draw status text
        mov     ecx, [sc.work_text]
        or      ecx, 90000000h
        mcall   4, <10,70>, , [status]
 
; draw editbox
        edit_boxes_set_sys_color edit1, editboxes_end, sc
        invoke  edit_box_draw, edit1
 
        cmp     [identifier], 0
        je     @f
; draw progressbar
        invoke  progressbar_draw, pb
  @@:
        mcall   12, 2   ; end window draw
 
dont_draw:
        ret
 
;---------------------------------------------------------------------
; Data area
;-----------------------------------------------------------------------------
align   4
@IMPORT:
 
library lib_http,       'http.obj', \
        box_lib,        'box_lib.obj', \
        proc_lib,       'proc_lib.obj'
 
import  lib_http, \
        HTTP_get,       'get', \
        HTTP_receive,   'receive', \
        HTTP_free,      'free'
 
import  box_lib, \
        edit_box_draw,    'edit_box_draw', \
        edit_box_key,     'edit_box_key', \
        edit_box_mouse,   'edit_box_mouse', \
        progressbar_draw, 'progressbar_draw', \
        progressbar_prog, 'progressbar_progress'
 
import  proc_lib, \
        OpenDialog_Init,  'OpenDialog_init', \
        OpenDialog_Start, 'OpenDialog_start'
 
 
fileinfo        dd 2
  .offset       dd 0, 0
  .size         dd 0
  .buffer       dd 0
                db 0
                dd fname_buf
 
fileopen        dd 7
                dd 0                    ; flags
                dd fname_buf            ; parameters
                dd 0                    ; reserved
                dd 0                    ; reserved
                db "/sys/@open", 0      ; path
 
edit1           edit_box 400, 5, 10, 0xffffff, 0x0000ff, 0x0080ff, 0x000000, 0x90000000, URLMAXLEN, url, mouse_dd, ed_focus+ed_always_focus, 0, 0
editboxes_end:
 
identifier      dd 0
btn_text        dd sz_download
status          dd sz_null
sz_download     db 'Download', 0
sz_cancel       db ' Cancel ', 0
sz_open         db '  Open  ', 0
sz_exit         db '  Exit  ', 0
 
sz_null         db 0
sz_downloading  db 'Downloading..', 0
sz_complete     db 'Download completed', 0
sz_err_create   db 'Could not create the local file!', 0
sz_err_full     db 'Disk full!', 0
sz_err_http     db 'HTTP error!', 0
title           db 'HTTP Downloader', 0
 
OpenDialog_data:
.type                   dd 1    ; Save
.procinfo               dd procinfo
.com_area_name          dd communication_area_name
.com_area               dd 0
.opendir_path           dd temp_dir_path
.dir_default_path       dd communication_area_default_path
.start_path             dd open_dialog_path
.draw_window            dd dont_draw
.status                 dd 0
.openfile_patch         dd fname_buf
.filename_area          dd filename_area
.filter_area            dd filter
.x:
.x_size                 dw 420  ; Window X size
.x_start                dw 200  ; Window X position
.y:
.y_size                 dw 320  ; Window y size
.y_start                dw 140  ; Window Y position
 
communication_area_name         db 'FFFFFFFF_open_dialog',0
open_dialog_path                db '/sys/File Managers/opendial',0
communication_area_default_path db '/sys',0
 
filter:
dd      0
db      0
 
pb:
.value          dd 0
.left           dd 5
.top            dd 45
.width          dd 300
.height         dd 16
.style          dd 1
.min            dd 0
.max            dd 0
.back_color     dd 0xefefef
.progress_color dd 0xc8c8c8
.frame_color    dd 0x94aece
.frame_color2   dd 0xffffff
 
include_debug_strings
 
download_file_path db '/tmp0/1/', 0
 
IM_END:
 
url             rb URLMAXLEN
sc              system_colors
offset          dd ?
mouse_dd        dd ?
 
filename_area   rb 256
temp_dir_path   rb 4096
procinfo        rb 1024
fname_buf       rb 4096
text_work_area  rb 1024
 
I_END:

Rev 7092	Rev 9568
1	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;	1	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
2	;; ;;	2	;; ;;
3	;; Copyright (C) KolibriOS team 2014-2017. All rights reserved. ;;	3	;; Copyright (C) KolibriOS team 2014-2017. All rights reserved. ;;
4	;; Distributed under terms of the GNU General Public License ;;	4	;; Distributed under terms of the GNU General Public License ;;
5	;; ;;	5	;; ;;
6	;; downloader.asm - HTTP client for KolibriOS ;;	6	;; downloader.asm - HTTP client for KolibriOS ;;
7	;; ;;	7	;; ;;
8	;; hidnplayr@kolibrios.org ;;	8	;; hidnplayr@kolibrios.org ;;
9	;; ;;	9	;; ;;
10	;; GNU GENERAL PUBLIC LICENSE ;;	10	;; GNU GENERAL PUBLIC LICENSE ;;
11	;; Version 2, June 1991 ;;	11	;; Version 2, June 1991 ;;
12	;; ;;	12	;; ;;
13	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;	13	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
14		14
15	URLMAXLEN = 65535	15	URLMAXLEN = 65535
16		16
17	__DEBUG__ = 1	17	__DEBUG__ = 1
18	__DEBUG_LEVEL__ = 1	18	__DEBUG_LEVEL__ = 1
19		19
20		20
21	format binary as ""	21	format binary as ""
22	use32	22	use32
23	org 0x0	23	org 0x0
24		24
25	db 'MENUET01' ; header	25	db 'MENUET01' ; header
26	dd 0x01 ; header version	26	dd 0x01 ; header version
27	dd START ; entry point	27	dd START ; entry point
28	dd IM_END ; image size	28	dd IM_END ; image size
29	dd I_END+0x1000 ; required memory	29	dd I_END+0x1000 ; required memory
30	dd I_END+0x1000 ; esp	30	dd I_END+0x1000 ; esp
31	dd url	31	dd url
32	dd 0x0 ; I_Path	32	dd 0x0 ; I_Path
33		33
34		34
35	include '../../macros.inc'	35	include '../../macros.inc'
36	include '../../proc32.inc'	36	include '../../proc32.inc'
37	include '../../dll.inc'	37	include '../../dll.inc'
38	include '../../debug-fdo.inc'	38	include '../../debug-fdo.inc'
39	include '../../develop/libraries/box_lib/trunk/box_lib.mac'	39	include '../../develop/libraries/box_lib/trunk/box_lib.mac'
40	include '../../develop/libraries/http/http.inc'	40	include '../../develop/libraries/http/http.inc'
41		41
42	virtual at 0	42	virtual at 0
43	http_msg http_msg	43	http_msg http_msg
44	end virtual	44	end virtual
45		45
46		46
47	START:	47	START:
48	mcall 68, 11 ; init heap so we can allocate memory dynamically	48	mcall 68, 11 ; init heap so we can allocate memory dynamically
49		49
50	; load libraries	50	; load libraries
51	stdcall dll.Load, @IMPORT	51	stdcall dll.Load, @IMPORT
52	test eax, eax	52	test eax, eax
53	jnz mainloop.exit	53	jnz mainloop.exit
54		54
55	; wanted events	55	; wanted events
56	mcall 40, EVM_REDRAW + EVM_KEY + EVM_BUTTON + EVM_MOUSE + EVM_MOUSE_FILTER	56	mcall 40, EVM_REDRAW + EVM_KEY + EVM_BUTTON + EVM_MOUSE + EVM_MOUSE_FILTER
57		57
58	; prepare filename buffers	58	; prepare filename buffers
59	mov edi, fname_buf	59	mov edi, fname_buf
60	mov esi, download_file_path	60	mov esi, download_file_path
61	@@:	61	@@:
62	lodsb	62	lodsb
63	stosb	63	stosb
64	test al, al	64	test al, al
65	jnz @r	65	jnz @r
66		66
67	; Initialise OpenDialog	67	; Initialise OpenDialog
68	invoke OpenDialog_Init, OpenDialog_data	68	invoke OpenDialog_Init, OpenDialog_data
69		69
70	; If user provided parameters, start download right away!	70	; If user provided parameters, start download right away!
71	cmp byte[url], 0	71	cmp byte[url], 0
72	jne display_url_and_download	72	jne display_url_and_download
73		73
74	mov [OpenDialog_data.draw_window], draw_window	74	mov [OpenDialog_data.draw_window], draw_window
75		75
76	redraw:	76	redraw:
77	call draw_window	77	call draw_window
78		78
79	mainloop:	79	mainloop:
80	mcall 10 ; wait here for event	80	mcall 10 ; wait here for event
81	cmp eax, EV_REDRAW	81	cmp eax, EV_REDRAW
82	je redraw	82	je redraw
83	cmp eax, EV_KEY	83	cmp eax, EV_KEY
84	je .key	84	je .key
85	cmp eax, EV_BUTTON	85	cmp eax, EV_BUTTON
86	je .button	86	je .button
87	cmp eax, EV_MOUSE	87	cmp eax, EV_MOUSE
88	je .mouse	88	je .mouse
89	jmp mainloop	89	jmp mainloop
90		90
91	.key:	91	.key:
92	mcall 2 ; read key	92	mcall 2 ; read key
93	invoke edit_box_key, edit1	93	invoke edit_box_key, edit1
94	cmp ax, 13 shl 8	94	cmp ax, 13 shl 8
95	je download	95	je download
96	jmp mainloop	96	jmp mainloop
97		97
98	.button:	98	.button:
99	mcall 17 ; get id	99	mcall 17 ; get id
100	cmp ah, 1 ; button id=1 ?	100	cmp ah, 1 ; button id=1 ?
101	je .exit	101	je .exit
102		102
103	cmp [btn_text], sz_download	103	cmp [btn_text], sz_download
104	je download	104	je download
105		105
106	cmp [btn_text], sz_open	106	cmp [btn_text], sz_open
107	je open_file	107	je open_file
108		108
109	.exit:	109	.exit:
110	mcall -1 ; exit	110	mcall -1 ; exit
111		111
112	.mouse:	112	.mouse:
113	invoke edit_box_mouse, edit1	113	invoke edit_box_mouse, edit1
114	jmp mainloop	114	jmp mainloop
115		115
116		116
117	open_file:	117	open_file:
118	mcall 70, fileopen	118	mcall 70, fileopen
119	jmp mainloop	119	jmp mainloop
120		120
121	display_url_and_download:	121	display_url_and_download:
122	xor al, al	122	xor al, al
123	mov ecx, 4096	123	mov ecx, 4096
124	mov edi, url	124	mov edi, url
125	repne scasb	125	repne scasb
126	sub edi, url+1	126	sub edi, url+1
127	mov [edit1.size], edi	127	mov [edit1.size], edi
128		128
129	download:	129	download:
130	; Extract the filename from URL	130	; Extract the filename from URL
131	mov edi, url	131	mov edi, url
132	xor al, al	132	xor al, al
133	mov ecx, URLMAXLEN	133	mov ecx, URLMAXLEN
134	repne scasb	134	repne scasb
135	mov esi, edi	135	mov esi, edi
136	dec esi	136	dec esi
137	dec esi	137	dec esi
138	std	138	std
139	.loop:	139	.loop:
140	lodsb	140	lodsb
141	cmp al, '/'	141	cmp al, '/'
142	je .done	142	je .done
143	test al, al	143	test al, al
144	jnz .loop	144	jnz .loop
145	.done:	145	.done:
146	cld	146	cld
147	mov ecx, edi	147	mov ecx, edi
148	sub ecx, esi	148	sub ecx, esi
149	inc esi	149	inc esi
150	inc esi	150	inc esi
151	mov edi, filename_area	151	mov edi, filename_area
152	rep movsb	152	rep movsb
153		153
154	; invoke OpenDialog	154	; invoke OpenDialog
155	invoke OpenDialog_Start, OpenDialog_data	155	invoke OpenDialog_Start, OpenDialog_data
156	mcall 40, EVM_REDRAW + EVM_BUTTON + EVM_STACK	156	mcall 40, EVM_REDRAW + EVM_BUTTON + EVM_STACK
157	call draw_window	157	call draw_window
158		158
159	; Create the local file	159	; Create the local file
160	mov [fileinfo], 2 ; create/write to file	160	mov [fileinfo], 2 ; create/write to file
161	xor eax, eax	161	xor eax, eax
162	mov [fileinfo.offset], eax	162	mov [fileinfo.offset], eax
163	mov [fileinfo.offset+4], eax	163	mov [fileinfo.offset+4], eax
164	mov [fileinfo.size], eax	164	mov [fileinfo.size], eax
165	mcall 70, fileinfo	165	mcall 70, fileinfo
166	test eax, eax	166	test eax, eax
167	jnz create_error	167	jnz create_error
168		168
169	; Start the download	169	; Start the download
170	invoke HTTP_get, url, 0, FLAG_STREAM or FLAG_REUSE_BUFFER, 0	170	invoke HTTP_get, url, 0, FLAG_STREAM or FLAG_REUSE_BUFFER, 0
171	test eax, eax	171	test eax, eax
172	jz get_error	172	jz get_error
173		173
174	mov [identifier], eax	174	mov [identifier], eax
175	mov [offset], 0	175	mov [offset], 0
176	mov [btn_text], sz_cancel	176	mov [btn_text], sz_cancel
177	mov [status], sz_downloading	177	mov [status], sz_downloading
178	or [edit1.flags], ed_figure_only	178	or [edit1.flags], ed_figure_only
179	and [edit1.flags], not ed_focus	179	and [edit1.flags], not ed_focus
180	push [sc.work]	180	push [sc.work]
181	pop [edit1.color]	181	pop [edit1.color]
182	call draw_window	182	call draw_window
183		183
184	jmp download_loop	184	jmp download_loop
185		185
186	get_error:	186	get_error:
187	mov [btn_text], sz_exit	187	mov [btn_text], sz_exit
188	mov [status], sz_err_http	188	mov [status], sz_err_http
189	jmp redraw	189	jmp redraw
190		190
191	create_error:	191	create_error:
192	mov [btn_text], sz_exit	192	mov [btn_text], sz_exit
193	mov [status], sz_err_create	193	mov [status], sz_err_create
194	jmp redraw	194	jmp redraw
195		195
196	download_loop:	196	download_loop:
197	mcall 10	197	mcall 10
198	cmp eax, EV_REDRAW	198	cmp eax, EV_REDRAW
199	je .redraw	199	je .redraw
200	cmp eax, EV_BUTTON	200	cmp eax, EV_BUTTON
201	je .button	201	je .button
202		202
203	invoke HTTP_receive, [identifier]	203	invoke HTTP_receive, [identifier]
204	test eax, eax	204	test eax, eax
205	jz got_data	205	jz got_data
206	jmp download_loop	206	jmp download_loop
207		207
208	.redraw:	208	.redraw:
209	call draw_window	209	call draw_window
210	jmp download_loop	210	jmp download_loop
211		211
212	.button:	212	.button:
213	jmp http_free	213	jmp http_free
214		214
215	got_data:	215	got_data:
216	mov ebp, [identifier]	216	mov ebp, [identifier]
217	test [ebp + http_msg.flags], 0xffff0000 ; error?	217	test [ebp + http_msg.flags], 0xffff0000 ; error?
218	jnz http_error	218	jnz http_error
219		219
220	cmp [fileinfo], 3 ; Did we write before?	220	cmp [fileinfo], 3 ; Did we write before?
221	je .write	221	je .write
222		222
223	test [ebp + http_msg.flags], FLAG_CONTENT_LENGTH	223	test [ebp + http_msg.flags], FLAG_CONTENT_LENGTH
224	jz .first_write	224	jz .first_write
225		225
226	mov eax, [ebp + http_msg.content_length]	226	mov eax, [ebp + http_msg.content_length]
227	mov [pb.max], eax	227	mov [pb.max], eax
228		228
229	DEBUGF 1, "new file size=%u\n", eax	229	DEBUGF 1, "new file size=%u\n", eax
230	mov [fileinfo], 4 ; set end of file	230	mov [fileinfo], 4 ; set end of file
231	mov [fileinfo.offset], eax ; new file size	231	mov [fileinfo.offset], eax ; new file size
232	mcall 70, fileinfo	232	mcall 70, fileinfo
233	test eax, eax	233	test eax, eax
234	jnz write_error	234	jnz write_error
235		235
236		236
237	.first_write:	237	.first_write:
238	mov [fileinfo], 3 ; write to existing file	238	mov [fileinfo], 3 ; write to existing file
239	.write:	239	.write:
240	mov ecx, [ebp + http_msg.content_received]	240	mov ecx, [ebp + http_msg.content_received]
241	sub ecx, [offset]	241	sub ecx, [offset]
242	jz .no_data ; more then 0 data bytes?	242	jz .no_data ; more then 0 data bytes?
243		243
244	mov [fileinfo.size], ecx	244	mov [fileinfo.size], ecx
245	mov eax, [ebp + http_msg.content_ptr]	245	mov eax, [ebp + http_msg.content_ptr]
246	mov [fileinfo.buffer], eax	246	mov [fileinfo.buffer], eax
247	mov ebx, [offset]	247	mov ebx, [offset]
248	mov [fileinfo.offset], ebx	248	mov [fileinfo.offset], ebx
249	DEBUGF 1, "Writing to disk: size=%u offset=%u\n", ecx, ebx	249	DEBUGF 1, "Writing to disk: size=%u offset=%u\n", ecx, ebx
250	mcall 70, fileinfo	250	mcall 70, fileinfo
251	test eax, eax ; check error code	251	test eax, eax ; check error code
252	jnz write_error	252	jnz write_error
253	cmp ebx, ecx ; check if all bytes were written to disk	253	cmp ebx, ecx ; check if all bytes were written to disk
254	jne write_error	254	jne write_error
255		255
256	mov eax, [ebp + http_msg.content_received]	256	mov eax, [ebp + http_msg.content_received]
257	mov [offset], eax	257	mov [offset], eax
258	mov [pb.value], eax	258	mov [pb.value], eax
259		259
260	invoke progressbar_draw, pb	260	invoke progressbar_draw, pb
261		261
262	.no_data:	262	.no_data:
263	test [ebp + http_msg.flags], FLAG_GOT_ALL_DATA	263	test [ebp + http_msg.flags], FLAG_GOT_ALL_DATA
264	jz download_loop	264	jz download_loop
265		265
266	; Download completed successfully	266	; Download completed successfully
267	mov [status], sz_complete	267	mov [status], sz_complete
268	mov [pb.progress_color], 0x0000c800 ; green	268	mov [pb.progress_color], 0x0000c800 ; green
269	mov [btn_text], sz_open	269	mov [btn_text], sz_open
270	jmp http_free	270	jmp http_free
271		271
272	write_error:	272	write_error:
273	mov [status], sz_err_full	273	mov [status], sz_err_full
274	mov [pb.progress_color], 0x00c80000 ; red	274	mov [pb.progress_color], 0x00c80000 ; red
275	mov [btn_text], sz_exit	275	mov [btn_text], sz_exit
276	jmp http_free	276	jmp http_free
277		277
278	http_error:	278	http_error:
279	mov [status], sz_err_http	279	mov [status], sz_err_http
280	mov [pb.progress_color], 0x00c80000 ; red	280	mov [pb.progress_color], 0x00c80000 ; red
281	mov [btn_text], sz_exit	281	mov [btn_text], sz_exit
282	; jmp http_free	282	; jmp http_free
283		283
284	http_free:	284	http_free:
285	mcall 40, EVM_REDRAW + EVM_BUTTON	285	mcall 40, EVM_REDRAW + EVM_BUTTON
286	push [ebp + http_msg.content_received]	286	push [ebp + http_msg.content_received]
287	pop [pb.value]	287	pop [pb.value]
288		288
289	mov ecx, [ebp + http_msg.content_ptr]	289	mov ecx, [ebp + http_msg.content_ptr]
290	test ecx, ecx	290	test ecx, ecx
291	jz @f	291	jz @f
292	mcall 68, 13 ; free the buffer	292	mcall 68, 13 ; free the buffer
293	@@:	293	@@:
294	invoke HTTP_free, [identifier] ; free headers and connection	294	invoke HTTP_free, [identifier] ; free headers and connection
295	jmp redraw	295	jmp redraw
296		296
297	draw_window:	297	draw_window:
298	mcall 12, 1 ; start window draw	298	mcall 12, 1 ; start window draw
299		299
300	; get system colors	300	; get system colors
301	mcall 48, 3, sc, 40	301	mcall 48, 3, sc, 40
302		302
303	; draw window	303	; draw window
304	mov edx, [sc.work]	304	mov edx, [sc.work]
305	or edx, 0x34000000	305	or edx, 0x34000000
306	mcall 0, <50, 420>, <350, 120>, , 0, title	306	mcall 0, <50, 420>, <350, 120>, , 0, title
307		307
308	; draw button	308	; draw button
309	mcall 8, <320,75>, <50,24>, 22, [sc.work_button]	309	mcall 8, <320,75>, <50,24>, 22, [sc.work_button]
310		310
311	; draw button text	311	; draw button text
312	mov ecx, [sc.work_button_text]	312	mov ecx, [sc.work_button_text]
313	or ecx, 90000000h	313	or ecx, 90000000h
314	mcall 4, <325,56>, , [btn_text]	314	mcall 4, <325,56>, , [btn_text]
315		315
316	; draw status text	316	; draw status text
317	mov ecx, [sc.work_text]	317	mov ecx, [sc.work_text]
318	or ecx, 90000000h	318	or ecx, 90000000h
319	mcall 4, <10,70>, , [status]	319	mcall 4, <10,70>, , [status]
320		320
321	; draw editbox	321	; draw editbox
322	edit_boxes_set_sys_color edit1, editboxes_end, sc	322	edit_boxes_set_sys_color edit1, editboxes_end, sc
323	invoke edit_box_draw, edit1	323	invoke edit_box_draw, edit1
324		324
325	cmp [identifier], 0	325	cmp [identifier], 0
326	je @f	326	je @f
327	; draw progressbar	327	; draw progressbar
328	invoke progressbar_draw, pb	328	invoke progressbar_draw, pb
329	@@:	329	@@:
330	mcall 12, 2 ; end window draw	330	mcall 12, 2 ; end window draw
331		331
332	dont_draw:	332	dont_draw:
333	ret	333	ret
334		334
335	;---------------------------------------------------------------------	335	;---------------------------------------------------------------------
336	; Data area	336	; Data area
337	;-----------------------------------------------------------------------------	337	;-----------------------------------------------------------------------------
338	align 4	338	align 4
339	@IMPORT:	339	@IMPORT:
340		340
341	library lib_http, 'http.obj', \	341	library lib_http, 'http.obj', \
342	box_lib, 'box_lib.obj', \	342	box_lib, 'box_lib.obj', \
343	proc_lib, 'proc_lib.obj'	343	proc_lib, 'proc_lib.obj'
344		344
345	import lib_http, \	345	import lib_http, \
346	HTTP_get, 'get', \	346	HTTP_get, 'get', \
347	HTTP_receive, 'receive', \	347	HTTP_receive, 'receive', \
348	HTTP_free, 'free'	348	HTTP_free, 'free'
349		349
350	import box_lib, \	350	import box_lib, \
351	edit_box_draw, 'edit_box', \	351	edit_box_draw, 'edit_box_draw', \
352	edit_box_key, 'edit_box_key', \	352	edit_box_key, 'edit_box_key', \
353	edit_box_mouse, 'edit_box_mouse', \	353	edit_box_mouse, 'edit_box_mouse', \
354	progressbar_draw, 'progressbar_draw', \	354	progressbar_draw, 'progressbar_draw', \
355	progressbar_prog, 'progressbar_progress'	355	progressbar_prog, 'progressbar_progress'
356		356
357	import proc_lib, \	357	import proc_lib, \
358	OpenDialog_Init, 'OpenDialog_init', \	358	OpenDialog_Init, 'OpenDialog_init', \
359	OpenDialog_Start, 'OpenDialog_start'	359	OpenDialog_Start, 'OpenDialog_start'
360		360
361		361
362	fileinfo dd 2	362	fileinfo dd 2
363	.offset dd 0, 0	363	.offset dd 0, 0
364	.size dd 0	364	.size dd 0
365	.buffer dd 0	365	.buffer dd 0
366	db 0	366	db 0
367	dd fname_buf	367	dd fname_buf
368		368
369	fileopen dd 7	369	fileopen dd 7
370	dd 0 ; flags	370	dd 0 ; flags
371	dd fname_buf ; parameters	371	dd fname_buf ; parameters
372	dd 0 ; reserved	372	dd 0 ; reserved
373	dd 0 ; reserved	373	dd 0 ; reserved
374	db "/sys/@open", 0 ; path	374	db "/sys/@open", 0 ; path
375		375
376	edit1 edit_box 400, 5, 10, 0xffffff, 0x0000ff, 0x0080ff, 0x000000, 0x90000000, URLMAXLEN, url, mouse_dd, ed_focus+ed_always_focus, 0, 0	376	edit1 edit_box 400, 5, 10, 0xffffff, 0x0000ff, 0x0080ff, 0x000000, 0x90000000, URLMAXLEN, url, mouse_dd, ed_focus+ed_always_focus, 0, 0
377	editboxes_end:	377	editboxes_end:
378		378
379	identifier dd 0	379	identifier dd 0
380	btn_text dd sz_download	380	btn_text dd sz_download
381	status dd sz_null	381	status dd sz_null
382	sz_download db 'Download', 0	382	sz_download db 'Download', 0
383	sz_cancel db ' Cancel ', 0	383	sz_cancel db ' Cancel ', 0
384	sz_open db ' Open ', 0	384	sz_open db ' Open ', 0
385	sz_exit db ' Exit ', 0	385	sz_exit db ' Exit ', 0
386		386
387	sz_null db 0	387	sz_null db 0
388	sz_downloading db 'Downloading..', 0	388	sz_downloading db 'Downloading..', 0
389	sz_complete db 'Download completed', 0	389	sz_complete db 'Download completed', 0
390	sz_err_create db 'Could not create the local file!', 0	390	sz_err_create db 'Could not create the local file!', 0
391	sz_err_full db 'Disk full!', 0	391	sz_err_full db 'Disk full!', 0
392	sz_err_http db 'HTTP error!', 0	392	sz_err_http db 'HTTP error!', 0
393	title db 'HTTP Downloader', 0	393	title db 'HTTP Downloader', 0
394		394
395	OpenDialog_data:	395	OpenDialog_data:
396	.type dd 1 ; Save	396	.type dd 1 ; Save
397	.procinfo dd procinfo	397	.procinfo dd procinfo
398	.com_area_name dd communication_area_name	398	.com_area_name dd communication_area_name
399	.com_area dd 0	399	.com_area dd 0
400	.opendir_path dd temp_dir_path	400	.opendir_path dd temp_dir_path
401	.dir_default_path dd communication_area_default_path	401	.dir_default_path dd communication_area_default_path
402	.start_path dd open_dialog_path	402	.start_path dd open_dialog_path
403	.draw_window dd dont_draw	403	.draw_window dd dont_draw
404	.status dd 0	404	.status dd 0
405	.openfile_patch dd fname_buf	405	.openfile_patch dd fname_buf
406	.filename_area dd filename_area	406	.filename_area dd filename_area
407	.filter_area dd filter	407	.filter_area dd filter
408	.x:	408	.x:
409	.x_size dw 420 ; Window X size	409	.x_size dw 420 ; Window X size
410	.x_start dw 200 ; Window X position	410	.x_start dw 200 ; Window X position
411	.y:	411	.y:
412	.y_size dw 320 ; Window y size	412	.y_size dw 320 ; Window y size
413	.y_start dw 140 ; Window Y position	413	.y_start dw 140 ; Window Y position
414		414
415	communication_area_name db 'FFFFFFFF_open_dialog',0	415	communication_area_name db 'FFFFFFFF_open_dialog',0
416	open_dialog_path db '/sys/File Managers/opendial',0	416	open_dialog_path db '/sys/File Managers/opendial',0
417	communication_area_default_path db '/sys',0	417	communication_area_default_path db '/sys',0
418		418
419	filter:	419	filter:
420	dd 0	420	dd 0
421	db 0	421	db 0
422		422
423	pb:	423	pb:
424	.value dd 0	424	.value dd 0
425	.left dd 5	425	.left dd 5
426	.top dd 45	426	.top dd 45
427	.width dd 300	427	.width dd 300
428	.height dd 16	428	.height dd 16
429	.style dd 1	429	.style dd 1
430	.min dd 0	430	.min dd 0
431	.max dd 0	431	.max dd 0
432	.back_color dd 0xefefef	432	.back_color dd 0xefefef
433	.progress_color dd 0xc8c8c8	433	.progress_color dd 0xc8c8c8
434	.frame_color dd 0x94aece	434	.frame_color dd 0x94aece
435	.frame_color2 dd 0xffffff	435	.frame_color2 dd 0xffffff
436		436
437	include_debug_strings	437	include_debug_strings
438		438
439	download_file_path db '/tmp0/1/', 0	439	download_file_path db '/tmp0/1/', 0
440		440
441	IM_END:	441	IM_END:
442		442
443	url rb URLMAXLEN	443	url rb URLMAXLEN
444	sc system_colors	444	sc system_colors
445	offset dd ?	445	offset dd ?
446	mouse_dd dd ?	446	mouse_dd dd ?
447		447
448	filename_area rb 256	448	filename_area rb 256
449	temp_dir_path rb 4096	449	temp_dir_path rb 4096
450	procinfo rb 1024	450	procinfo rb 1024
451	fname_buf rb 4096	451	fname_buf rb 4096
452	text_work_area rb 1024	452	text_work_area rb 1024
453		453
454	I_END:	454	I_END:

Subversion Repositories Kolibri OS

(root)/programs/network/downloader/downloader.asm @ 6977 – Rev 7092 → 9568