|
@@ -1387,8 +1387,13 @@ static int kcm_attach(struct socket *sock, struct socket *csock,
|
|
if (!csk)
|
|
if (!csk)
|
|
return -EINVAL;
|
|
return -EINVAL;
|
|
|
|
|
|
- /* We must prevent loops or risk deadlock ! */
|
|
|
|
- if (csk->sk_family == PF_KCM)
|
|
|
|
|
|
+ /* Only allow TCP sockets to be attached for now */
|
|
|
|
+ if ((csk->sk_family != AF_INET && csk->sk_family != AF_INET6) ||
|
|
|
|
+ csk->sk_protocol != IPPROTO_TCP)
|
|
|
|
+ return -EOPNOTSUPP;
|
|
|
|
+
|
|
|
|
+ /* Don't allow listeners or closed sockets */
|
|
|
|
+ if (csk->sk_state == TCP_LISTEN || csk->sk_state == TCP_CLOSE)
|
|
return -EOPNOTSUPP;
|
|
return -EOPNOTSUPP;
|
|
|
|
|
|
psock = kmem_cache_zalloc(kcm_psockp, GFP_KERNEL);
|
|
psock = kmem_cache_zalloc(kcm_psockp, GFP_KERNEL);
|