kolibrios-fun/programs/network/pasta/pasta.asm
hidnplayr 7e3fbc4991 Pasta: fixed bug when there is no data on clipboard, use keep-alive connection.
git-svn-id: svn://kolibrios.org@5770 a494cfbc-eb01-0410-851d-a64ba20cac60
2015-08-24 13:46:18 +00:00

283 lines
7.9 KiB
NASM

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; ;;
;; Copyright (C) KolibriOS team 2014-2015. All rights reserved. ;;
;; Distributed under terms of the GNU General Public License ;;
;; ;;
;; pasta.asm - Paste text to paste.kolibrios.org from a file or ;;
;; from clipboard. ;;
;; ;;
;; GNU GENERAL PUBLIC LICENSE ;;
;; Version 2, June 1991 ;;
;; ;;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
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 param
dd 0 ; I_Path
include '../../macros.inc'
include '../../proc32.inc'
include '../../dll.inc'
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 error
; Got parameters?
cmp byte[param], 0
je clipboard
; Yep, try to read the file.
mcall 68, 12, 32768
test eax, eax
jz error
mov [file_struct.buf], eax
mov [clipboard_data], eax
mcall 70, file_struct
cmp eax, 6
jne error_free_clip
mov [clipboard_data_length], ebx
mov eax, [clipboard_data]
jmp escape
clipboard:
; Get number of slots on the clipboard
mcall 54, 0
cmp eax, -1
je error
; Get last item on clipboard
mov ecx, eax
dec ecx
inc ebx
mcall 54
cmp eax, -1
je error
cmp eax, 1
je error
; Verify if we can work with it
mov [clipboard_data], eax
cmp dword[eax + 4], 0 ; text ?
jne error_free_clip
; Save length in [clipboard_data_length]
mov ecx, dword[eax]
sub ecx, 12
mov [clipboard_data_length], ecx
; Skip clipboard containter params for escape proc
add eax, 12
escape:
; Escape all characters that need escaping
invoke HTTP_escape, eax
test eax, eax
jz error_free_clip
mov [clipboard_data_length], ebx
push eax
mcall 68, 13, [clipboard_data]
pop [clipboard_data]
; Connect to the server
invoke HTTP_get, sz_url, 0, FLAG_BLOCK or FLAG_KEEPALIVE, 0
test eax, eax
jz error_free_clip
mov [identifier], eax
.again:
invoke HTTP_receive, [identifier]
test eax, eax
jnz .again
invoke HTTP_find_header_field, [identifier], sz_set_cookie
mov edi, cookie
test eax, eax
jz .no_cookie
mov esi, eax
.cookie_loop:
lodsb
stosb
cmp al, 13
je .cookie_end
cmp al, 10
je .cookie_end
cmp al, 0
je .cookie_end
jmp .cookie_loop
.cookie_end:
dec edi
.no_cookie:
mov ax, 0x0a0d
stosw
xor al, al
stosb
mov ecx, [clipboard_data_length]
add ecx, sz_paste_head.length + sz_paste_tail.length
invoke HTTP_post, sz_url, [identifier], FLAG_BLOCK, sz_cookie, sz_ctype, ecx
test eax, eax
jz error_free_clip_disconnect
; Send the data to the server
mov ecx, [eax + http_msg.socket]
mcall 75, 6, , sz_paste_head, sz_paste_head.length, 0
mcall 75, 6, , [clipboard_data], [clipboard_data_length], 0
mcall 75, 6, , sz_paste_tail, sz_paste_tail.length, 0
; Free the data
mcall 68, 13, [clipboard_data]
.again2:
invoke HTTP_receive, [identifier]
test eax, eax
jnz .again2
invoke HTTP_disconnect, [identifier]
mov ebp, [identifier]
cmp [ebp + http_msg.status], 302 ; found
jne error_free_http
invoke HTTP_find_header_field, [identifier], sz_location
test eax, eax
jz error_free_http
push eax
mov esi, sz_failed
mov edi, paste_url
mov ecx, 2
rep movsd
pop esi
.url_loop:
lodsb
stosb
cmp al, 13
je .url_end
cmp al, 10
je .url_end
cmp al, 0
je .url_end
jmp .url_loop
.url_end:
dec edi
lea ecx, [edi - paste_url + 4]
mov eax, '" -t'
stosd
mov ax, 'O'
stosw
mov [notify_struct.msg], paste_url
mcall 70, notify_struct
mcall 54, 3 ; Delete last slot in the clipboard
mov dword[paste_url-4], ecx
mov dword[paste_url+0], 0 ; Text
mov dword[paste_url+4], 1 ; cp0866
mcall 54, 2, , paste_url-4 ; Write URL to the clipboard
invoke HTTP_free, [identifier]
mcall -1
error_free_http:
invoke HTTP_free, [identifier]
jmp error
error_free_clip_disconnect:
invoke HTTP_disconnect, [identifier]
invoke HTTP_free, [identifier]
error_free_clip:
mcall 68, 13, [clipboard_data]
error:
mov [notify_struct.msg], sz_failed
mcall 70, notify_struct
mcall -1
;---------------------------------------------------------------------
; Data area
;-----------------------------------------------------------------------------
align 4
@IMPORT:
library lib_http, 'http.obj'
import lib_http, \
HTTP_get, 'get', \
HTTP_post, 'post', \
HTTP_receive, 'receive', \
HTTP_find_header_field, 'find_header_field', \
HTTP_free, 'free', \
HTTP_escape, 'escape', \
HTTP_disconnect, 'disconnect'
IM_END:
file_struct:
dd 0 ; read file
dd 0 ; offset
dd 0 ; reserved
dd 32768 ; max file size
.buf dd 0 ; buffer ptr
db 0
dd param
notify_struct:
dd 7 ; run application
dd 0
.msg dd paste_url
dd 0
dd 0
db '/sys/@notify', 0
sz_url db 'http://paste.kolibrios.org/', 0
sz_set_cookie db 'set-cookie', 0
sz_location db 'location', 0
sz_ctype db 'application/x-www-form-urlencoded', 0
sz_failed db '"Pasta!',10,'Paste failed!" -E', 0
sz_paste_head db 'code='
.length = $ - sz_paste_head
sz_paste_tail db '%0D%0A&language=text&webpage='
.length = $ - sz_paste_tail
sz_cookie db 'Cookie: '
cookie db 0
rb 1024
paste_url rb 1024
param rb 1024
identifier dd 0
clipboard_data dd 0
clipboard_data_length dd 0
I_END: