法律分析：
若禁止玩手机是公司明令禁止的，或是多次玩手机严重违反了单位规章制度，公司可以解除合同，除此之外，公司的行为不合法。
法律依据：
《中华人民共和国劳动法》
第二十一条 在试用期中，除劳动者有本法第三十九条和第四十条第一项、第二项规定的情形外，用人单位不得解除劳动合同。用人单位在试用期解除劳动合同的，应当向劳动者说明理由。
第三十九条 为过失性辞退情形，第四十条为无过失辞退情形。