Custom attachment file types

Hi,

I need to restrict attachment file types when user sends a new message (there is currently no restriction and user can upload whatever file he wants).
I find it in the core code but it means at each plugin update, my changes will not be able anymore.
Is there any other way to custom file types ?

Thanks !