From 3e032c563ff857c49e141ef88694e2423cfa5ee3 Mon Sep 17 00:00:00 2001
From: Florian Pritz <bluewind@xssn.at>
Date: Mon, 26 Jul 2010 17:21:29 +0200
Subject: fb: support filesize checking

Signed-off-by: Florian Pritz <bluewind@xssn.at>
---
 fb | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/fb b/fb
index aca94c8..1ff49ae 100755
--- a/fb
+++ b/fb
@@ -11,12 +11,13 @@
 # Optional: xclip
 #----------------------------------------------------
 
-VERSION="0.6.4"
+VERSION="0.6.5"
 
 DELETE=
 EXTENSION=""
 GET=
 PASTEBIN="http://paste.xinu.at"
+WARNSIZE=10485760
 USERAGENT="fb-client/$VERSION"
 CLIPBOARD=""
 
@@ -26,6 +27,13 @@ do_upload() {
     EXTRA="-F extension=$EXTENSION"
   fi
   TMPFILE=`mktemp "$TMPDIR/data.XXXXXX"`
+  if [ `stat -c %s "$1"` -gt "$WARNSIZE" ]; then
+    WARNSIZE=`curl -s "$PASTEBIN/file/get_max_size"`
+    if [ `stat -c %s "$1"` -gt "$WARNSIZE" ]; then
+      echo "Warning: Your upload is too big and would be rejected. Maximum size is: $WARNSIZE bytes. Skipping..." >&2
+      return
+    fi
+  fi
   curl -# -n -L -A $USERAGENT $EXTRA -F "file=@$1" "$PASTEBIN/file/do_upload" > $TMPFILE
   sed '$d' $TMPFILE >&2
   URL=`tail -1 $TMPFILE`
-- 
cgit v1.2.3-24-g4f1b