Well there are some existing scenarios of WSN which you can modify according to your requirments. Have a look at this paper related to using NS 2 for WSN
If you can go with NS-3, I recommend going through the whole Tutorial in order to understand how a simulation script works, and then get into modifying the different classes involved in order to prototype your protocol.
Go for it: https://www.nsnam.org/docs/tutorial/html/introduction.html
Hello guys, for those of you wanting to use NS-3 for your research, I recommend going through the tutorial.
There, you will understand how NS-3 scripts are build, how to combine examples to get the result you need, and what to modify in order to get it to work.
Be patient, it can take time. But the good thing is that most of the intricacies of the protocols are already written and tested. Using the logging modules you can get a grasp of what is happening under the hood. In the end, it is a very efficient way to test your ideas.
Spend time trying to grasp it. It totally worths it.