X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;ds=sidebyside;f=src%2Fprotocol_edge.c;h=74ace0cc6d05cfeb42dad8f5821c383e978ef36d;hb=8538d6c661a083774e5b54163eaf5cdbcadcda6d;hp=4efb1eceb0bdfbac7723d0050620055253f2b4ff;hpb=c023ad12147aa88810629c110ea6b1ab94267196;p=meshlink

diff --git a/src/protocol_edge.c b/src/protocol_edge.c
index 4efb1ece..74ace0cc 100644
--- a/src/protocol_edge.c
+++ b/src/protocol_edge.c
@@ -108,12 +108,12 @@ bool add_edge_h(meshlink_handle_t *mesh, connection_t *c, const char *request) {
 
 	// Check if devclasses are valid
 
-	if(from_devclass < 0 || from_devclass > _DEV_CLASS_MAX) {
+	if(from_devclass < 0 || from_devclass >= DEV_CLASS_COUNT) {
 		logger(mesh, MESHLINK_ERROR, "Got bad %s from %s: %s", "ADD_EDGE", c->name, "from devclass invalid");
 		return false;
 	}
 
-	if(to_devclass < 0 || to_devclass > _DEV_CLASS_MAX) {
+	if(to_devclass < 0 || to_devclass >= DEV_CLASS_COUNT) {
 		logger(mesh, MESHLINK_ERROR, "Got bad %s from %s: %s", "ADD_EDGE", c->name, "to devclass invalid");
 		return false;
 	}