On Mon, 2009-08-24 at 10:28 -0700, Dan Smith wrote:
> 
> +static int sock_restore_flags(struct socket *sock,
> +                             struct ckpt_hdr_socket *h)
> +{
> +       int ret;
> +       int v = 1;
> +       unsigned long sk_flags = h->sock.flags;
> +       unsigned long sock_flags = h->socket.flags;
> +
> +       if (test_and_clear_bit(SOCK_URGINLINE, &sk_flags)) {
> +               ret = sock_setsockopt(sock, SOL_SOCKET, SO_OOBINLINE,
> +                                     (char *)&v, sizeof(v));
> +               if (ret)
> +                       return ret;
> +       }
> +
> +       if (test_and_clear_bit(SOCK_KEEPOPEN, &sk_flags)) {
> +               ret = sock_setsockopt(sock, SOL_SOCKET, SO_KEEPALIVE,
> +                                     (char *)&v, sizeof(v));
> +               if (ret)
> +                       return ret;
> +       }

Would it make more sense to do this programatically?

cr_sock_restore_flag(struct socket *sock, unsigned long sck_flag,
                     struct ckpt_hdr_socket *h, unsigned long sock_flag)
{
        unsigned long sk_flags = h->sock.flags;
        unsigned long sock_flags = h->socket.flags;

        
        if (!test_and_clear_bit(sk, &sk_flags))
                return 0;

        return sock_setsockopt(sock, sock_flag, SO_OOBINLINE,
                               (char *)&v, sizeof(v));
}

Then, each call becomes:

        ret = cr_sock_restore_flag(sock, cr_sock, SOCK_URGINLINE, SO_OOBINLINE);
        if (ret)
                return ret;

        ret = cr_sock_restore_flag(sock, cr_sock, SOCK_KEEPOPEN, SO_KEEPALIVE);
        if (ret)
                return ret;

Or, you could spell the flags out in a (better named) structure:

struct sock_flagpair
{
        unsigned long sock_flag;
        unsigned long sk_flag;
}       

struct sock_flagpair sock_flagpairs[] = {
        { SOCK_URGINLINE, SO_OOBINLINE },
        { SOCK_KEEPOPEN, SO_KEEPALIVE},
        ...
};

And just walk through the array to do the restore:

static int sock_restore_flags(struct socket *sock,
                             struct ckpt_hdr_socket *h)
{
        int i;

        for (i = 0; i < ARRAY_SIZE(sock_flagpairs); i++) {
                int ret;
                unsigned long sock_flag = sock_flagpairs[i].sock_flag;
                unsigned long sk_flag = sock_flagpairs[i].sk_flag;
                ret = cr_sock_restore_flag(sock, cr_sock, sock_flag, sk_flag);
                if (ret)
                        break;
        }
        ...
}

-- Dave

_______________________________________________
Containers mailing list
[email protected]
https://lists.linux-foundation.org/mailman/listinfo/containers

_______________________________________________
Devel mailing list
[email protected]
https://openvz.org/mailman/listinfo/devel

Reply via email to