From: Rob Ashton <robashton@codeofrob.com>
Date: Thu, 9 Apr 2015 18:01:38 +0100
Subject: [PATCH] WS accepts iodata, so byte_size won't work unless we flatten


diff --git a/src/cow_ws.erl b/src/cow_ws.erl
index c89c17a..c025b38 100644
--- a/src/cow_ws.erl
+++ b/src/cow_ws.erl
@@ -580,7 +580,7 @@ masked_frame({binary, Payload}, _) ->
 	[<< 1:1, 0:3, 2:4, 1:1, Len/bits >>, MaskKeyBin, mask(iolist_to_binary(Payload), MaskKey, <<>>)].
 
 payload_length(Payload) ->
-	case byte_size(Payload) of
+	case iolist_size(Payload) of
 		N when N =< 125 -> << N:7 >>;
 		N when N =< 16#ffff -> << 126:7, N:16 >>;
 		N when N =< 16#7fffffffffffffff -> << 127:7, N:64 >>