diff mbox

[net-next] samples/bpf: set max locked memory to ulimited

Message ID 1466568358-37217-1-git-send-email-u9012063@gmail.com
State Accepted, archived
Delegated to: David Miller
Headers show

Commit Message

William Tu June 22, 2016, 4:05 a.m. UTC 
Signed-off-by: William Tu <u9012063@gmail.com>
---
 samples/bpf/sockex2_user.c | 3 +++
 samples/bpf/sockex3_user.c | 3 +++
 2 files changed, 6 insertions(+)

Comments

Alexei Starovoitov June 22, 2016, 4:57 a.m. UTC | #1 
On Tue, Jun 21, 2016 at 09:05:58PM -0700, William Tu wrote:
> Signed-off-by: William Tu <u9012063@gmail.com>

Acked-by: Alexei Starovoitov <ast@kernel.org>
David Miller June 25, 2016, 4:04 p.m. UTC | #2 
From: William Tu <u9012063@gmail.com>
Date: Tue, 21 Jun 2016 21:05:58 -0700

> Signed-off-by: William Tu <u9012063@gmail.com>

Applied.
diff mbox

Patch

diff --git a/samples/bpf/sockex2_user.c b/samples/bpf/sockex2_user.c
index 29a276d..8a4085c 100644
--- a/samples/bpf/sockex2_user.c
+++ b/samples/bpf/sockex2_user.c
@@ -5,6 +5,7 @@ 
 #include "bpf_load.h"
 #include <unistd.h>
 #include <arpa/inet.h>
+#include <sys/resource.h>
 
 struct pair {
 	__u64 packets;
@@ -13,11 +14,13 @@  struct pair {
 
 int main(int ac, char **argv)
 {
+	struct rlimit r = {RLIM_INFINITY, RLIM_INFINITY};
 	char filename[256];
 	FILE *f;
 	int i, sock;
 
 	snprintf(filename, sizeof(filename), "%s_kern.o", argv[0]);
+	setrlimit(RLIMIT_MEMLOCK, &r);
 
 	if (load_bpf_file(filename)) {
 		printf("%s", bpf_log_buf);
diff --git a/samples/bpf/sockex3_user.c b/samples/bpf/sockex3_user.c
index 2617772..d4184ab 100644
--- a/samples/bpf/sockex3_user.c
+++ b/samples/bpf/sockex3_user.c
@@ -5,6 +5,7 @@ 
 #include "bpf_load.h"
 #include <unistd.h>
 #include <arpa/inet.h>
+#include <sys/resource.h>
 
 struct flow_keys {
 	__be32 src;
@@ -23,11 +24,13 @@  struct pair {
 
 int main(int argc, char **argv)
 {
+	struct rlimit r = {RLIM_INFINITY, RLIM_INFINITY};
 	char filename[256];
 	FILE *f;
 	int i, sock;
 
 	snprintf(filename, sizeof(filename), "%s_kern.o", argv[0]);
+	setrlimit(RLIMIT_MEMLOCK, &r);
 
 	if (load_bpf_file(filename)) {
 		printf("%s", bpf_log_buf);