Packages
phoenix
1.2.1
1.8.9
1.8.8
1.8.7
1.8.6
1.8.5
1.8.4
1.8.3
1.8.2
1.8.1
1.8.0
1.8.0-rc.4
1.8.0-rc.3
1.8.0-rc.2
1.8.0-rc.1
1.8.0-rc.0
1.7.24
1.7.23
1.7.22
1.7.21
1.7.20
1.7.19
1.7.18
1.7.17
1.7.16
1.7.15
1.7.14
1.7.13
1.7.12
1.7.11
1.7.10
1.7.9
1.7.8
1.7.7
1.7.6
1.7.5
1.7.4
1.7.3
1.7.2
1.7.1
1.7.0
1.7.0-rc.3
1.7.0-rc.2
1.7.0-rc.1
1.7.0-rc.0
1.6.17
1.6.16
1.6.15
1.6.14
1.6.13
1.6.12
1.6.11
1.6.10
1.6.9
1.6.8
1.6.7
1.6.6
1.6.5
1.6.4
1.6.3
1.6.2
1.6.1
1.6.0
1.6.0-rc.1
1.6.0-rc.0
1.5.15
1.5.14
1.5.13
1.5.12
1.5.11
1.5.10
1.5.9
1.5.8
1.5.7
1.5.6
1.5.5
1.5.4
1.5.3
1.5.2
1.5.1
1.5.0
1.5.0-rc.0
1.4.18
1.4.17
1.4.16
1.4.15
1.4.14
1.4.13
1.4.12
1.4.11
1.4.10
1.4.9
1.4.8
1.4.7
1.4.6
1.4.5
1.4.4
1.4.3
1.4.2
1.4.1
1.4.0
1.4.0-rc.3
1.4.0-rc.2
1.4.0-rc.1
1.4.0-rc.0
1.3.5
1.3.4
1.3.3
1.3.2
1.3.1
1.3.0
1.3.0-rc.3
1.3.0-rc.2
1.3.0-rc.1
1.3.0-rc.0
1.2.5
1.2.4
1.2.3
1.2.2
1.2.1
1.2.0
1.2.0-rc.1
1.2.0-rc.0
1.1.9
1.1.8
1.1.7
1.1.6
1.1.5
1.1.4
1.1.3
1.1.2
1.1.1
1.1.0
1.0.6
1.0.5
1.0.4
1.0.3
1.0.2
1.0.1
1.0.0
0.17.1
0.17.0
0.16.1
0.16.0
0.15.0
0.14.0
0.13.1
0.13.0
0.12.0
0.11.0
0.10.0
0.9.0
0.8.0
0.7.2
0.7.1
0.7.0
0.6.2
0.6.1
0.6.0
0.5.0
0.4.1
0.4.0
0.3.1
0.3.0
0.2.11
0.2.10
0.2.9
0.2.8
0.2.7
0.2.6
0.2.5
0.2.4
0.2.3
0.2.2
0.2.1
0.2.0
0.1.0
Productive. Reliable. Fast. A productive web framework that does not compromise speed or maintainability.
Security advisory:
This version has known vulnerabilities.
View advisories
Current section
Files
Jump to
Current section
Files
lib/phoenix/transports/websocket.ex
defmodule Phoenix.Transports.WebSocket do
@moduledoc """
Socket transport for websocket clients.
## Configuration
The websocket is configurable in your socket:
transport :websocket, Phoenix.Transports.WebSocket,
timeout: :infinity,
serializer: Phoenix.Transports.WebSocketSerializer,
transport_log: false
* `:timeout` - the timeout for keeping websocket connections
open after it last received data, defaults to 60_000ms
* `:transport_log` - if the transport layer itself should log and, if so, the level
* `:serializer` - the serializer for websocket messages
* `:check_origin` - if we should check the origin of requests when the
origin header is present. It defaults to true and, in such cases,
it will check against the host value in `YourApp.Endpoint.config(:url)[:host]`.
It may be set to `false` (not recommended) or to a list of explicitly
allowed origins
* `:code_reloader` - optionally override the default `:code_reloader` value
from the socket's endpoint
## Serializer
By default, JSON encoding is used to broker messages to and from clients.
A custom serializer may be given as module which implements the `encode!/1`
and `decode!/2` functions defined by the `Phoenix.Transports.Serializer`
behaviour.
The `encode!/1` function must return a tuple in the format
`{:socket_push, :text | :binary, String.t | binary}`.
"""
@behaviour Phoenix.Socket.Transport
def default_config() do
[serializer: Phoenix.Transports.WebSocketSerializer,
timeout: 60_000,
transport_log: false]
end
## Callbacks
import Plug.Conn, only: [fetch_query_params: 1, send_resp: 3]
alias Phoenix.Socket.Broadcast
alias Phoenix.Socket.Transport
@doc false
def init(%Plug.Conn{method: "GET"} = conn, {endpoint, handler, transport}) do
{_, opts} = handler.__transport__(transport)
conn =
conn
|> code_reload(opts, endpoint)
|> fetch_query_params()
|> Transport.transport_log(opts[:transport_log])
|> Transport.force_ssl(handler, endpoint, opts)
|> Transport.check_origin(handler, endpoint, opts)
case conn do
%{halted: false} = conn ->
params = conn.params
serializer = Keyword.fetch!(opts, :serializer)
case Transport.connect(endpoint, handler, transport, __MODULE__, serializer, params) do
{:ok, socket} ->
{:ok, conn, {__MODULE__, {socket, opts}}}
:error ->
send_resp(conn, 403, "")
{:error, conn}
end
%{halted: true} = conn ->
{:error, conn}
end
end
def init(conn, _) do
send_resp(conn, :bad_request, "")
{:error, conn}
end
@doc false
def ws_init({socket, config}) do
Process.flag(:trap_exit, true)
serializer = Keyword.fetch!(config, :serializer)
timeout = Keyword.fetch!(config, :timeout)
if socket.id, do: socket.endpoint.subscribe(socket.id, link: true)
{:ok, %{socket: socket,
channels: %{},
channels_inverse: %{},
serializer: serializer}, timeout}
end
@doc false
def ws_handle(opcode, payload, state) do
msg = state.serializer.decode!(payload, opcode: opcode)
case Transport.dispatch(msg, state.channels, state.socket) do
:noreply ->
{:ok, state}
{:reply, reply_msg} ->
encode_reply(reply_msg, state)
{:joined, channel_pid, reply_msg} ->
encode_reply(reply_msg, put(state, msg.topic, msg.ref, channel_pid))
{:error, _reason, error_reply_msg} ->
encode_reply(error_reply_msg, state)
end
end
@doc false
def ws_info({:EXIT, channel_pid, reason}, state) do
case Map.get(state.channels_inverse, channel_pid) do
nil -> {:ok, state}
{topic, join_ref} ->
new_state = delete(state, topic, channel_pid)
encode_reply Transport.on_exit_message(topic, join_ref, reason), new_state
end
end
@doc false
def ws_info(%Broadcast{event: "disconnect"}, state) do
{:shutdown, state}
end
def ws_info({:socket_push, _, _encoded_payload} = msg, state) do
format_reply(msg, state)
end
def ws_info(_, state) do
{:ok, state}
end
@doc false
def ws_terminate(_reason, _state) do
:ok
end
@doc false
def ws_close(state) do
for {pid, _} <- state.channels_inverse do
Phoenix.Channel.Server.close(pid)
end
end
defp put(state, topic, join_ref, channel_pid) do
%{state | channels: Map.put(state.channels, topic, channel_pid),
channels_inverse: Map.put(state.channels_inverse, channel_pid, {topic, join_ref})}
end
defp delete(state, topic, channel_pid) do
case Map.fetch(state.channels, topic) do
{:ok, ^channel_pid} ->
%{state | channels: Map.delete(state.channels, topic),
channels_inverse: Map.delete(state.channels_inverse, channel_pid)}
{:ok, _newer_pid} ->
%{state | channels_inverse: Map.delete(state.channels_inverse, channel_pid)}
end
end
defp encode_reply(reply, state) do
format_reply(state.serializer.encode!(reply), state)
end
defp format_reply({:socket_push, encoding, encoded_payload}, state) do
{:reply, {encoding, encoded_payload}, state}
end
defp code_reload(conn, opts, endpoint) do
reload? = Keyword.get(opts, :code_reloader, endpoint.config(:code_reloader))
if reload?, do: Phoenix.CodeReloader.reload!(endpoint)
conn
end
end