公司如果有相关规章制度规定了这种情况,则不违法。劳动法对用人单位和劳动者双方都有保护和约束,只要相关规定不违反法律,劳动者有义务遵守公司规章制度,如果违反,用人单位有权依据公司制度的对违章者作出惩罚
如果单位的规章制度明确写明上班期间不许吸烟,你违反规定,企业是有权利进行惩罚的
只要企业的规定没有违反相关的法律规定,那就是合法有效的。当然,重大事项需经职工代表大会表决通过。