cakedb 0.0.6 → 0.0.7
Sign up to get free protection for your applications and to get access to all the features.
- data/lib/cakedb.rb +19 -11
- metadata +1 -1
data/lib/cakedb.rb
CHANGED
@@ -3,7 +3,7 @@
|
|
3
3
|
require 'socket'
|
4
4
|
|
5
5
|
class CakeDB
|
6
|
-
def initialize( server="127.0.0.1", port=8888 )
|
6
|
+
def initialize( server="127.0.0.1", port=8888, loggingLevel=0 )
|
7
7
|
#puts "Connection Initiated"
|
8
8
|
@server = TCPSocket.new(server, port)
|
9
9
|
@sids = {}
|
@@ -51,6 +51,7 @@ class CakeDB
|
|
51
51
|
result = Array.new
|
52
52
|
|
53
53
|
#How much data we gotta get?
|
54
|
+
|
54
55
|
recv_total = @server.recv(4).unpack("L>")[0]
|
55
56
|
#puts "total is #{recv_total}"
|
56
57
|
if recv_total > 0
|
@@ -61,12 +62,18 @@ class CakeDB
|
|
61
62
|
header_whole = @server.recv(12)
|
62
63
|
header = header_whole.unpack("Q>L>")
|
63
64
|
if header[0] == nil || header[1] == nil
|
64
|
-
|
65
|
+
if loggingLevel == 1
|
66
|
+
puts "You have been eaten by a Grue"
|
67
|
+
end
|
68
|
+
raise "You have been eaten by a Grue"
|
65
69
|
end
|
66
70
|
if header_whole.bytesize !=12
|
67
|
-
|
68
|
-
|
69
|
-
|
71
|
+
if loggingLevel == 1
|
72
|
+
puts "Incomplete header"
|
73
|
+
puts "Length: #{header[1]}"
|
74
|
+
puts "TS: #{header[0]}"
|
75
|
+
end
|
76
|
+
raise "Header Incomplete"
|
70
77
|
end
|
71
78
|
|
72
79
|
#recieve the data, header says how much
|
@@ -80,12 +87,13 @@ class CakeDB
|
|
80
87
|
puts "Incomplete read: #{result[counter]["data"].bytesize} / #{header[1]}"
|
81
88
|
end
|
82
89
|
if result[counter]["data"].bytesize != header[1]
|
83
|
-
|
84
|
-
|
85
|
-
|
86
|
-
|
87
|
-
|
88
|
-
|
90
|
+
if loggingLevel == 1
|
91
|
+
puts "Incomplete read on payload receive"
|
92
|
+
puts "Raw Data: #{resultt[counter]}"
|
93
|
+
puts "Length: #{header[1]} bytes"
|
94
|
+
puts "Raw Data length: #{resultt[counter].bytesize - 12}"
|
95
|
+
puts "Total Length: #{recv_total}"
|
96
|
+
end
|
89
97
|
end
|
90
98
|
recv_total = recv_total - header[1] - 12
|
91
99
|
counter+=1
|