Basic protocol code.

1 files changed, 20 insertions, 2 deletions

diff --git a/zvolcreator.rb b/zvolcreator.rb
index 7d5dfc3..eb79069 100755
--- a/zvolcreator.rb
+++ b/zvolcreator.rb
@@ -2,11 +2,16 @@
 
 require 'zfs'
 require 'socket'
+require 'json'
 
 require_relative 'zfs_monkeypatch'
 load File.dirname($0) + "/" + 'config.rb'
 
+puts "Filesystem base: #{Config[:basefs]}"
+puts "Default user quota: #{Config[:default_user_quota]}"
+puts "Socket path: #{Config[:socketpath]}"
 sockdir = File.dirname(Config[:socketpath])
+puts "socket directory: #{sockdir}"
 
 # Check if socket directory exists, if not - create if with apropriate options
 Dir.mkdir(sockdir, 0700) if not File.exists?(sockdir)
@@ -15,8 +20,21 @@ File.umask(0077)
 server = UNIXServer.open(Config[:socketpath])
 loop do
     Thread.start(server.accept) do |client|
-        client.puts "i'm alive"
-        client.close
+        begin
+            client.puts({:status => 0, :content => {:app => "zvolcreator", :apiversion => 0}}.to_json)
+            response = JSON.parse(client.read)
+            raise ArgumentError.new("Bad status") if response["status"] != 0
+
+            client.puts({:status => 0, :content => {:basefs => Config[:basefs], :default_user_quota => Config[:default_user_quota]}}.to_json)
+            response = JSON.parse(client.read)
+            raise ArgumentError.new("Bad status") if response["status"] != 0
+            raise ArgumentError.new("No user specified") if not response["content"]["user"]
+            raise ArgumentError.new("Volume size not specified") if not response["content"]["volsize"]
+        rescue JSON::ParserError, ArgumentError => e
+            client.puts({:status => 1, :content => e.to_s})
+            client.close
+            Thread.kill
+        end
     end
 end