Skip to content

Commit

Permalink
net/packet: convert po->pressure to an atomic flag
Browse files Browse the repository at this point in the history
Not only this removes some READ_ONCE()/WRITE_ONCE(),
this also removes one integer.

Signed-off-by: Eric Dumazet <edumazet@google.com>
Signed-off-by: David S. Miller <davem@davemloft.net>
  • Loading branch information
Eric Dumazet authored and David S. Miller committed Mar 17, 2023
1 parent 61edf47 commit 791a3e9
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 7 deletions.
14 changes: 8 additions & 6 deletions net/packet/af_packet.c
Original file line number Diff line number Diff line change
Expand Up @@ -1307,22 +1307,23 @@ static int __packet_rcv_has_room(const struct packet_sock *po,

static int packet_rcv_has_room(struct packet_sock *po, struct sk_buff *skb)
{
int pressure, ret;
bool pressure;
int ret;

ret = __packet_rcv_has_room(po, skb);
pressure = ret != ROOM_NORMAL;

if (READ_ONCE(po->pressure) != pressure)
WRITE_ONCE(po->pressure, pressure);
if (packet_sock_flag(po, PACKET_SOCK_PRESSURE) != pressure)
packet_sock_flag_set(po, PACKET_SOCK_PRESSURE, pressure);

return ret;
}

static void packet_rcv_try_clear_pressure(struct packet_sock *po)
{
if (READ_ONCE(po->pressure) &&
if (packet_sock_flag(po, PACKET_SOCK_PRESSURE) &&
__packet_rcv_has_room(po, NULL) == ROOM_NORMAL)
WRITE_ONCE(po->pressure, 0);
packet_sock_flag_set(po, PACKET_SOCK_PRESSURE, false);
}

static void packet_sock_destruct(struct sock *sk)
Expand Down Expand Up @@ -1409,7 +1410,8 @@ static unsigned int fanout_demux_rollover(struct packet_fanout *f,
i = j = min_t(int, po->rollover->sock, num - 1);
do {
po_next = pkt_sk(rcu_dereference(f->arr[i]));
if (po_next != po_skip && !READ_ONCE(po_next->pressure) &&
if (po_next != po_skip &&
!packet_sock_flag(po_next, PACKET_SOCK_PRESSURE) &&
packet_rcv_has_room(po_next, skb) == ROOM_NORMAL) {
if (i != j)
po->rollover->sock = i;
Expand Down
2 changes: 1 addition & 1 deletion net/packet/internal.h
Original file line number Diff line number Diff line change
Expand Up @@ -117,7 +117,6 @@ struct packet_sock {
spinlock_t bind_lock;
struct mutex pg_vec_lock;
unsigned long flags;
int pressure;
int ifindex; /* bound device */
__be16 num;
struct packet_rollover *rollover;
Expand Down Expand Up @@ -146,6 +145,7 @@ enum packet_sock_flags {
PACKET_SOCK_TP_LOSS,
PACKET_SOCK_HAS_VNET_HDR,
PACKET_SOCK_RUNNING,
PACKET_SOCK_PRESSURE,
};

static inline void packet_sock_flag_set(struct packet_sock *po,
Expand Down

0 comments on commit 791a3e9

Please sign in to comment.