From e282e8608cfad97b9c7a80d05602814b2beb79e4 Mon Sep 17 00:00:00 2001
From: eleksir <eleksir@exs-elm.ru>
Date: Tue, 14 Apr 2020 01:41:32 +0300
Subject: [PATCH] joyproxy.pm: enforce size restrictions before actual
 downloading

---
 lib/joyproxy.pm | 35 ++++++++++++++++++++++++-----------
 1 file changed, 24 insertions(+), 11 deletions(-)

diff --git a/lib/joyproxy.pm b/lib/joyproxy.pm
index acacab2..59ac026 100644
--- a/lib/joyproxy.pm
+++ b/lib/joyproxy.pm
@@ -3,7 +3,7 @@ package joyproxy;
 use strict;
 use warnings "all";
 use lib qw(./lib ./vendor_perl);
-use Fcntl qw(O_RDONLY O_WRONLY);
+use Fcntl qw(O_RDONLY O_WRONLY O_CREAT O_TRUNC);
 use HTTP::Tiny;
 use URI::URL;
 
@@ -48,6 +48,7 @@ sub __urlencode($) {
 	my $str = shift;
 	my $urlobj = url $str;
 	$str = $urlobj->as_string;
+	$urlobj = undef;
 	undef $urlobj;
 	return $str;
 }
@@ -67,10 +68,10 @@ sub __dlfunc($) {
 	$file = "/tmp/joyproxy/" . $file;
 	$url = __urlencode($url);
 
-	my $http = HTTP::Tiny->new();
-	my $response = $http->mirror(
+	my $http = HTTP::Tiny->new(max_size => 10485760);
+
+	my $response = $http->get(
 		$url,
-		$file,
 		{
 			headers => {
 				"Accept" => '*/*',
@@ -84,15 +85,27 @@ sub __dlfunc($) {
 	);
 
 	if ($response->{success}) {
-		my $filesize = (stat($file))[7];
-
-		if ($filesize > 10485760) {
-			unlink $file;
+		sysopen (FILE, $file, O_WRONLY|O_CREAT|O_TRUNC) or do {
+			$http = undef;
+			$response = undef;
 			return (undef, undef);
-		} else {
-			return ($file, $filesize);
-		}
+		};
+
+		my $filesize = length($response->{content});
+
+		syswrite (FILE, $response->{content}, $filesize) or do {
+			$http = undef;
+			$response = undef;
+			return (undef, undef);
+		};
+
+		close FILE;
+		$http = undef;
+		$response = undef;
+		return ($file, $filesize);
 	} else {
+		$http = undef;
+		$response = undef;
 		return (undef, undef);
 	}
 }