This is why I didn't take my kids on a plane until the youngest was 7. But even when I did, I was a bit worried about something like this.

Thankfully, giving him a tablet loaded with videos he likes kept him occupied. In my experience, don't really on plane Wi-Fi! It's always sucked (been unusable) for me. Someone else here mentioned about bringing candy/snacks, which is also a very good idea.