[erlang] gen_tcp传输文件原型

有相关的代码需要使用gen_tcp传输文件 把写的原型发出来

-module(tcp).
-export([client/1,server/1]).

client(Port) ->
    Host = "localhost",
    {ok,Sock} = gen_tcp:connect(Host,Port,[binary,{packet,0}]),
    ok = gen_tcp:send(Sock,"/tmp/ecstore.tar"),
    receive_data(Sock,[]).

receive_data(Socket,SoFar) ->
    receive
        {tcp,Socket,Bin} ->
            receive_data(Socket,[Bin|SoFar]);
        {tcp_closed,Socket} ->
            file:write_file("/tmp/test1.tar",list_to_binary(lists:reverse(SoFar)))
    end.

server(Port) ->
    case gen_tcp:listen(Port,[binary,{packet,0}, %% 每个应用程序消息都从一个4字节长的头部开始
                                       {reuseaddr,true},  %% 
                                       {active,true}]) of
        {ok,Listen} ->
            spawn(fun()-> connect(Listen) end);
        {error,Why} ->
            io:format("~p~n",[Why])
    end.

connect(Listen) ->
    case gen_tcp:accept(Listen) of %% 等待暂停并等待一个连接
        {ok,Socket} ->
            spawn(fun() -> connect(Listen) end),
            loop(Socket);
        {error,Why} ->
            io:format("~p~n",[Why])
    end.

loop(Socket) ->
    receive
        {tcp,Socket,Bin} ->
            Str = binary_to_list(Bin),
            %%io:format("Server received = ~p~n",[Str]),
            T = case file:read_file(Str) of
                {ok,Bin1} ->
                    binary_to_list(Bin1);
                {error,Reason} ->
                    io_lib:format("~p~n",[Reason])
            end,
            gen_tcp:send(Socket,[T]),
            gen_tcp:close(Socket);
        {tcp_closed,Socket} ->
            io:format("Server socket closed~n")
    end.